LeetCode——杨辉三角 i-ii
Q:给出一个值numRows,生成杨辉三角的前numRows行
例如,给出 numRows = 5,
返回
[↵ [1],↵ [1,1],↵ [1,2,1],↵ [1,3,3,1],↵ [1,4,6,4,1]↵]
A:
public static ArrayList<ArrayList<Integer>> generate(int numRows) {
ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<>();
if (numRows == 0)
return arrayLists;
ArrayList<Integer> array = new ArrayList<>();
for (int i = 1; i <= numRows; i++) {
ArrayList<Integer> arrayList = new ArrayList<>();
for (int j = 0; j < i; j++) {
if (j == 0 || j == i - 1)
arrayList.add(1);
else
arrayList.add(array.get(j - 1) + array.get(j));
}
array = arrayList;
arrayLists.add(arrayList);
}
return arrayLists;
}
Q:给出一个索引k,返回杨辉三角的第k行
例如,k=3,
返回[1,3,3,1].
备注:
你能将你的算法优化到只使用O(k)的额外空间吗?
A:
注意,行数是从0开始的。使用从后往前计算。
public static ArrayList<Integer> getRow(int rowIndex) {
ArrayList<Integer> array = new ArrayList<>();
for (int i = 0; i <= rowIndex; i++) {
for (int j = i - 1; j > 0; j--) {
array.set(j, array.get(j) + array.get(j - 1));
}
array.add(1);
}
return array;
}