【PAT】python 1005 继续(3n+1)猜想 (25 point(s))

最开始用列表超时了,然后想着用集合,最后改成了哈希表。用s存储所有过程中验证过的数,最后判断所给的数是否在s里,如果不在则符合要求。

代码如下:

n=int(input())
num=list(map(int,input().split()))
s={}
for i in num:
    while(i!=1):
        if(i%2==1):
            i=3*i+1
        i=i/2
        if i in s:
            break
        s[i]=1
num=sorted(num,reverse=True)
l=len(num)
flag=0
for i in range(l):  
    if(num[i] not in s):
        if flag==1:
            print("",num[i],end='')
        if flag==0:
            print(num[i],end='')
        flag=1
        
        

你可能感兴趣的:(PAT,哈希表,Collatz猜想,算法优化,数字处理,编程)