一、题目
函数原型: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; }