PAT 1005 繼續(3n+1)猜想 python程式碼滿分

wyx翔發表於2020-12-08

1005 繼續(3n+1)猜想

卡拉茲(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

程式碼如下:

# 遞迴函式
def step(n,cover):
    if n==1:
        pass
    else:
        if n%2==0:
            n = n//2
        else:
            n = (3*n+1)//2
        cover.append(n)
        step(n,cover)

#輸入
k = eval(input())
num = input().split(' ')
num = [int(i) for i in num]

out = []
total = []
for i in range(len(num)):
    cover = []
    step(num[i],cover)
    total.append(cover)
for i in range(k):
    a = num[i]
    b = 0
    for j in total:
        if a not in j:
            b+=1
    if b == k:
        out.append(a)
out.sort(reverse=True)#降序
#輸出
for i in out:
    if i ==out[len(out)-1]:
        print(i)
    else:
        print(i,end=' ')

得分

在這裡插入圖片描述

相關文章