题目
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行
我的代码
class Solution {
public List<List<Integer>> generate(int numRows) {
List<Integer> firstline = new ArrayList();
firstline.add(1);
List <Integer> secondline = new ArrayList();
secondline.add(1);
secondline.add(1);
List<List<Integer>> ans = new ArrayList();
if (numRows == 0) {return ans;}
ans.add(firstline);
if (numRows == 1) {return ans;}
ans.add(secondline);
if (numRows == 2) {return ans;}
List<Integer> lastturn = secondline;
for (int i=2; i<numRows; i++){
List<Integer> thisturn = new ArrayList();
thisturn.add(1);
for (int j=1; j<lastturn.size(); j++){
thisturn.add(lastturn.get(j-1) + lastturn.get(j));
}
thisturn.add(1);
ans.add(thisturn);
lastturn = thisturn;
}
return ans;
}
}
小结
两个问题:
1.List的定义:
本来我想当然的以为可以像例题解答那样,list = [[1],[1,1]]来直接声明。报错后意识到这只是接口,必须用add实现(?我也不懂是不是只能一个一个add进去)
2.没有考虑到numRows==0的情况,导致第一次提交解答错误。