问题描述:
Given an index k, return the kth row of the Pascal’s triangle.
For example, given k = 3,
Return [1,3,3,1].
Note:
Could you optimize your algorithm to use only O(k) extra space?
我的解决思路:
先画出二维数组看规律
0 0
0 1 0
0 1 1 0
0 1 2 1 0
0 1 3 3 1 0
0 1 4 6 4 1 0
.
.
.
比如求当j=2,4 = 1(nums[j-1])+3(nums[j])时,需要把nums[j]用一个变量nums[0](此时nums[0]这个位置不需要用到)保存起来,但是执行后nums[j]=nums[j-1]+nums[j]后nums[j]的值会改变,所以使用一个中间变量temp来保存。可能表述的不是很明确,望见谅.
java代码:
public class Solution {
public List<Integer> getRow(int rowIndex) {
int[] nums = new int[rowIndex+2];
for(int i=0;i<=rowIndex+1;i++)
nums[i] = 0;
nums[1]=1;
for(int i=2;i<=rowIndex+1;i++)
for(int j=1;j<=i;j++){
int temp = nums[j];
nums[j]=nums[0]+nums[j];
nums[0]=temp;
}
List list = new ArrayList();
for(int i=1;i<=rowIndex+1;i++)
list.add(nums[i]);
return list;
}
}
下面附上几组测试用例:
关于Pascal’s Triangle的相关问题,可以看