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