LeetCode每日一题-2.12-119. 杨辉三角 II

题目描述

解法 模拟计算

根据杨辉三角的性质逐行计算
设一个开辟能容纳杨辉三角的二维数组，每一行通过变量i对应，列通过j控制：每行的元素对应dp[i][0] ~ dp[i][j]
容易发现除了边界为1，其他的公式为
dp[i][j] = dp[i - 1][j - 1] + dp[i - 1][j];

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

空间复杂度O(k^2)
空间复杂度O(k^2)

优化

将空间复杂度降为O(k)
挺难想到的：
由于每一行在计算的时候只涉及 （上一行的 （ 对应当前元素下标的） 元素） 以及 （其上一个其元素）
所以只要在一个数组里面修改，就能达到换行的效果
同时为了保证上一个元素在本元素计算之前保留自己的值，应从尾到头反向遍历。
0号元素保持为1

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

时间复杂度O(k^2)
空间复杂度O(k)