Python：實現楊輝三角（藍橋杯）

問題描述

楊輝三角形又稱Pascal三角形，它的第i+1行是(a+b)ⁱ的展開式的系數。

它的一個重要性質是：三角形中的每個數字等於它兩肩上的數字相加。

下面給出了楊輝三角形的前4行：

   1

  1 1

 1 2 1

1 3 3 1

給出n，輸出它的前n行。

輸入格式

輸入包含一個數n。

輸出格式

輸出楊輝三角形的前n行。每一行從這一行的第一個數開始依次輸出，中間使用一個空格分隔。請不要在前面輸出多余的空格。

樣例輸入

4

樣例輸出

1
1 1
1 2 1
1 3 3 1

數據規模與約定

1 <= n <= 34。

def yanghui():
    line = [1]
    while True:
        yield line
        line = [1] + [line[i] + line[i + 1] for i in range(len(line) - 1)] + [1]


def run():
    n = int(input())
    flag = 0
    for i in yanghui():
        print(" ".join(str(j) for j in i))
        flag += 1
        if flag == n:
            break


if __name__ == '__main__':
    run()