bit-map牛刀小试：数组test[X]的值全部在区间[1, 8000]中， 现要输出test中重复的数。要求:1. 不能改变原数组； 2.时间复杂度为O(X)；3.除test外空间不超过1KB

       先来看看这个题目：数组test[X]的值全部在区间[1, 8000]中， 现要输出test中重复的数。要求:1. 不能改变原数组; 2.时间复杂度为O(X);3.除test外空间不超过1KB.

 

       好， 我们先给出一个不限空间的解法（为了程序方便， 假设X为10, 实际上可能很大）：

 

#include <iostream>
using namespace std;

#define X 10
#define N 8000

// 输出重复的数字
void printDup(const int test[], int n)
{
	int a[N] = {0};
	int i = 0;
	for(i = 0; i < n; i++)
	{
		a[test[i] - 1]++;
	}

	for(i = 0; i < N; i++) // 注意， 此处是N而不是n
	{
		if(a[i] > 1)
		{
			cout << i + 1 << endl;
		}
	}
}

int main(void) 
{
	int test[X] = {1, 2, 3, 4, 2, 5, 6, 7, 5, N};
	printDup(test, X);
	
	return 0;
}

 

        结果为：

2

5

 

       显然， 上述程序在空间上超标（且当X<N=8000时， 时间超标）， 究其原因是： 让一个int去存一个二值状态， 太浪费空间了， 和不用一个bit来存呢？ 所以， 我们自然想到了用bit-map来操作， 如下：

 

#include <iostream>
using namespace std;

#define X 10

#define BIT_INT 32   // 1个int可以标志32个坑
#define SHIFT 5
#define MASK 0x1f
#define N 8000
int a[1 + N / BIT_INT]; // 需要1 + N / BIT_INT 个整数来标志N个事物

// 将所有位都初始化为0状态
void setAllZero()
{
	memset(a, 0, (1 + N / BIT_INT) * sizeof(int));
}

// 设置第i位为1
void setOne(int i)
{
	a[i >> SHIFT] |= (1 << (i & MASK));
}

// 设置第i位为1
void setZero(int i)
{
	a[i >> SHIFT] &= ~(1 << (i & MASK));
}

// 检查第i位的值
int getState(int i)
{
	return (a[i >> SHIFT] & (1 << (i & MASK))) && 1;
}

// 输出重复的数字
void printDup(const int test[], int n)
{
	int i = 0;
	for(i = 0; i < n; i++)
	{
		int state = getState(test[i] - 1);
		if(0 == state)
		{
			setOne(test[i] - 1);
		}
		else 
		{
			cout << test[i] << endl;
		}

	}
}

int main(void) 
{
	setAllZero();
	int test[X] = {1, 2, 3, 4, 2, 5, 6, 7, 5, N};
	printDup(test, X);
	
	return 0;
}

       结果为：

 

2

5

 

      且满足题目要求。 但是， 我随后发现这个程序还有个问题: 当test数组中某元素出现次数大于2时， 会重复输出， 比如：

 

#include <iostream>
using namespace std;

#define X 10

#define BIT_INT 32   // 1个int可以标志32个坑
#define SHIFT 5
#define MASK 0x1f
#define N 8000
int a[1 + N / BIT_INT]; // 需要1 + N / BIT_INT 个整数来标志N个事物

// 将所有位都初始化为0状态
void setAllZero()
{
	memset(a, 0, (1 + N / BIT_INT) * sizeof(int));
}

// 设置第i位为1
void setOne(int i)
{
	a[i >> SHIFT] |= (1 << (i & MASK));
}

// 设置第i位为1
void setZero(int i)
{
	a[i >> SHIFT] &= ~(1 << (i & MASK));
}

// 检查第i位的值
int getState(int i)
{
	return (a[i >> SHIFT] & (1 << (i & MASK))) && 1;
}

// 输出重复的数字
void printDup(const int test[], int n)
{
	int i = 0;
	for(i = 0; i < n; i++)
	{
		int state = getState(test[i] - 1);
		if(0 == state)
		{
			setOne(test[i] - 1);
		}
		else 
		{
			cout << test[i] << endl;
		}

	}
}

int main(void) 
{
	setAllZero();
	int test[X] = {1, 2, 3, 4, 2, 5, 6, 2, 5, N}; // 2出现3次
	printDup(test, X);
	
	return 0;
}

      结果为：

 

2

2

5

 

     我想了一下， 暂时没有想到只打印2, 5且符合题意的方法。 如果大家有好的思路， 欢迎赐教