Lintcode521.去除重复元素

**描述**

给一个整数数组,去除重复的元素。

你应该做这些事

1.在原数组上操作
2.将去除重复之后的元素放在数组的开头
3.返回去除重复元素之后的元素个数

不需要保持原数组的顺序

样例
例1:

输入:
nums = [1,3,1,4,4,2]
输出:
[1,3,4,2,?,?]
4

解释:

  1. 将重复的整数移动到 nums 的尾部 => nums = [1,3,4,2,?,?].
  2. 返回 nums 中唯一整数的数量 => 4.
    事实上我们并不关心你把什么放在了 ? 处, 只关心没有重复整数的部分.
    例2:

输入:
nums = [1,2,3]
输出:
[1,2,3]
3
挑战
1.O(n)时间复杂度.
2.O(nlogn)时间复杂度但没有额外空间

public static int deduplication(int[] nums) {
    // 使用map来存储每个元素出现的次数
    Map<Integer, Integer> mapData = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        if (mapData.containsKey(nums[i])) {
            mapData.put(nums[i], mapData.get(nums[i]++));
        } else {
            mapData.put(nums[i], 1);
        }
    }
    //i用于将不重复以及重复的其中一个放于前面
    int i = 0;
    //j用于存储重复元素
    int j = nums.length - 1;
    for (Integer num : mapData.keySet()) {
        //重复元素将其置于后面直至其出现次数为1
        while (mapData.get(num) > 1) {
            nums[j--] = num;
            mapData.put(num, mapData.get(num)-1);
        }
        //只为1的情况则将其放入数组的前面
        nums[i++] = num;
        mapData.put(num, mapData.get(num)-1);
    }
    return i;
}

猜你喜欢

转载自blog.csdn.net/leohu_v5/article/details/91411583