diff options
author | jack <[email protected]> | 2024-06-14 13:13:15 -0600 |
---|---|---|
committer | jack <[email protected]> | 2024-06-14 13:13:15 -0600 |
commit | d343a3b18dbaca9297d4f7b2280c4a16f0a5c8aa (patch) | |
tree | 96df922dae2a5073d9a50ea70c7aae6aa37e3ebe /remove_element.cpp | |
download | leetcode-d343a3b18dbaca9297d4f7b2280c4a16f0a5c8aa.tar.gz leetcode-d343a3b18dbaca9297d4f7b2280c4a16f0a5c8aa.zip |
initial
Diffstat (limited to 'remove_element.cpp')
-rw-r--r-- | remove_element.cpp | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/remove_element.cpp b/remove_element.cpp new file mode 100644 index 0000000..644f239 --- /dev/null +++ b/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); +} |