浙大PAT乙级1005 “继续(3n+1)猜想 ”python解答部分正确_求教如何修改
卡拉兹(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中提交有问题,测试点无法通过,求大佬解答,如何修改,谢谢!
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次随机测试, 如果两个函数结果不同则提示. 在我这边暂时没出现问题, 麻烦你再提交下.
n = int(input())
nums = list(map(int, input().split(' ')))
calcs = []
keys = [n]
def collatz_conjecture(number):
while number != 1:
if number % 2 == 0:
# 偶数
number /= 2
elif number % 2 == 1:
# 奇数
number = number * 3 + 1
if number not in calcs:
calcs.append(int(number))
collatz_conjecture(n)
for num in nums:
if num in calcs:
continue
collatz_conjecture(num)
if num not in keys:
keys.append(num)
keys.sort(reverse=True)
print(" ".join([str(item) for item in keys]))