一、题目
函数原型:int** generate(int numRows, int* returnSize, int** returnColumnSizes)
参数解析:numRows是指明要求前几行杨辉三角
returnSize是返回指针数组的元素个数
returnColumnSizes是指明杨辉三角每一行有几个元素
二、思路既然需要函数返回值类型是int**,那么需要返回一个指针数组。先创建一个指针数组ret,并为其申请空间。returnSize是指针数组元素个数,那么也就等于numRows。每一个指针数组中的一个元素又是一个数组,returnColumnSizes就表示该数组的大小。
经过观察杨辉三角,发现每一行除去第一个和最后一个元素,其余元素值都等于上一行邻近的两个数之和。因此设置头指针与尾指针,头指针初始指向每一行第二个元素,尾指针初始指向每一行倒数第二个元素。设置循环,新的一行值用上一行的相邻两个元素相加得到,每一行首尾元素值为1,无需计算。
当杨辉三角行数小于等于3时,每一行的值都可以直接赋值,无需计算。当行数大于等于4时,需要利用循环和首位指针得到每一行的元素值。最后返回指针数组ret。
cpp
int** generate(int numRows, int* returnSize, int** returnColumnSizes)
{
*returnSize = numRows;//返回数组个数等于numRows
int** ret = (int**)malloc(sizeof(int*) * numRows);//返回的数组指针
*returnColumnSizes = malloc(sizeof(int)* numRows);
int left = 0;//头指针
int right = 0;//尾指针
int i = 0;
//int j = 0;
for (i = 0; i < numRows; i++)
{
(*returnColumnSizes)[i] = i + 1;
ret[i] = (int*)malloc(sizeof(int) * (i + 1));
if (i == 0 )
{
ret[i][0] = 1;
}
if (i == 1)
{
ret[i][0] = 1;
ret[i][1] = 1;
}
if (i == 2)
{
ret[i][0] = 1;
ret[i][1] = 2;
ret[i][2] = 1;
left = 1;
right = i - 1;
}
if (i >= 3)
{
ret[i][0] = 1;
ret[i][i] = 1;
left = 1;
right = i - 1;
while (left <= right)
{
ret[i][left] = ret[i - 1][left] + ret[i - 1][left - 1];
ret[i][right] = ret[i - 1][right] + ret[i - 1][right - 1];
left++;
right--;
}
left = 1;
right = i - 1;
}
}
return ret;
}