diff --git a/solution/0000-0099/0026.Remove Duplicates from Sorted Array/README.md b/solution/0000-0099/0026.Remove Duplicates from Sorted Array/README.md index deb7645d78123..0a0e171d48ee3 100644 --- a/solution/0000-0099/0026.Remove Duplicates from Sorted Array/README.md +++ b/solution/0000-0099/0026.Remove Duplicates from Sorted Array/README.md @@ -120,6 +120,16 @@ public: }; ``` +```cpp +class Solution { +public: + int removeDuplicates(vector& nums) { + nums.erase(unique(nums.begin(), nums.end()), nums.end()); + return nums.size(); + } +}; +``` + ### **Go** ```go diff --git a/solution/0000-0099/0026.Remove Duplicates from Sorted Array/Solution.cpp b/solution/0000-0099/0026.Remove Duplicates from Sorted Array/Solution.cpp index 32e8c18858c09..905f169e9d64b 100644 --- a/solution/0000-0099/0026.Remove Duplicates from Sorted Array/Solution.cpp +++ b/solution/0000-0099/0026.Remove Duplicates from Sorted Array/Solution.cpp @@ -1,10 +1,7 @@ class Solution { public: int removeDuplicates(vector& nums) { - int i = 0; - for (int& num : nums) - if (i < 1 || num != nums[i - 1]) - nums[i++] = num; - return i; + nums.erase(unique(nums.begin(), nums.end()), nums.end()); + return nums.size(); } -}; \ No newline at end of file +};