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不能被数列中的其他数字所覆盖。

现在给定一系列待验证的数字，我们只需要验证其中的几个关键数，就可以不必再重复验证余下的数字。你的任务就是找出这些关键数字，并按从大到小的顺序输出它们。

输入格式：每个测试输入包含1个测试用例，第1行给出一个正整数K(<100)，第2行给出K个互不相同的待验证的正整数n(1<n<=100)的值，数字间用空格隔开。

输出格式：每个测试用例的输出占一行，按从大到小的顺序输出关键数字。数字间用1个空格隔开，但一行中最后一个数字后没有空格。

输入样例：

6
3 5 6 7 8 11

输出样例：

7 6

#include<iostream>
#include<algorithm>
using namespace std;
int a[101];									//存放输入的第二行的k个数字 
int sto[100000];							//！！！存放经过(3n+1)猜想过程中的数，测试之前只开了1000，导致第四个点一直是错误，后来改成100000就通过了 
bool cmp(int a,int b)						//sort的降序排列函数 
{
	return a>b;
}
int main()
{
	int k;
	cin>>k;
	int kt=k;								//最后会用sort给剩下的元素排序，所以要用kt减去被覆盖过的元素个数，最后kt剩下的就是最终要输出的元素 
	for(int i=0;i<k;i++)
	{
		cin>>a[i];
	}
	int m=0;								//m是累计猜想过程中出现的数字 
	for(int j=0;j<k;j++)
	{
		int tmp;							//因为不能在循环的过程中改变a[]中的元素 ，所以用一个中间变量代替 
		tmp=a[j];
		int key=0;
		while((tmp!=1)&&(tmp>0))			//因为被覆盖的数字会赋成-1，所以用tmp>0来减少循环次数 
		{
			key=1;
			if(tmp%2==0)
			{
				tmp=tmp/2;
				sto[m++]=tmp;
			}
			else
			{
				tmp=(tmp*3+1)/2; 
				sto[m++]=tmp;
			}
		}
		if(key==1)							//减少-1的影响 
		{
			for(int d=0;d<m;d++)
			{
				if(sto[d]>0)
				{
					for(int t=0;t<k;t++)
					{
						if((sto[d]==a[t])&&a[t]>0)
						{
							a[t]=-1;		//因为每个元素都不相同将两个数组中覆盖的元素都赋成-1 
							sto[d]=-1;
							--kt;
						}
					}
				}	
			}
		}
	}
	sort(a,a+k,cmp);						//降序排列，把-1都放在后面，方便读取 
	for(int end=0;end<kt;end++)				//kt的个数是最终剩下的关键数 的个数 
	{
		if(end!=kt-1)cout<<a[end]<<" ";		 
		else cout<<a[end];
	}
	return 0;
}