算法leetcode|80. 删除有序数组中的重复项 II(rust重拳出击)



80. 删除有序数组中的重复项 II:

给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。

不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。

说明

为什么返回数值是整数,但输出的答案是数组呢?

请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

样例 1:

输入:
	
	nums = [1,1,1,2,2,3]
	
输出:
	
	5, nums = [1,1,2,2,3]
	
解释:
	
	函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3。 不需要考虑数组中超出新长度后面的元素。

样例 2:

输入:
	
	nums = [0,0,1,1,1,1,2,3,3]
	
输出:
	
	7, nums = [0,0,1,1,2,3,3]
	
解释:
	
	函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3。不需要考虑数组中超出新长度后面的元素。

提示:

  • 1 <= nums.length <= 3 * 104
  • -104 <= nums[i] <= 104
  • nums 已按升序排列

分析:

  • 面对这道算法题目,二当家的再次陷入了沉思。
  • 这道题和 26. 删除有序数组中的重复项 非常相似,不同的是,本题允许相同元素最多有一次重复。
  • 如果要删除重复,使用映射表等数据结构去计数是比较直观的想法,但是题目要求只能使用常数级别的额外空间,所以不能靠装备只能靠技巧了。
  • 题目中有一个非常重要的信息,那就是有序,这意味着相同的数字都会挨在一起,也就是最多允许当前下标和前一个下标的数字相同,判断当前下标和前面的前面的下标数字是否相同即可快速判断重复,即 nums[i]nums[i - 2] 相同时,就要删除该元素。
  • 由于要求是原地修改,所以我们需要一个新数组的长度(本身返回值也要求新数组的长度),还需要一个遍历下标,即双指针,只需要遍历一次,并且只需要常数级的额外空间,非常高效。
  • 另外如果数组元素少于3个(或者说数组长度小于/等于2个),是肯定满足题意的,可以直接返回,不需要做任何处理。

题解:

rust:

impl Solution {
    
    
    pub fn remove_duplicates(nums: &mut Vec<i32>) -> i32 {
    
    
        let n = nums.len();
        if n <= 2 {
    
    
            return n as i32;
        }
        let (mut slow, mut fast) = (2, 2);
        while fast < n {
    
    
            if nums[slow - 2] != nums[fast] {
    
    
                nums[slow] = nums[fast];
                slow += 1;
            }
            fast += 1;
        }
        return slow as i32;
    }
}

go:

func removeDuplicates(nums []int) int {
    
    
	n := len(nums)
	if n <= 2 {
    
    
		return n
	}
	slow, fast := 2, 2
	for fast < n {
    
    
		if nums[slow-2] != nums[fast] {
    
    
			nums[slow] = nums[fast]
			slow++
		}
		fast++
	}
	return slow
}

c++:

class Solution {
    
    
public:
    int removeDuplicates(vector<int>& nums) {
    
    
        const int n = nums.size();
        if (n <= 2) {
    
    
            return n;
        }
        int slow = 2, fast = 2;
        while (fast < n) {
    
    
            if (nums[slow - 2] != nums[fast]) {
    
    
                nums[slow] = nums[fast];
                ++slow;
            }
            ++fast;
        }
        return slow;
    }
};

python:

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        slow = 2
        for fast in range(2, len(nums)):
            if nums[fast] != nums[slow - 2]:
                nums[slow] = nums[fast]
                slow += 1
        return slow


java:

class Solution {
    
    
    public int removeDuplicates(int[] nums) {
    
    
        final int n = nums.length;
        if (n <= 2) {
    
    
            return n;
        }
        int slow = 2, fast = 2;
        while (fast < n) {
    
    
            if (nums[slow - 2] != nums[fast]) {
    
    
                nums[slow] = nums[fast];
                ++slow;
            }
            ++fast;
        }
        return slow;
    }
}

非常感谢你阅读本文~
欢迎【点赞】【收藏】【评论】三连走一波~
放弃不难,但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子:https://le-yi.blog.csdn.net/ 博客原创~


猜你喜欢

转载自blog.csdn.net/leyi520/article/details/132802590