Add solution and test-cases for problem 341

Author: Cori1109

leetcode/301-400/0341.Flatten-Nested-List-Iterator/README.md

@@ -1,26 +1,36 @@
 # [341.Flatten Nested List Iterator][title]
 
-> [!WARNING|style:flat]
-> This question is temporarily unanswered if you have good ideas. Welcome to [Create Pull Request PR](https://github.com/kylesliu/awesome-golang-algorithm)
-
 ## Description
+You are given a nested list of integers `nestedList`. Each element is either an integer or a list whose elements may also be integers or other lists. Implement an iterator to flatten it.
 
-**Example 1:**
+Implement the `NestedIterator` class:
 
+- `NestedIterator(List<NestedInteger> nestedList)` Initializes the iterator with the nested list `nestedList`.
+- `int next()` Returns the next integer in the nested list.
+- `boolean hasNext()` Returns `true` if there are still some integers in the nested list and `false` otherwise.
+Your code will be tested with the following pseudocode:
 ```
-Input: a = "11", b = "1"
-Output: "100"
+initialize iterator with nestedList
+res = []
+while iterator.hasNext()
+    append iterator.next() to the end of res
+return res
 ```
+If `res` matches the expected flattened list, then your code will be judged as correct.
 
-## 题意
-> ...
+**Example 1:**
 
-## 题解
+```
+Input: nestedList = [[1,1],2,[1,1]]
+Output: [1,1,2,1,1]
+Explanation: By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,1,2,1,1].
+```
 
-### 思路1
-> ...
-Flatten Nested List Iterator
-```go
+__Example 2:__
+```
+Input: nestedList = [1,[4,[6]]]
+Output: [1,4,6]
+Explanation: By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,4,6].
 ```
 
 

leetcode/301-400/0341.Flatten-Nested-List-Iterator/Solution.go

@@ -1,5 +1,69 @@
 package Solution
 
-func Solution(x bool) bool {
-	return x
+type NestedInteger struct {
+	Val   int
+	isInt bool
+	list  []*NestedInteger
+}
+
+func (this NestedInteger) IsInteger() bool {
+	return this.isInt
+}
+
+func (this NestedInteger) GetInteger() int {
+	return this.Val
+}
+
+func (this NestedInteger) SetInteger(value int) {
+
+}
+func (this NestedInteger) Add(ele NestedInteger) {
+
+}
+
+func (this NestedInteger) GetList() []*NestedInteger {
+	return this.list
+}
+
+type NestedIterator struct {
+	Raw []*NestedInteger
+	idx int
+}
+
+func Constructor3(nestedList []*NestedInteger) *NestedIterator {
+	raw := make([]*NestedInteger, 0)
+	deepTraverse(nestedList, &raw)
+	return &NestedIterator{
+		Raw: raw,
+		idx: 0,
+	}
+}
+
+func deepTraverse(nestedList []*NestedInteger, raw *[]*NestedInteger) {
+	for idx, ele := range nestedList {
+		if ele.IsInteger() {
+			*raw = append(*raw, nestedList[idx])
+			continue
+		}
+		deepTraverse(ele.GetList(), raw)
+	}
+}
+
+func (this *NestedIterator) Next() int {
+	r := this.Raw[this.idx].GetInteger()
+	this.idx++
+	return r
+}
+
+func (this *NestedIterator) HasNext() bool {
+	return this.idx < len(this.Raw)
+}
+
+func Solution(nestedList []*NestedInteger) []int {
+	o := Constructor3(nestedList)
+	expect := make([]int, 0)
+	for o.HasNext() {
+		expect = append(expect, o.Next())
+	}
+	return expect
 }

leetcode/301-400/0341.Flatten-Nested-List-Iterator/Solution_test.go

@@ -10,12 +10,22 @@ func TestSolution(t *testing.T) {
 	//	测试用例
 	cases := []struct {
 		name   string
-		inputs bool
-		expect bool
+		inputs []*NestedInteger
+		expect []int
 	}{
-		{"TestCase", true, true},
-		{"TestCase", true, true},
-		{"TestCase", false, false},
+		{"TestCase1", []*NestedInteger{{Val: 1, isInt: true}, {isInt: false, list: []*NestedInteger{{Val: 2, isInt: true}, {Val: 3, isInt: true}}}, {Val: 1, isInt: true, list: nil}}, []int{1, 2, 3, 1}},
+		{"TestCase2", []*NestedInteger{{Val: 1, isInt: true}, {Val: 2, isInt: true}, {Val: 3, isInt: true}}, []int{1, 2, 3}},
+		{"TestCase3", []*NestedInteger{
+			{Val: 1, isInt: true},
+			{isInt: false, list: []*NestedInteger{
+				{isInt: false, list: []*NestedInteger{{Val: 2, isInt: true}}},
+				{isInt: false, list: []*NestedInteger{
+					{Val: 3, isInt: true},
+					{Val: 4, isInt: true},
+				}},
+			}},
+			{Val: 5, isInt: true}},
+			[]int{1, 2, 3, 4, 5}},
 	}
 
 	//	开始测试