aboutsummaryrefslogtreecommitdiff
path: root/other/remove_element.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'other/remove_element.cpp')
-rw-r--r--other/remove_element.cpp28
1 files changed, 28 insertions, 0 deletions
diff --git a/other/remove_element.cpp b/other/remove_element.cpp
new file mode 100644
index 0000000..644f239
--- /dev/null
+++ b/other/remove_element.cpp
@@ -0,0 +1,28 @@
+#include "leetcode.h"
+
+class Solution {
+public:
+ int removeElement(vector<int> &nums, int val) {
+ int idx = 0, k = nums.size();
+ while (idx < k) {
+ if (nums[idx] == val) {
+ // cout << "idx " << idx << endl;
+ // cout << "k " << k << endl;
+ nums[idx] = nums[k - 1];
+ --k;
+ } else
+ ++idx;
+ }
+ return k;
+ }
+};
+
+int main() {
+ Solution obj;
+ vector<int> nums = {0, 1, 2, 2, 3, 0, 4, 2};
+ int val = 2;
+ for (int i = 0; i < nums.size(); i++)
+ cout << nums[i] << ' ';
+ cout << endl << nums.size() << endl;
+ cout << obj.removeElement(nums, val);
+}