题目描述
给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii
这里创建杨辉三角的方法不变,可以参照118的方法
1.这里每行的第一个元素和最后的一个元素都是1
2.其他位置的元素按照如下公式
ret[i][j] = ret[i - 1][j - 1] + ret[i - 1][j];
3.只不过这里需要创建rowIndex+1行的数组大小,因为这里返回ret[rowIndex]行
vector<int> getRow(int rowIndex) {
vector<vector<int>> ret(rowIndex + 1); //这里初始化结果矩阵为numRows行
for(int i = 0; i < rowIndex + 1; i++) {
ret[i].resize(i + 1); //每一行有多少个元素
for(int j = 0; j < i + 1; j++) {
if(j == 0 || j == i) { //这里也可以直接针对第0个和最后一个元素直接赋值,其他的元素使用循环遍历,那么这里就不需要判断了。
ret[i][j] = 1;
} else {
ret[i][j] = ret[i - 1][j - 1] + ret[i - 1][j];
}
}
}
return ret[rowIndex];
}
滚动数组法:
1.因为这里当前行之和上一行有关系,所以这里使用滚动数组的方法
2.使用两个数组保存当前行和上一行。
vector<int> getRow(int rowIndex) {
vector<int> pre, cur; //这里定义当前数组和前一个数组
for(int i = 0; i < rowIndex + 1; i++) {
cur.resize(i + 1); //定义当前的列表的长度
cur[0] = cur[i] = 1;
for(int j = 1; j < i ; j++) { //利用上一次的信息
cur[j] = pre[j - 1] + pre[j]; //这里第一次pre为null的时候,不会执行这里
}
pre = cur; //替换掉pre的数组
}
return cur;
}
上面是使用两个数组的空间,下面这里还可以优化为一个数组的空间
当前项只和前面一行的第j-1个和第j个有关。所以计算第几行的时候,可以先计算当前行和上一行的值
这里计算第j行的时候,从后面向前面遍历,这样用到上一行的第j和第j-1的时候,还是上一行的值。
int main() {
int n = 5;
getRow(n);
return 0;
}