杨辉三角

杨辉三角

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

思路和代码

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

杨辉三角2

上面返回的前k行，这一题是返回某一行，同样的代码可以解决。只需要修改一下返回即可。

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<vector<int>> ans(rowIndex+1);
        if(rowIndex == 0)
            return {1};
        for(int i=0; i<rowIndex+1; i++)
        {
            for(int j=0; j<=i; j++)
            {
                if(j==0 || j==i)
                    ans[i].push_back(1);
                else
                {
                    ans[i].push_back(ans[i-1][j-1] + ans[i-1][j]);
                }
            }
        }
        return ans[rowIndex];
    }
};

通过递推可以发现，可以直接求出。
参考链接

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> res(rowIndex + 1);
        res[0] = 1;
        for (int i = 1; i <= rowIndex; ++i) {
            for (int j = i; j >= 1; --j) {
                res[j] += res[j - 1];
            }
        }
        return res;
    }
};