标签:style class blog c code java
Remove Duplicates from Sorted Array II
Follow up for "Remove Duplicates":
What if duplicates are allowed at
most twice?
For example,
Given sorted array A = [1,1,1,2,2,3]
,
Your function should return length = 5
, and A is
now [1,1,2,2,3]
.
我是Pick One!的,所以一开始没关心"Remove Duplicates"中的要求,不知道是排好序的,直接开始做了。
代码比较简单,建立map,存放每个元素与相应的出现次数。
class Solution { public: int removeDuplicates(int A[], int n) { if(n <= 2) return n; int index = 2; int iter = 2; for(; iter < n; iter ++) { if(A[iter] != A[index-2]) { A[index] = A[iter]; index ++; } } return index; } };
后来发现是排好序的,那么使用in-place做法就可以了。
使用index记录新的A数组下一个的位置,使用iter扫描原始A数组。
核心思想就是:
如果iter扫到的当前元素在index之前已经存在两个(注意,由于A是排好序的,因此只需要判断前两个就行),
那么iter继续前进。否则将iter指向的元素加入index,index与iter一起前进。
class Solution { public: int removeDuplicates(int A[], int n) { if(n <= 2) return n; int index = 2; int iter = 2; for(; iter < n; iter ++) { if(A[iter] != A[index-2]) { A[index] = A[iter]; index ++; } } return index; } };
【LeetCode】Remove Duplicates from Sorted Array II,布布扣,bubuko.com
【LeetCode】Remove Duplicates from Sorted Array II
标签:style class blog c code java
原文地址:http://www.cnblogs.com/ganganloveu/p/3737536.html