leetcode 442. 数组中重复的数据 (思维) &448

给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。

找到所有出现两次的元素。

你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?

示例:

输入:
[4,3,2,7,8,2,3,1]

输出:
[2,3]

最开始毫无思路,结果发现每个数的范围都是[1,n],所以可以考虑这种做法,如果某个位置的数不属于这个位置,就把他交换到对应得位置,即1-N的数,分别对应1-N的位置。如果碰到某个数和他位置上的数相同时,就证明这个数重复了。因为题目要求时找出所有的重复的数,所以我们把这个数扔到结果集合后,就把这个位置的数修改位边界外的一个数,我是把他修改为0.
代码如下:

class Solution {
    
    
public:
    vector<int> findDuplicates(vector<int>& nums) {
    
    
        vector<int> res;
        for(int i=0;i<nums.size();++i){
    
    
            while(nums[i]-1 != i && nums[i] != 0){
    
    
                if(nums[nums[i]-1] == nums[i]){
    
    
                    res.push_back(nums[i]);
                    nums[i] = 0;
                }
                else{
    
    
                    swap(nums[nums[i]-1],nums[i]);
                }
            }
        }
        return res;
    }
};

然后提交看了下,发现自己的速度只超过了25%的人,发现有一种更好的做法,思路如下:
对每个数a进行判断如果a > 0,就修改为a < 0,如果a < 0,那么证明这个位置已经被修改过了,即a这个数出现了两次。

代码如下:

class Solution {
    
    
public:
    vector<int> findDuplicates(vector<int>& nums) {
    
    
        vector<int> res;
        for(int i=0;i<nums.size();++i){
    
    
            int pos = abs(nums[i]);
            if(nums[pos-1] > 0)
                nums[pos-1] = -nums[pos-1];
            else
                res.push_back(pos);
        }
        return res;
    }
};

对于448这个题目,第一种做法,只需要考虑剩下的为0的位置,就是没有出现的数,第二中做法,剩下的仍为正数的位置,就是没有出现的数。

猜你喜欢

转载自blog.csdn.net/zhao5502169/article/details/90199701