题目:
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
思路:
观察杨辉三角如下:
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
…
可以看到:
- 每一行的第一列为 1;
- 当 i(行数) = j(列数) 时,对应位置数字为1;
- 当 i > j 时,a[i][j] = a[i-1][j-1]+a[i-1][j]。
这样我们可以用动态规划来解决问题。
设置2个列表row和preList,row来存当前行的数据,preList列表来存前一行的数据。每次row的第一个元素设置为1,最后一个元素设置为1,中间的元素设置为preList(j-1)+preList(j),最后将row放入traingle中即可。
public List<List<Integer>> generate(int numRows) {
List<List<Integer>> traingle = new ArrayList<>();
if(numRows <= 0)return traingle;
//第一行数字为1
traingle.add(new ArrayList());
traingle.get(0).add(1);
//其他行数字
for(int i = 1; i < numRows; i++){
List<Integer> row = new ArrayList<>();
List<Integer> preList = traingle.get(i-1);
//第一列数字为1
row.add(1);
//每一行的第1列开始赋值
for(int j = 1; j < i; j++){
row.add(preList.get(j - 1) + preList.get(j));
}
//i=j时,数字为1
row.add(1);
//最后将row数组加入traaiingle中就好了
traingle.add(row);
}
return traingle;
}
小结
这道题之前做过,现在来做直到思路,但是写出的代码还是不尽人意,以后还是要多练习。