给定一个非负整数 numRows,生成杨辉三角的前numRows行。
杨辉三角示意图在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
思路:每个数都是上一个list的currentIndex-1和currentIndex的和。如果currentIndex-1或者currentIndex在上一个list不存在就置为0。
public class PascalTriangle {
public void test() {
List<List<Integer>> list = generate(11);
LogUtil.Companion.d("list->" + list.toString());
}
public List<List<Integer>> generate(int numRows) {
List<List<Integer>> result = new ArrayList<>();
if (numRows == 0) {
return result;
}
int num = 0;
for (int i = 0; i < numRows; i++) {
num++;
List<Integer> list = new ArrayList<>();
if (i == 0) {
list.add(1);
result.add(list);
continue;
} else if (i == 1) {
list.add(1);
list.add(1);
result.add(list);
continue;
}
List<Integer> lastList = result.get(result.size() - 1);
for (int j = 0; j < num; j++) {
int pre = (j - 1 >= 0) ? lastList.get(j - 1) : 0;
int current = (j < lastList.size()) ? lastList.get(j) : 0;
list.add(pre + current);
}
result.add(list);
LogUtil.Companion.d("i->" + i + " lastList->" + lastList.toString());
}
return result;
}
}
获取杨辉三角某一行~
public List<Integer> getRow(int rowIndex) {
List<Integer> list = new ArrayList<>();
int num = 0;
List<Integer> preList = new ArrayList<>();
for (int i = 0; i < rowIndex + 1; i++) {
num++;
list.clear();
if (i == 0) {
list.add(1);
preList = new ArrayList<>(list);
LogUtil.Companion.d(" 0 lastList->" + preList.toString());
continue;
} else if (i == 1) {
list.add(1);
list.add(1);
preList = new ArrayList<>(list);
LogUtil.Companion.d(" 1 lastList->" + preList.toString());
continue;
}
for (int j = 0; j < num; j++) {
int pre = (j - 1 >= 0) ? preList.get(j - 1) : 0;
int current = (j < preList.size()) ? preList.get(j) : 0;
list.add(pre + current);
LogUtil.Companion.d("pre->" + pre + " current->" + current + " total->" + (pre + current));
}
preList = new ArrayList<>(list);
LogUtil.Companion.d("i->" + i + " lastList->" + preList.toString());
}
return list;
}
网友评论