leetcode:118. 杨辉三角

一、题目

leetcode:118. 杨辉三角_第1张图片

函数原型: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;
}

你可能感兴趣的:(leetcode刷题训练营,leetcode,算法,职场和发展)