1. 程式人生 > >15.楊輝三角II

15.楊輝三角II

給定一個非負索引 k,其中 k ≤ 33,返回楊輝三角的第 行。

在楊輝三角中,每個數是它左上方和右上方的數的和。

示例:

輸入: 3
輸出: [1,3,3,1]

程式碼:

public static List<Integer> getRow(int rowIndex) {
	List<List<Integer>> list = new ArrayList<List<Integer>>();
	if (rowIndex < 0) {
		return new ArrayList<Integer>();
	} else if (rowIndex == 0) {
		List<Integer> li = new ArrayList<Integer>();
		li.add(1);
		return li;
	}

	for (int i = 1; i <= rowIndex + 1; i++) {
		List li = new ArrayList();
		if (i == 1) {
			li.add(1);
			list.add(li);
			continue;
		}
		for (int j = 1; j <= i; j++) {
			if (j == 1) {
				li.add(1);
			} else {
				List<Integer> last = list.get(i - 2);
				int a = last.get(j - 2);
				int b = last.size() >= j ? last.get(j - 1) : 0;
				li.add(a + b);
			}
		}
		list.add(li);
	}
	return list.get(list.size() - 1);
}