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

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，100]的值，数字间用空格隔开。

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

输入样例：
6
3 5 6 7 8 11
输出样例：
7 6

注意：

• 第一次提交异常退出，刚开始定义的数组长度为101，只想着记录输入的数字是不是关键字，但是如果是超过35的奇数，就会发生溢出

代码：

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

int main()
{
    vector<int> res(301);//输入可能是99，所以定义长度为301
    int N;
    cin >> N;
    for(int i=0; i<N; i++){
        int temp;
        cin >> temp;
        if(res[temp] > 0){//该数已经验证过，随后的数也都验证过
            continue;
        }
        res[temp]++;//该数暂定为关键字
        while(temp != 1){
            if(temp%2 == 1){
                temp = 3*temp+1;
            }
            temp /= 2;
            if(res[temp] > 0){//该数已经验证过，随后的数也都验证过
                res[temp] += 2;
                break;
            }
            res[temp] += 2;//该数不是关键字
        }
    }
    bool a = true;
    for(int i=res.size()-1; i>1; i--){
        if(res[i] == 1){
            if(!a)
                cout << " ";
            cout << i;
            a = false;
        }
    }
    return 0;
}