LeetCode_119

题目:

给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 行。

题目要求O(K)空间复杂度,即只开一个大小为k的数组。

解析: 

题目要求只能开一个数组,则只能利用滚动的思想,每一行与上一行有关。在杨辉三角中,每个数是它左上方和右上方的数的和:

将三角转化为数组的话(空位补0):

1 0 0

1 1 0

1 2 1

每个数就是它上一行正上方和左上方的数的和。设置数组的初始值1 1,遍历数组将之后的每一行的值求出,填到数组中,再在末尾补个1。但是有个问题就是不可以从前往后遍历数组,因为这样的话会出现上一行的值被覆盖的情况,例如:

第一行:1 1

第二行:1 2 1

第三行:1 3 4 1

前两行没有问题,但到了第三行时,求出第二个数为1+2=3,将2覆盖了,第三个数就成了1+3=4,,出现了错误,后面都会发生同样的错误。所以应该从后往前遍历,先在末尾补个1,不会影响前边的值。代码如下:

public List<Integer> getRow(int rowIndex) {
		List<Integer> result = new ArrayList<Integer>(rowIndex + 1);
		result.add(1);
		if (rowIndex == 0)
			return result;
		result.add(1);
		if (rowIndex == 1)
			return result;
		for (int i = 2; i <= rowIndex; i++) {
			result.add(1);
			for (int j = i - 1; j > 0; j--)
				result.set(j, result.get(j) + result.get(j - 1));
		}
		return result;
	}

猜你喜欢

转载自blog.csdn.net/Q_M_X_D_D_/article/details/81428950
119