leetcode 算法-杨辉三角
题目
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
解题思路
杨辉三角大家都很熟悉,如上图所示,杨辉三角中除了第一个和最后一个元素为1外,每一个元素等于上一行左右肩上的元素之和。
知道这一点之后,我们就可以根据上一行来获得下一行的元素。然后除第一行每一行和第二行外其他行都可以通过上一行来获得。
当numRows=0,1时,作为递归的出口,直接返回指定结果;
当numRows>=2时,则先递归调用获取numRows-1的结果,然后遍历此结果,根据生成规则产生下一行,之后将numRows-1的结果和下一行的结果合并,就是最终的杨辉三角。
代码
class Solution {
public List<List<Integer>> generate(int numRows) {
List<List<Integer>> result = null;
List<Integer> list = null;
if(numRows<2){
result = new ArrayList<>();
if (numRows == 0) return result;
list = new ArrayList<>();
list.add(1);
result.add(list);
return result;
}else {
result = generate(numRows - 1);
list = result.get(result.size()-1);
List<Integer> list3 = new ArrayList<Integer>(list.size()+1);
list3.add(1);
for (int i = 1; i < list.size(); i++) {
list3.add(list.get(i-1)+list.get(i));
}
list3.add(1);
result.add(list3);
return result;
}
}
}