gpt4 book ai didi

python - 我需要以特定方式安排我的列表 - python

转载 作者:行者123 更新时间:2023-11-28 20:31:50 26 4
gpt4 key购买 nike

基本上,用户输入任何正数,然后程序应安排一个列表,其中包含直到输入的数字为止的所有正数,以便列表中 2 个连续元素的总和是一个平方数。如果输入的数字不允许排序,我只想让程序打印错误。这是到目前为止的代码:

u = int(input("ENTER: "))
l = []
for i in range(1, u + 1):
l.append(i)
o = l
t = []
for elem in l:
for x in o:
p = elem + x
p = math.sqrt(p)
if p%1 == 0:
if x == elem:
break
else:
t.append(x)
t.append(elem)

例如,如果我输入 15,那么最后的列表 t 看起来像这样:

[3, 1, 8, 1, 15, 1, 1, 3, 6, 3, 13, 3, 5, 4, 12, 4, 4, 5, 11, 5, 3, 6, 10, 6, 2, 7, 9, 7, 1, 8, 7, 9, 6, 10, 15, 10, 5, 11, 14, 11, 4, 12, 13, 12, 3, 13, 12, 13, 2, 14, 11, 14, 1, 15, 10, 15]

该列表包含理论上可行的所有对,我很难安排此列表,以便每个数字都出现一次,并且每个连续的数字都遵循上述属性。

所以我最终要寻找的列表是这样的:

[8, 1, 15, 10, 6, 3, 13, 12, 4, 5, 11, 14, 2, 7, 9]

提前感谢您提供的任何帮助。

最佳答案

这将是一种带有递归函数的蛮力方法:

import math

def f(temp, numbers):
for i, j in zip(temp[:-1], temp[1:]):
sqrt = math.sqrt(i+j)
if int(sqrt) != sqrt:
return False
if not numbers:
return temp
for i in numbers:
result = f(temp + [i], [j for j in numbers if j != i])
if result:
break
return result

n = int(input("Arrange numbers from 1 to ").strip())
numbers = list(range(1, n+1))
print("Input:", numbers)
print("Output:", f([], numbers))

示例 4:

Arrange numbers from 1 to 4
Input: [1, 2, 3, 4]
Output: False

15 的例子:

Arrange numbers from 1 to 15
Input: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
Output: [8, 1, 15, 10, 6, 3, 13, 12, 4, 5, 11, 14, 2, 7, 9]

关于python - 我需要以特定方式安排我的列表 - python,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53691261/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com