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;
}