[已解决]python杨辉三角问题

阿育王

          1
         / \
        1   1
       / \ / \
      1   2   1
     / \ / \ / \
    1   3   3   1
   / \ / \ / \ / \
  1   4   6   4   1
/ \ / \ / \ / \ / \
1   5   10  10  5   1
把每一行看做一个list，试写一个generator，不断输出下一行的list：

# -*- coding: utf-8 -*-

def triangles():
        此处写代码，没思路！！



# 期待输出:
# [1]
# [1, 1]
# [1, 2, 1]
# [1, 3, 3, 1]
# [1, 4, 6, 4, 1]
# [1, 5, 10, 10, 5, 1]
# [1, 6, 15, 20, 15, 6, 1]
# [1, 7, 21, 35, 35, 21, 7, 1]
# [1, 8, 28, 56, 70, 56, 28, 8, 1]
# [1, 9, 36, 84, 126, 126, 84, 36, 9, 1]
n = 0
results = []
for t in triangles():
    results.append(t)
    n = n + 1
    if n == 10:
        break

for t in results:
    print(t)

if results == [
    [1],
    [1, 1],
    [1, 2, 1],
    [1, 3, 3, 1],
    [1, 4, 6, 4, 1],
    [1, 5, 10, 10, 5, 1],
    [1, 6, 15, 20, 15, 6, 1],
    [1, 7, 21, 35, 35, 21, 7, 1],
    [1, 8, 28, 56, 70, 56, 28, 8, 1],
    [1, 9, 36, 84, 126, 126, 84, 36, 9, 1]
]:
    print('测试通过!')
else:
    print('测试失败!')

最佳答案

月排行榜 / 总排行榜

傻眼貓咪

2022-4-25 08:35:32

def triangles():
        res = []
        while True:
                if len(res) < 2:
                        res.append(1)
                else:
                        temp = [1]
                        for i in range(1, len(res)):
                                temp.append(res[i] + res[i-1])
                        temp.append(1)
                        res = temp
                yield res

for n, i in enumerate(triangles()):
        if n > 10: # 预设想输出多少行，不然无限输出
                break
        print(i)

跳转到最佳答案楼层

傻眼貓咪

def triangles():
        res = []
        while True:
                if len(res) < 2:
                        res.append(1)
                else:
                        temp = [1]
                        for i in range(1, len(res)):
                                temp.append(res[i] + res[i-1])
                        temp.append(1)
                        res = temp
                yield res

for n, i in enumerate(triangles()):
        if n > 10: # 预设想输出多少行，不然无限输出
                break
        print(i)

Stubborn

思路？给定你任意一个状态 Ni , 假如Ni为 [1, 8, 28, 56, 70, 56, 28, 8, 1],，你将如何生成下一个状态Ni+1 ,并确保所有状态通用？
Ni =  [1, 8,  28, 56, 70, 56, 28, 8, 1, 0]
r =    [0, 1,  8,  28, 56, 70, 56, 28, 8, 1]
R =   [1, 9, 36,  84, 126, 126, 84, 36, 9, 1]
或者
res = [a+b for a,b in zip(Ni, Ni[1:])]
res.append(1)
res.insert(0, 1)


另外你应该知道如何编写generator吧？

wp231957

Stubborn 发表于 2022-4-25 09:18
思路？给定你任意一个状态 Ni , 假如Ni为 [1, 8, 28, 56, 70, 56, 28, 8, 1],，你将如何生成下一个状态Ni+1 ...

麻烦看看这个代码是否还有优化的地方：
网上看到的最NB的代码 是 使用推导式   一个中括号搞定杨辉三角
我这个代码 咋糅合也 变不成一句代码

def f(a):
    n=[1]
    yield n
    for x in range(a):
        n.append(0)
        n.insert(0,0)
        n=[value+n[idx+1] for idx,value in enumerate(n) if idx<len(n)-1]
        yield n
 
a=f(10)
for x in a:
    print(x)

'''
[1]
[1, 1]
[1, 2, 1]
[1, 3, 3, 1]
[1, 4, 6, 4, 1]
[1, 5, 10, 10, 5, 1]
[1, 6, 15, 20, 15, 6, 1]
[1, 7, 21, 35, 35, 21, 7, 1]
[1, 8, 28, 56, 70, 56, 28, 8, 1]
[1, 9, 36, 84, 126, 126, 84, 36, 9, 1]
[1, 10, 45, 120, 210, 252, 210, 120, 45, 10, 1]
'''    

Stubborn

wp231957 发表于 2022-4-25 10:07
麻烦看看这个代码是否还有优化的地方：
网上看到的最NB的代码 是 使用推导式   一个中括号搞定杨辉三角，你看到的什么样的
...

一般都是根据上一个状态，推导下一个状态的呀

def gen(n):
    yield [1,]
    yield [1, 1]
    r = [1, 1]
    while n:
        r = [1] + [a+b for a,b in zip(r, r[1:])] + [1]
        yield r
        n -= 1

qq1151985918

wp231957 发表于 2022-4-25 10:07
麻烦看看这个代码是否还有优化的地方：
网上看到的最NB的代码 是 使用推导式   一个中括号搞定杨辉三角
...

一行没能实现，两行OK不

f = lambda a,b:1 if a==b or b==0 else f(a-1,b)+f(a-1,b-1)
print([[f(x,y) for y in range(x+1)] for x in range(10)])

傻眼貓咪

太厉害了，大佬们的代码让我大开眼界。

wp231957

qq1151985918 发表于 2022-4-25 12:37
一行没能实现，两行OK不

又去查看了当初浏览的网页，有些记忆错误
代码虽然都很短，但是常规算法  都是先写了一个函数，然后才是一行输出，这和你的没啥区别

真正的一行输出   是引用了三方库