PAT乙 1005. 继续(3n+1)猜想 (25)

题目地址：PAT乙1005.继续（3n+1）猜想（25）

题目描述：

    卡拉兹(Callatz)猜想已经在1001中给出了描述。在这个题目里，情况稍微有些复杂。当我们验证卡拉兹猜想的时候，为了避免重复计算，可以记录下递推过程中遇到的每一个数。例如对n=3进行验证的时候，我们需要计算3、5、8、4、2、1，则当我们对n=5、8、4、2进行验证的时候，就可以直接判定卡拉兹猜想的真伪，而不需要重复计算，因为这4个数已经在验证3的时候遇到过了，我们称5、8、4、2是被3“覆盖”的数。我们称一个数列中的某个数n为“关键数”，如果n不能被数列中的其他数字所覆盖。现在给定一系列待验证的数字，我们只需要验证其中的几个关键数，就可以不必再重复验证余下的数字。你的任务就是找出这些关键数字，并按从大到小的顺序输出它们。

程序：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;


int main()
{
	int a[1000], key[10000] = {0};
	vector <int> container;
	int K;
	int i = 0;
	cin >> K;
	
	while(K--)
	{
		int num;
		cin >> num;
		a[i++] = num; 
		while(num != 1)
		{
			if (num%2 == 0)
			{
				key[num/2] = 1;
				num /= 2;
			}
			else
			{
				key[(num*3+1)/2] = 1;
				num = (num*3+1)/2;
			}
		}
	}

	for (int j = 0; j < i; ++j)
	{
		if (key[a[j]] != 1)
		{
			container.push_back(a[j]);
 		}
	}
	sort(container.begin(), container.end());
	vector<int>::reverse_iterator p = container.rbegin();
	int s = 0;
	for (;p != container.rend(); p++)
	{
		if (s != 0)
		{
			cout << " " << *p;
		}
		else
		{
			cout << *p;
			s = 1;
		}
	}
	cout<<endl;
	return 0;
}