一. 数组_二维数组及滚动数组_119. 杨辉三角 II

题目描述
给定一个非负索引 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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值