【leetcode C语言实现】442_数组中重复的数据

题目描述

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

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

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

示例：

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

输出:
[2,3]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-all-duplicates-in-an-array

解题思路

数组中每一个元素都在1~n之间，故依次遍历整个数组中的每个元素，对数组中以(元素-1)为下标的元素加上n，对于重复两次的元素，其最终大小应为(原始值+2n)，而只出现过一次的元素其大小必然小于2n，根据这个结论可以找出数组中所有出现两次的元素。因为最终返回的是一个数组，且题目要求不用到任何额外空间，所以将符合要求的元素从中依次替换掉原数组中的元素。

代码

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* findDuplicates(int* nums, int numsSize, int* returnSize)
{
    int i;
    *returnSize = 0;

    for(i = 0; i < numsSize; i++)
    {
        nums[(nums[i] - 1) % numsSize] += numsSize;
    }
    for(i = 0; i < numsSize; i++)
    {
        if(nums[i] > 2 * numsSize)
        {
            nums[*returnSize] = i + 1;
            (*returnSize)++;
        }
    }

    return nums;
}

测试代码及结果

void main()
{
    int a[] = {4,3,2,7,8,2,3,1};
    int * return_size;
    int * nums = findDuplicates(a, sizeof(a)/sizeof(int), return_size);
    for(int i = 0; i < (*return_size); i++)
    {
        printf("%d ", nums[i]);
    } 
    printf("\n");
}

执行结果

时间复杂度O(n)，空间复杂度O(1)。