问题描述
我需要编写一个函数 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)