Leetcode 442. 数组中重复的数据 C++ 重要

题目描述

给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
示例:

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

输出:
[2,3]

解答

这种做法来自评论区。时间复杂度 O ( N ) O(N) ,空间复杂度 O ( 1 ) O(1)
做法:
因为数组元素已经规定了是在[1,n]之间,所以问题可以变得简单。数组大小是n,可能的元素一共有n个,即一个数组下标对于一个数。
以数组中的元素值减去1作为下标。如果对应的元素大于零,那么说明是第一次出现,乘以负一,标记这个下标对应元素已经出现过一次。如果下次再碰到为负数,那么就将元素保存。
注意:
这个题,已经说明了重复最多两次。如果是出现多次,那么要注意去重。

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

猜你喜欢

转载自blog.csdn.net/yuanliang861/article/details/89298225