LeetCode 118 杨辉三角

给定一个非负整数 numRows，生成杨辉三角的前 numRows 行。

在杨辉三角中，每个数是它左上方和右上方的数的和。

示例:

输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]

这个问题可以使用动态规划的方法来解决

• 声明一个数组dp
• 将数组的边界全部初始化为1，其他元素全部设置为0
• 为0的元素通过状态转移方程dp[i][j]=dp[i-1][j-1]+dp[i-1][j]求出,即可得到杨辉三角

C++

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        //dp[i][j]=dp[i-1][j-1]+dp[i-1][j]
        vector<vector<int> >res;
        for(int i=1;i<=numRows;i++)
        {
            vector<int>temp(i);
            res.push_back(temp);
        }
        for(int i=0;i<res.size();i++)
        {
            res[i][0]=res[i][res[i].size()-1]=1;
        }
        for(int i=0;i<res.size();i++)
        {
            for(int j=0;j<i+1;j++)
            {
                if(res[i][j]==0)
                {
                    res[i][j]=res[i-1][j-1]+res[i-1][j];
                }
            }
        }
        return res;
    }
};

Python

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        dp=[[0]*n for n in range(1,numRows+1)]
        for i in range(0,numRows):
            dp[i][0]=dp[i][-1]=1
        for i in range(0,numRows):
            for j in range(i+1):
                if dp[i][j]==0:
                    dp[i][j]=dp[i-1][j-1]+dp[i-1][j]
        return dp