LeetCode-80. Remove Duplicates from Sorted Array II

来源:互联网 发布:淘宝买家提取神器 编辑:程序博客网 时间:2024/05/22 14:09

题目描述

  1. 给定一个升序排序的int数组,允许每个数字最多重复两次,多余的数字不允许出现在数组的0-(n-1)的位置上,n即为去重后数组的长度。
  2. 例如:给定数组【1,1,1,2,2,3】,返回去重后数组的长度为5,并且数组的前5个元素为【1,1,2,2,3】.

解题思路

  1. 利用两个位置指针first和second,变量tem记录上一个元素,count表示与tem相等的元素的个数,first代表当前元素的位置,该位置的元素需要与tem作比较,如果两元素相同,count++;反之tem赋值为first位置的元素,同时count置为1。如果count大于2,说明该位置元素是多余的,first++;反之,该位置的元素不是多余的,需要与second位置的元素交换,同时first和second都加1。

代码

public class Solution {    public int removeDuplicates(int[] nums) {        if(nums==null || nums.length==0)            return 0;        int first=1;        int second=1;        int count=1;        int tem=nums[0];        for(;first<nums.length;){            if(nums[first]==tem){                count++;            }else{                tem=nums[first];                count=1;            }            if(count>2){                first++;            }else{                nums[second]=nums[first];                first++;                second++;            }        }        return second;    }}