PAT测试，1005题：继续(3n+1)猜想

Question

不知道哪里有错，总是最后一个测试不通过：
请教大家，到底哪里考虑不好？

题目如下：

卡拉兹(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 <stdio.h>

int main(int argc, char **agrv) {
    int n;
    int SIZE = 200;
    int keys[SIZE];
    int num[100];
    scanf("%d", &n);
    for (int i = 0; i < SIZE; ++i) {
        keys[i] = -1;
    }
    for (int j = 0; j < n; ++j) {
        scanf("%d", &num[j]);
    }

    /*
     * -1：  未被处理
     * 0：   计算过
     * 1：   关键数
     */
    for (int k = 0; k < n; ++k) {
        int data = num[k];
        if (keys[data] == -1) {
            // 标记为"关键数"
            keys[data] = 1;
        } else {    // 已被计算则跳过
            continue;
        }
        if (data % 2 == 0) {
            data /= 2;
        } else {
            data = (3 * data + 1) / 2;
        }
        while (data != 1 && keys[data] != 0) {
            keys[data] = 0;
            if (data % 2 == 0) {
                data /= 2;
            } else {
                data = (3 * data + 1) / 2;
            }
        }
    }
    int fist = 1;
    for (int l = SIZE - 1; l >= 0; l--) {
        if (keys[l] == 1) {
            if (!fist) {
                printf(" %d", l);
            } else {
                printf("%d", l);
                fist = 0;
            }
        }
    }
    return 0；
}

另外又用Python写了一次，对于想到的参数都是正确的，但是系统判定还是有一部分错：

DONE = 0  # 已被计算
UNI = 1  # 关键数

n = int(input())
nums = [int(x) for x in input().split(' ')]

uni = {}


def cut(number):
    if number % 2 == 0:
        number //= 2
    else:
        number = (3 * number + 1) // 2
    return number


for data in nums:
    if data not in uni:
        uni[data] = UNI  # 第一次遇到这个数，标记为关键数
    else:
        uni[data] = DONE  # 第二次遇到这个数，标记为已被计算
        continue
    data = cut(data)
    while data != 1:
        uni[data] = DONE
        data = cut(data)

out = sorted([str(x) for x in uni if uni[x] == UNI], key=lambda x: x, reverse=True)
print(' '.join(out))