一个清明节过的我都已经凌乱了==之前的学习状态也需要好好调整回来了。假期之前把HashTable的题目刷完了,现在想把以前刷过的Array的题目再刷一遍,总结一下。恩
题目:Given numRows, generate the first numRows of Pascal’s triangle.
For example, given numRows = 5, Return
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
其实就是杨辉三角。118是获得杨辉三角的前n行,而119是获得杨辉三角的第n行。下面我们来看一下具体思路:
要获得前n行,我们知道杨辉三角的第n行中每个元素是n-1行中两个元素求和得到的。所以只需要使用List来保存上一行的结果然后遍历求和即可。思路很简单,代码入下:
public List<List<Integer>> generate(int numRows) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
List<Integer> row, pre = null;
for (int i = 0; i < numRows; ++i) {
row = new ArrayList<Integer>();
for (int j = 0; j <= i; ++j)
if (j == 0 || j == i)
row.add(1);
else
row.add(pre.get(j - 1) + pre.get(j));
pre = row;
res.add(row);
}
return res;
}
对于119题,我们只需要获得第n行,一种简单单的方法就是采用上面的思路,只返回第n行。这里我们可以将代码简化到只使用一个List,然后遍历自身==代码入下:
public List<Integer> getRow1(int rowTndex){
List<Integer> res = new ArrayList<>();
for(int i=0; i<rowTndex+1; i++){
res.add(0, 1);
for(int j=1; j<res.size(); j++)
res.set(j, res.get(j) + res.get(j+1));
}
return res;
}
当然,另外一种更加简单的有效的方法就是使用数学推倒的方式获得第n行中每个元素的公式。可以观察一下其规律,就可以得到下面的代码:
public List<Integer> getRow(int rowIndex) {
List<Integer> res = new ArrayList<>();
res.add(1);
int k = rowIndex;
for( int i = 1; i <= rowIndex; i++, k-- ){
// Using double to remove the overflow
double tmp = ((double)res.get(i-1)*k)/i;
res.add( (int)tmp );
}
return res;
}