整数对列表

如何解决整数对列表

我需要编写一个函数 GrowingSum(n)，其中 n 是一个正整数。这 函数应该返回整数对 (i; j) 的列表，使得

0 < i < n-1,0 < j < n-1

这些对应该以增加的顺序出现 元素的总和。

例如， 对于 n = 4，合法的输出是

[(0; 0); (0; 1); (1; 0); (0; 2); (1; 1); (2; 0); (0; 3); (1; 2); 
 (2; 1); (3; 0); (1; 3); (2; 2); (3; 1); (2; 3); (3; 2); (3; 3)].

有没有人在 python 中有一个算法？

解决方法

类似下面的东西？它并没有真正优化，因为生成没有排序并且排序是在之后完成的，但是除非性能有问题，否则这应该可以正常工作

n = 4
lst = sorted([(i,j) for i in range(n) for j in range(n)],key=lambda x: sum(x))

,

我建议 itertools.product 生成两倍于 0 到 n-1 的值的笛卡尔积，以及 sorted(,key=sum) 以获得正确的顺序

from itertools import product

def fct(n: int):
    return sorted(product(range(n),range(n)),key=sum)