记得之前用 python/JS 实现过,用 java 重写.
一些结论:
- ArrayList.add(index,value) 数组在特定位置添加元素后,之后元素整体后移;
- 每次都是先添加首位元素,之后计算 [1-tmp.size()-2]之间的元素;
- 求全量杨辉三角则 res.add(new ArrayList(tmp));
- 求某行的是直接 return tmp;
求全量杨辉三角 LeetCode 118
public class GenPascalTriangle118 {
public List<List<Integer>> generate(int numRows) {
List<List<Integer>> res = new ArrayList<>();
if(numRows == 0) return res;
//作为临时的数组,存放上次填充后的结果
List<Integer> tmp = new ArrayList<>();
for(int i = 0 ;i < numRows +1 ;i++){
//每次新的数组都是先在首部加1(整体向后移动)
tmp.add(0,1);
for(int j = 1;j < tmp.size() - 1 ;j++){
tmp.set(j,tmp.get(j) + tmp.get(j+1));
}
res.add(new ArrayList<Integer>(tmp));
}
return res;
}
public static void main(String[] args){
GenPascalTriangle118 p = new GenPascalTriangle118();
p.generate(2);
}
}
求某行杨辉三角 LeetCode 119 O(k) 空间复杂度
public class PascalTriangle {
public List<Integer> getRow(int rowIndex) {
List<Integer> l = new ArrayList<>();
if(rowIndex < 0) return l;
int i = 0;
while (i < rowIndex+1){
//改变了目前元素在该位置(如果有的话)和所有后续元素向右移动(将添加一个到其索引)
l.add(0,1);
/* 1 - > 1 1
-> 1 1 1 先在 l 的首位加 1,之后调整后面的值,这里不存在未new即set的情况
* */
for(int j = 1; j < l.size() - 1; j++){
//如何保证最后一个元素是1?
l.set(j,l.get(j)+l.get(j+1));
}
i++;
}
return l;
}
public static void main(String[] args){
PascalTriangle p = new PascalTriangle();
List<Integer> ll = p.getRow(4);
for(int i = 0; i < ll.size(); i++){
System.out.print(ll.get(i) + "->");
}
}
}