leetcode 杨辉三角 II

pascals-triangle-ii
方法 1

func getRow1(rowIndex int) []int {
	rowIndex += 1

	var res [][]int
	for i := 0; i < rowIndex; i++ {
		r := make([]int, 0)
		for j := 0; j < i+1; j++ {
			if j == 0 || j == i {
				r = append(r, 1)
			} else {
				r = append(r, res[i-1][j]+res[i-1][j-1])
			}
		}
		res = append(res, r)
	}
	return res[len(res)-1]
}

方法 1 说白了就是在杨辉三角 1 的解决方法上返回最后的元素。和杨辉三角 1 相比没有任何优化。这里需要注意的是杨辉三角 2 起始行是从第 0 行开始的。推荐使用方法 2 简洁方便。

---

方法 2

func getRow2(rowIndex int) []int {
	rowIndex+=1
	res := make([]int, rowIndex)
	for i := 0; i < rowIndex; i++ {
		res[i] = 1
		for j := i - 1; j > 0; j-- {
			res[j] += res[j-1]
		}
	}
	return res
}

需要注意内层循环
假设 rowIndex = 5

第一次循环：i = 0，[1 0 0 0 0 0]，内层循环条件不满足。

第二次循环：i = 1，[1 1 0 0 0 0]，内层循环条件不满足。

第三次循环：i = 2，[1 1 1 0 0 0]，内层循环条件满足，第一次循环，res[1] = 2，[1 2 1 0 0 0] 然后退出。

第四次循环：i = 3，[1 2 1 1 0 0]，
内层循环条件满足，第一次循环，res[2] = 3，[1 2 3 1 0 0]，第二次循环 res[1] = 3，[1 3 3 1 0 0] 然后退出。

第五次循环：i = 4，[1 3 3 1 1 0]，
内层循环条件满足，第一次循环，res[3] = 4，[1 3 3 4 1 0]，第二次循环 res[2] = 6，[1 3 6 4 1 0]，第三次循环 res[1] = 4，[1 4 6 4 1 0] 然后退出。

第六次循环：i = 5，[1 4 6 4 1 1]，
内层循环条件满足，第一次循环，res[4] = 5，[1 4 6 4 5 1]，第二次循环 res[3] = 10，[1 4 6 10 5 1]，第三次循环 res[2] = 10，[1 4 10 10 5 1]，第四次循环 res[1] = 5，[1 5 10 10 5 1]

最终返回数组 [1 5 10 10 5 1] 。