Распечатка списка (содержащего целые числа) в лексикографическом порядке
Задача: Вам даны три целых числа x,y и z вместе с целым числом n. Вы должны напечатать список всех возможных координат, где сумма не равна n. Печать Распечатать список в лексикографическом порядке. Ниже мой код. Прекрасно работает, за исключением печати в лексикографическом порядке. Ниже приведен мой код. Есть ли лучший подход для получения лексикографического порядка списка, содержащего целые числа?
from itertools import combinations
lst=[]
if __name__ == '__main__':
x = int(input())
y = int(input())
z = int(input())
n = int(input())
lst=[[a,b,c] for b in range(y+1) for c in range(z+1) for a in range(x+1) ]
finallst=[]
for items in combinations(lst,3):
for nums in items:
x=sum(nums)
if x!=n and nums not in finallst:
finallst.append(nums)
f_finallst= (sorted(map(str,(finallst)))) #converted to string to get lexicographic order
print (f_finallst)
My result=['[0, 0, 0]', '[0, 0, 1]', '[0, 1, 0]', '[1, 0, 0]', '[1, 1, 1]']
Expected result=[[0, 0, 0], [0, 0, 1], [0, 1, 0], [1, 0, 0], [1, 1, 1]]
2 ответа
Решение
print([coords
for coords in itertools.product(range(x+1), range(y+1), range(z+1))
if sum(coords) != n])
x = int(input())
y = int(input())
z = int(input())
n = int(input())
lists=[[i,j,k] for i in range(x+1)
for j in range(y+1)
for k in range(z+1) if (i+j+k)!=n]
print(lists)
Используя map(str, finallst)
, вы приведете каждый элемент в списке к str
, Вы хотите сохранить элементы такими, какие они есть, но использовать str
в качестве ключа сортировки.
f_finallst= sorted(finallst, key=str)
Я обнаружил, что добавление лямбда-функции, которая в основном работает как ключ для сравнения сортировки, поможет. Подробное объяснение читайте здесь
if __name__ == '__main__':
x = int(input())
y = int(input())
z = int(input())
n = int(input())
orginal_list = [[i, j, k] for k in range(z+1) for j in range(y+1) for i in range(x+1) if i + j + k != n]
sorted_list = sorted(orginal_list, key = lambda i: (len(i), i))
print(sorted_list)