蓝桥杯,杨辉三角形,Python

题目描述

问题描述

杨辉三角形又称Pascal三角形,它的第i+1行是(a+b)i的展开式的系数。
它的一个重要性质是:三角形中的每个数字等于它两肩上的数字相加。

下面给出了杨辉三角形的前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。

本题不难,但是因为踩了一个坑所以花了好长时间,我们先看错误代码:

n = int(input(''))
arr = [[]]*n
for
i in range(n): for j in range(i+1): if j == 0 or i == j: arr[i].append(1) else: arr[i].append(arr[i-1][j-1] + arr[i-1][j]) print(" ".join(str(a) for a in arr[i]))

n=4时,错误代码输出如下:

1
1 1 1
1 1 1 1 2 1
1 1 1 1 2 1 1 2 2 1

再输出arr如下:

[[1, 1, 1, 1, 2, 1, 1, 2, 2, 1], 
[1, 1, 1, 1, 2, 1, 1, 2, 2, 1], 
[1, 1, 1, 1, 2, 1, 1, 2, 2, 1], 
[1, 1, 1, 1, 2, 1, 1, 2, 2, 1]]

可以看到arr的所有元素都相同,那么这是为什么呢?这是因为arr的元素在内存中都指向同一片区域,这是arr的声明方式有问题,arr = [[]]*n将同一片内存区域赋值给了n个指针,我们看以下区别:

>>> arr=[[]]*4
>>> arr[0].append(1)
>>> print(arr)
[[1], [1], [1], [1]]
>>> Arr=[[],[],[],[]]
>>> Arr[0].append(1)
>>> print(Arr)
[[1], [], [], []]

因此本题的正确解法为:

n = int(input(''))
arr = []
for i in range(n):
    arr.append([])
    for j in range(i+1):
        if j == 0 or i == j:
            arr[i].append(1)
        else:
            arr[i].append(arr[i-1][j-1] + arr[i-1][j])
    print(" ".join(str(a) for a in arr[i]))

 

posted @ 2020-02-08 12:57  Marvin-wen  阅读(1391)  评论(0编辑  收藏  举报