1.题目
给定一个非负索引 rowIndex
,返回「杨辉三角」的第 rowIndex
行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3 输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0 输出: [1]
示例 3:
输入: rowIndex = 1 输出: [1,1]
2. 解题思路
此题与118.杨辉三角类似,先构建一个杨辉三角,然后返回第
rowIndex行数组即可。
- 首先,通过指针传递的方式设置返回的数组大小为
rowIndex+1
,这样可以在函数外部知道返回数组的大小。- 动态分配一个二维数组
ret
,其大小为(rowIndex+1)
行,每行的大小从1到rowIndex+1
不等。- 初始化第一列和最后一列的值,都是1。这是杨辉三角的一个特性,每一行的开始和结束都是1。
- 使用嵌套循环从第二列开始计算每一列的值,每一列的值是上一行相邻两个数的和。
- 最后,返回第
rowIndex
行的数组,注意这里只返回了一维数组,而不是整个二维数组。这是因为杨辉三角的每一行都是对称的,所以只需要返回一行即可。
int* getRow(int rowIndex, int* returnSize) {
// 首先,通过指针传递的方式设置返回的数组大小为rowIndex+1
*returnSize = rowIndex + 1;
// 动态分配一个二维数组ret,其大小为(rowIndex+1)行,每行的大小从1到rowIndex+1不等
int** ret = malloc(sizeof(int*) * (rowIndex + 1));
for (int i = 0; i <= rowIndex; ++i) {
// 为每一行分配空间,大小为i+1,因为每行至少有1个元素,且最后一行最多有rowIndex+1个元素
ret[i] = malloc(sizeof(int) * (i + 1));
}
// 初始化第一列和最后一列的值,都是1
for (int i = 0; i <= rowIndex; ++i) {
ret[i][0] = ret[i][i] = 1;
}
// 从第二列开始,每一列的值是上一行相邻两个数的和
for (int i = 1; i < rowIndex; ++i) {
for (int j = 1; j < i; ++j) {
ret[i][j] = ret[i - 1][j - 1] + ret[i - 1][j];
}
}
// 返回第rowIndex行的数组,注意这里只返回了一维数组,而不是整个二维数组
return ret[rowIndex];
}