【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 <vector>
#include <string>
#include <algorithm>
#include <sstream>
using std::istringstream;
using std::string;
using std::sort;
using std::reverse;
using std::vector;
using std::cin;
using std::cout;
using std::endl;
int main() {
	vector<int> original_int;
	vector<int> overrided_int; //	保存被“覆盖”的数
	vector<int> key_int;
	int K;
	//cin >> K;

	string line;
	getline(cin, line);
	istringstream count(line);
	count >> K;

	int value;
	getline(cin, line);
	istringstream record(line); // 使用stringstream来读取一行字符串中的数字
	for (unsigned i = 0; i < K; i++) {
		record >> value;
		original_int.push_back(value);
	}

	// 得到被“覆盖”的数
	for (unsigned i = 0; i < original_int.size(); i++) {
		int n = original_int[i]; //注意不要改变origin数组的值 后面还会用到
		while (n != 1) {
			if (n % 2 == 0) {
				n = n / 2;
			}
			else {
				n = (3 * n + 1) / 2;
			}
			overrided_int.push_back(n);
		}
	}
	
	// 找出没有被覆盖的数
	for (unsigned i = 0; i < original_int.size(); i++) {
		bool isKey = true;
		for (unsigned j = 0; j < overrided_int.size(); j++) {
			if (original_int[i] == overrided_int[j]) {
				isKey = false;
			}
		}
		if (isKey) {
			key_int.push_back(original_int[i]);
		}
	}
	// 排序(从小到大）
	sort(key_int.begin(), key_int.end()); 
	// 反转
	reverse(key_int.begin(), key_int.end());
	// 打印
	for (unsigned int i = 0; i < key_int.size(); i++) {
		if (i != key_int.size() - 1) {
			cout << key_int[i] << " ";
		}
		else {
			cout << key_int[i];
		}
		
	}
	return 0;
}