统计数组中各个元素出现的次数，元素取值范围为：1到N

一道非常巧妙的算法

问题描述

• 给定一个数组a,长度为N,元素取值范围为[1, N]
• 统计各个元素出现的次数，要求时间复杂度为O(N),空间复杂度为O(1)
• 可以改变原来的数组结构

解题思路

1. 从第一个元素开始遍历，每遍历到一个元素，将（该元素值-1记为index）作为一个下标值，令该下标对应的元素为index+1出现的次数
2. 若下标index为负值，说明元素的值已处理过（现在记录是次数），跳过
3. 若index为非负值；判断，若a[index]为正，则交换数据，若a[index]非正，则执行减一，且自身的值设为0（表示已经用过了）
4. 最后统计次数（次数是负数）

代码

#include<bits/stdc++.h>
using namespace std;

int main(){
	int n = 6;
	vector<int> a = { 6, 2, 4, 1, 2, 5 };
	int i = 0;
	while (i<n){
		int temp = a[i] - 1;       
		if (temp<0){      //判断该数据是否用过
			i++;
			continue;
		}
		if (a[temp] > 0){
			a[i] = a[temp];
			a[temp] = -1;
		}
		else
		{
			a[i] = 0;     //该数据已经使用过，且表示元素i+1出现0次
			a[temp]--;
		}
	}
	for (int j = 0; j < n; ++j){
		cout << j + 1 << "出现次数" << -a[j]<< endl;
	}
	return 0;
}