力扣：数组（四）

二维数组及滚动数组

力扣题目编号，二维数组及滚动数组：118、119、661、598、419

118：杨辉三角

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

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

示例:

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

解题思路

    定义一个数组元素为数组的数组，循环两次该数组，对每个数组的头和尾赋值为1，其他部分为上一个数组的ans[i-1][j-1] + ans[i-1][j]，即可。

实现代码

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        vector<vector<int>> ans(numRows);
        if(numRows == 0) {
            return ans; 
        }   
        for(int i = 0; i < numRows; 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;
    }
};

119：杨辉三角 II

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

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

示例:

输入: 3
输出: [1,3,3,1]
进阶：

你可以优化你的算法到 O(k) 空间复杂度吗？

解题思路

    刚开始思路进入了死循环，看错题目，以为是需要时间复杂度为O(n)，想了好久，还是觉得无法实现。看了大佬的代码才知道，题目与118题无异，就是只需要返回单个数组。
    利用杨辉三角前一行与后一行的关系，从最后面的元素进行更新，如果更新最前面的元素，会导致后面元素累加时，原数据已经发生变化。

实现代码

class Solution {
public:
    vector<int> getRow(int rowIndex) {
    	//第K行的vector大小为 rowIndex+1
        vector<int> kRows(rowIndex+1);
        //利用前一行求后一行，第K行要循环K遍
        for(int i = 0; i <= rowIndex; i++)
            {
            kRows[i] = 1;//行末尾为1
            for(int j = i; j > 1; j--)//每一行的更新过程
                {
                    kRows[j-1] = kRows[j-2] + kRows[j-1];
                }
            }
        return kRows; 
    }
};