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

//PAT1005
//题意：求出数列中不能被其他数覆盖的数，即关键数
//方案：直接使用集合，将被覆盖的数均放入其中，最后遍历数列，若不在集合中，就说明是关键数
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
#include <functional>

using namespace std;

void get_overlap_number(set<int>& st, int n)		//获取数列中被覆盖的数
{
	while ( n != 1 )
	{
		n = (n & 1) ? (3 * n + 1) >> 1 : n >> 1;
		if ( st.find( n ) == st.end() )				//不在集合中，被覆盖数放入集合
		{
			st.insert( n );
		}
		else										//在集合，直接跳出循环，无需再找
			break;
	}
}

int main()
{
	int n;
	cin>>n;
	vector<int> v;
	set<int> st;
	for ( int i=0; i<n; i++ )
	{
		int t;
		cin>>t;
		v.push_back( t );
		get_overlap_number(st, t);
	}
	vector<int> ans;
	for (int i=0; i<v.size(); i++)
	{
		if ( st.find(v[i]) == st.end() )		//遍历集合，若不在集合中，则是关键数
		{
			ans.push_back( v[i] );
		}
	}
	sort(ans.begin(), ans.end(), greater<int>() );		//排序，大到小
	for (int i=0; i<ans.size()-1; i++)
	{
		cout<<ans[i]<<" ";
	}
	cout<<ans[ans.size()-1];
	return 0;
}