Remove elements

include/s0283_move_zeroes.hpp

@@ -0,0 +1,13 @@
+#ifndef S0283_MOVE_ZEROES_HPP
+#define S0283_MOVE_ZEROES_HPP
+
+#include <vector>
+
+using namespace std;
+
+class Solution {
+ public:
+  void moveZeroes(vector<int>& nums);
+};
+
+#endif

notes/src/SUMMARY.md

@@ -1,6 +1,7 @@
 # 数组
 
 - [二分查找](./bin_search.md)
+- [移除元素](./remove_elements.md)
 
 # 经典代码
 

notes/src/remove_elements.md

@@ -0,0 +1,11 @@
+# 移除元素
+
+查找数组中的目标元素 target ，原地移除该元素，返回移除后的数组长度。
+
+双指针法，快指针用于查找新数组的元素，即不是 target 的元素，慢指针用于覆盖，即写入新数组的元素。
+
+## 相关题目
+
+- [27. 移除元素](https://leetcode.com/problems/remove-element/)
+- [24. 删除排序数组中的重复项](https://leetcode.com/problems/swap-nodes-in-pairs/)
+- [283. 移动零](https://leetcode.com/problems/move-zeroes/)

src/s0027_remove_element.cpp

@@ -1,18 +1,14 @@
 #include "s0027_remove_element.hpp"
 
 int Solution::removeElement(vector<int>& nums, int val) {
-  int size = nums.size();
-  if (size == 0) {
-    return 0;
-  }
-  int fast{0}, slow{0};
-  for (; fast < size; ++fast) {
-    if (nums.at(fast) == val) {
-      continue;
-    } else {
-      nums[slow] = nums.at(fast);
-      ++slow;
+  int len = nums.size();
+  int f{0}, s{0};
+  while (f < len) {
+    if (nums[f] != val) {
+      nums[s] = nums[f];
+      ++s;
     }
+    ++f;
   }
-  return slow;
+  return s;
 }

src/s0283_move_zeroes.cpp

@@ -0,0 +1,20 @@
+#include "s0283_move_zeroes.hpp"
+
+void Solution::moveZeroes(vector<int>& nums) {
+  int len = nums.size();
+  if (len <= 1) {
+    return;
+  }
+  int s{0}, f{0};
+  while (f < len) {
+    if (nums[f] != 0) {
+      nums[s] = nums[f];
+      ++s;
+    }
+    ++f;
+  }
+  while (s < len) {
+    nums[s] = 0;
+    ++s;
+  }
+}

tests/s0283_move_zeroes.cpp

@@ -0,0 +1,19 @@
+#include "s0283_move_zeroes.hpp"
+
+#include <gtest/gtest.h>
+
+TEST(Problem283, Case1) {
+  vector<int> i{0, 1, 0, 3, 12};
+  vector<int> o{1, 3, 12, 0, 0};
+  Solution solution;
+  solution.moveZeroes(i);
+  EXPECT_EQ(i, o);
+}
+
+TEST(Problem283, Case2) {
+  vector<int> i{0};
+  vector<int> o{0};
+  Solution solution;
+  solution.moveZeroes(i);
+  EXPECT_EQ(i, o);
+}