def yang_hui_triangle(floor=1):
    idx, root = 0, [1]
    while idx < floor:
        yield root
        temp = 0
        triangle = []
        for i in root:
            temp += i
            triangle += [temp]
            temp = i
        root = triangle + [root[-1]]
        idx += 1