题目描述
给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
提示:
0 <= rowIndex <= 33
进阶:
你可以优化你的算法到 O(rowIndex) 空间复杂度吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析及函数部分代码
这道题最开始的思路与杨辉三角解法相同,唯一需要注意的地方就是题目给定的索引再这道题里面是需要加一的。
class Solution {
public List<Integer> getRow(int rowIndex) {
List<Integer> List=new ArrayList<>();
int[][] nums=new int[rowIndex+1][rowIndex+1];
for(int i=0;i<=rowIndex;i++){
for(int j=0;j<=i;j++){
if(j==0||i==j){
nums[i][j]=1;
}else{
nums[i][j]=nums[i-1][j]+nums[i-1][j-1];
}
if(i==rowIndex){
List.add(nums[i][j]);
}
}
}
return List;
}
}
但是,不知道大家有没有注意到,这道题目的题目描述进阶要求是算法到 O(rowIndex) 空间复杂度,这种解法显然是不符合题目要求的,所以,还是要改进!!!
其实,杨辉三角各个位置上的数是有公式可以算出来的,大家可以参考一下官方解法
class Solution {
public List<Integer> getRow(int rowIndex) {
List<Integer> row = new ArrayList<Integer>();
row.add(1);
for (int i = 1; i <= rowIndex; ++i) {
row.add((int) ((long) row.get(i - 1) * (rowIndex - i + 1) / i));
}
return row;
}
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/pascals-triangle-ii/solution/yang-hui-san-jiao-ii-by-leetcode-solutio-shuk/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。