🎨27.移除元素

README.md

@@ -9,7 +9,7 @@
 - [20.有效的括号](leetcode/easy/20.go)
 - [21.合并两个有序链表](leetcode/easy/21.go)
 - [26.删除有序数组中的重复项](leetcode/easy/t26/solution.go)
-- [27.移除元素](leetcode/easy/27.go)
+- [27.移除元素](leetcode/easy/t27/solution.go)
 - [28.实现 strStr()](leetcode/easy/28.go)
 - [35.搜索插入位置](leetcode/easy/35.go)
 - [53.最大子序和](leetcode/easy/53.go)

leetcode/easy/t27/solution.go

@@ -0,0 +1,15 @@
+package t27
+
+// 27. 移除元素
+// https://leetcode.cn/problems/remove-element
+func removeElement(nums []int, val int) int {
+	slow, fast := 0, 0
+	for fast < len(nums) {
+		if nums[fast] != val { // 快指针不等于移除元素时 交换快慢指针的值 慢指针前进·
+			nums[slow] = nums[fast]
+			slow++
+		}
+		fast++
+	}
+	return slow
+}

leetcode/easy/t27/solution_test.go

@@ -0,0 +1,32 @@
+package t27
+
+import "testing"
+
+func TestRemoveElement(t *testing.T) {
+	tests := []struct {
+		name   string
+		nums   []int
+		val    int
+		expect int
+	}{
+		{
+			name:   "case 1",
+			nums:   []int{3, 2, 2, 3},
+			val:    3,
+			expect: 2,
+		},
+		{
+			name:   "case 2",
+			nums:   []int{0, 1, 2, 2, 3, 0, 4, 2},
+			val:    2,
+			expect: 5,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := removeElement(tt.nums, tt.val); got != tt.expect {
+				t.Errorf("removeElement() = %v, want %v", got, tt.expect)
+			}
+		})
+	}
+}