问题描述
给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。 在「杨辉三角」中,每个数是它左上方和右上方的数的和。 示例 1: 输入: rowIndex = 3 输出: [1,3,3,1] 示例 2: 输入: rowIndex = 0 输出: [1] 示例 3: 输入: rowIndex = 1 输出: [1,1] 提示: 0 <= rowIndex <= 33
方法1
根据杨辉三角的规律求出前 rowindex 行,然后返回需要的行。
代码
public List<Integer> getRow(int rowIndex) {
List<List<Integer>> lists = new ArrayList<>();
for(int i = 0;i <= rowIndex;i++){
List<Integer> aa = new ArrayList<>();
for(int j = 0;j <= i;j++){
if(j == 0 || j == i){
aa.add(1);
continue;
}
aa.add(lists.get(i - 1).get(j - 1) + lists.get(i - 1).get(j));
}
lists.add(aa);
}
return lists.get(rowIndex);
}
执行结果
方法2
也是根据规律求出前 rowindex 行,然后返回需要的行。但是我们只存储杨辉三角的 一半,在返回的时候对数据进行处理然后返回完整的杨辉三角的一行。
注意:
1. 在存储杨辉三角的时候,如何设置每行存储终点下标的设置
2. array[i - 1][j] 不存在
3. 还原 返回List 集合 开始还原的下标的设置
代码
public List<Integer> getRow(int rowIndex) {
List<List<Integer>> lists = new ArrayList<>();
for (int i = 0; i <= rowIndex; i++) {
List<Integer> aa = new ArrayList<>();
for (int j = 0; j <= i / 2; j++) {
if (j == 0) {
aa.add(1);
continue;
}
if (lists.get(i - 1).size() < j + 1) {
aa.add(lists.get(i - 1).get(j - 1) * 2);
} else {
aa.add(lists.get(i - 1).get(j - 1) + lists.get(i - 1).get(j));
}
}
lists.add(aa);
}
if (rowIndex > 0) {
for (int i = (rowIndex + 1) / 2 - 1; i >= 0; i--) {
lists.get(rowIndex).add(lists.get(rowIndex).get(i));
}
return lists.get(rowIndex);
}
return lists.get(0);
}
执行结果