主页 / 问题 / 1226646
Accepted
марат
Asked:2022-01-03 15:52:33 +0800 CST2022-01-03 15:52:33 +0800 CST

创建一个循环，将矩阵的前一个元素中的两个相邻数字相加

  • 772

由于一些转换，我正在编写一个获取下一系列数字的代码

1
1 1
1 2 1
1 3 3 1
1 4 6 4 1

行“1 4 6 4 1”是从上一行“1 3 3 1”得到的，如下：开始时总是1, 1+3=4, 3+3=6, 3+1=4,最后它总是1

这个想法是矩阵的前两个元素已经存在，循环从第一个元素开始，计算那里两个相邻元素的总和并将其写入第二个元素（我从零开始在矩阵中编号），我试图以某种方式实现它，但结果不是很好

table = [[1], [1, 1]]
n = int(input())
s = 0
for i in range(n - 2):
  for row in table[i]:
       s += sum(row)
  table.append([s])  
print(table)

在代码中修复什么以使其工作？

python

2 个回答

  1. pascal_triangle是一个无限帕斯卡三角行生成器。下一行是通过对前一行的相邻元素求和获得的。为此，请在上一行的左侧和右侧添加零：

    1 3 3 1 ->    0 1 3 3 1
              1 3 3 1 0
            + ---------
              1 4 6 4 1

    itertools.islice- 修剪无限生成器的工具：

    import itertools

def pascal_triangle():
    line = (1, )
    while True:
        yield line
        line = tuple(map(sum, zip((0, ) + line, line + (0, ))))

for line in itertools.islice(pascal_triangle(), 5):
    print(*line)

    1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
    • 6
  2. Best Answer
    def solve(n: int) -> list:
    retval = [(1,)]  # Начальное состояние
    for _ in range(n - 1):  # Делаем n-1 раз, т.к. первая строка уже задана
        last_row = retval[-1]  # Забираем последнюю строку
        next_row = map(sum, zip(last_row, last_row[1:]))  # Суммируем все пересекающиеся пары из последней строки
        wrapped_next_row = (1, *next_row, 1)  # Добавляем единички с концов
        retval.append(wrapped_next_row)  # Добавляем в итоговое значение очередную строку
    return retval

print(*solve(10), sep='\n')
    • 2

相关问题