leetcode题解-80. Remove Duplicates from Sorted Array II

来源:互联网 发布:oracle 数据库日志 编辑:程序博客网 时间:2024/06/05 04:29

题目:

Follow up for "Remove Duplicates":What if duplicates are allowed at most twice?For example,Given sorted array nums = [1,1,1,2,2,3],Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.

题目思路,因为数组已经是排序数组,所以我们只需要考虑让每个元素的重复个数限制在2以内即可,使用一个标志来标识。首先想到的最简单的思路应该是使用map来标记每个数的出现次数,大于2时不再进行操作,2以内的则对数组相应元素进行赋值。代码入下:

    public int removeDuplicates(int[] nums) {        int count = 0;        Map<Integer, Integer> map = new HashMap<>();        for(int i=0; i<nums.length; i++){            if(!map.containsKey(nums[i])) {                nums[count] = nums[i];                map.put(nums[i], 1);                count++;            }else if(map.get(nums[i]) == 1) {                nums[count] = nums[i];                map.put(nums[i], map.get(nums[i]) + 1);                count++;            }        }        return count;    }

这种方法因为使用了hashmap导致算法效率较低,我们考虑能不能仅使用一个标记来依次表示所有数字出现次数,就有了下面的代码,使用一个boolean型的flag标志前后相邻数字是否相等:

    public static int removeDuplicates1(int[] nums){        if(nums.length<=1)            return nums.length;        int count = 1;        boolean flag = false;        for(int i=1; i<nums.length; i++){            if(nums[i] == nums[i-1] && !flag){                nums[count] = nums[i];                count++;                flag = true;            }else if(nums[i] != nums[i-1]){                nums[count] = nums[i];                count++;                flag = false;            }        }        return count;    }

另外,在discuss中看到了一种极其简洁的代码,如下所示,很6:

    public int removeDuplicates2(int[] nums) {        int i = 0;        for (int n : nums)            if (i < 2 || n > nums[i-2])                nums[i++] = n;        return i;    }
0 0
原创粉丝点击