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] 。