浙大PAT乙级1005 “继续(3n+1)猜想 ”python解答部分正确_求教如何修改

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

from copy import deepcopy
n = int(input())#读入第一行的输入
arr = input()#读入第二行的输入
num = [n for _ in range(n)]
start = 0
k = 0
def Callatz(n):
    if n%2==0:#偶数
        n = n/2
    else:#奇数
        n = (3*n+1)/2
    return n

for i in list(range(len(arr))):#将第二行读入的字符串存入数组
    if arr[i]==' ' and i!=len(arr)-1:
        num[k] = int(str(arr[start:i]))
        start = i+1
        k += 1
    elif i==len(arr)-1:
        num[k] = int(str(arr[start:len(arr)]))
    else:
        pass
num.sort(reverse = True)#先从大到小排序
keywords = [1 for _ in range(len(num))]#用以标记是否是关键字的数组
for i in list(range(len(num))):
    a = num[i]
    while a!=1:
        a = Callatz(a)
        for j in list(range(len(num))):
            if i==j:
                pass
            else:
                if a==num[j]:
                    keywords[j]=0
                else:
                    pass
index = [0 for _ in range(len(num))]
for i in list(range(len(keywords))):#去除非关键字
    if keywords[i]!=0:
        index[i] = i;
    else:
        pass
index2 = deepcopy(index)
for i in index:
    if i==0:
        index2.remove(i)
for i in list(range(len(index2))):
    if i==len(index2)-1:
        print(num[index2[i]])
    else:
        print(num[index2[i]],end = ' ')
 

运行结果 

目前用题中给的例子测试输出结果和标准答案一致，但PAT中提交有问题，测试点无法通过，求大佬解答，如何修改，谢谢！

Answer 1

def f(ns):
    a = {}
    for x in ns:
        while True:
            if i := a.get(x):
                if i > 1:
                    break
                else:
                    a[x] += 1
            else:
                a[x] = 1
            if x == 1:
                break
            x = (3*x + 1) >> 1 if x & 1 else x >> 1
     
    res = [x for x in a.keys() if a[x] == 1 and x in ns]
    return sorted(res, reverse=True)

# test

def g(ns):
    seqs = {}
    for x in ns:
        t = x
        seqs[t] = set()
        while True:
            seqs[t].add(x)
            if x == 1:
                break
            x = (3*x + 1) >> 1 if x & 1 else x >> 1
            
    res = []
    for k1, v1 in seqs.items():
        flag = True
        for k2, v2 in seqs.items():
            if k1 == k2:
                continue
            if k1 in v2:
                flag = False
                break
        if flag:
            res.append(k1)
                
    return sorted(res, reverse=True)


from random import randint
for j in range(1000):
    ns = set([randint(1, 100) for _ in range(randint(1, 50))])
    a = f(ns)
    b = g(ns)
    if a != b:
        print('wrong')

昨天理解错了, 我又修改了一下. 主要想法是用字典记录每个路径中每个数字的出现次数, 如果有重复路径, 则次数必大于2, 在代码中, 如果大于2则截断. 最终留下那些次数为1的就是没有被覆盖的. f函数是实现, g函数是暴力实现. 下面是1k次随机测试, 如果两个函数结果不同则提示. 在我这边暂时没出现问题, 麻烦你再提交下.