2014年爱奇艺的校招题目-大题第八题

爱奇艺的一道笔试题目：

给定一个数组A中有N个元素，元素范围为1到N,1到N中元素有的没有出现，有的出现多次

要求：统计数组A中没有出现的元素和出现多次的元素个数

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

思路：

对于数组A = {1, 2, 2, 3}

A[1] = 1 > 0 时 A[1] == 1 ，A[1] = 1是新出现的数字，并且A[i]=1等于其下标，所以A[1] = -1 , 数组变为-1, 2, 2, 3

同理A[2] = 2 > 0 时 A[2] == 2 所以A[2] = -1 , 数组变为-1, -1, 2, 3

A[3] = 2 > 0 时 因为A[3] = 2 不是新出现的数字， 所以A[A[3]]减一，同时下标为3的数字目前没有出现过，所以A[3]设为0，此时数组变为-1， -2， 0， 3

A[4] = 3 > 0 时 因为A[4] = 2 是新出现的数字， 所以A[A[3]]设置为-1，同时下标为4的数字目前没有出现过，所以A[4]设为0，此时数组变为-1， -2， -1， 0

#include <iostream>
using namespace std;

#define NSIZ 1000

void getNum(int arr[], int n)
{
int i ,tmp;
for (i = 1;i <= n; ++i)
{
if (arr[i-1] > 0)
{
tmp = arr[arr[i - 1] - 1];
if (tmp < 0)
{
--arr[arr[i - 1] - 1];
arr[i - 1] = 0;
}
else if (arr[i - 1] > 0)
{
bool flag = i == arr[i - 1];
arr[arr[i - 1] - 1] = -1;

if (!flag )
{
if (tmp > 0)
{
arr[i-1] = tmp;
}
}
else
{
if (tmp == i)
{
arr[i - 1] = -1;
}
else
{
arr[i - 1] = 0;
}
}
--i;
}

}

}

for (i = 0;i < n;++i)
{

printf("%d %d\n", i + 1, -arr[i]);
}
}

int main()
{

int arr[] = {2, 2, 2, 4};
int n = sizeof(arr)/sizeof(arr[0]);
getNum(arr, n);
return 0;
}