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?
需要我们生成第 i 行的值。
同时,希望我们的算法的空间复杂度为O(k)
我是把这个问题当做数学问题来实现的。所花费的空间为O(k)
当n = 3 时候,数组为[1, 3, 3, 1]
其实,正是组合数,C30, C31 C32 C33
这里要注意的两点:
1. 组合数是对称的,C30 = C33 , C31 = C32;所以我们只需要计算一般的组合数就可以了。
2. 防止相乘的时候溢出。
对于Cnk 我们需要计算n * n - 1*..* (n - k +1)的值,同时计算k * k - 1*..*1 ,最后然后相除,这么做的话,很可能被除数会溢出的!
防止溢出的方法是直接将除法得到的结果累乘。如 n / k * (n - 1) / ( k - 1) * (n - k + 1) / 1。
运行时间:
代码:
public class PascalTriangleII {
public List<Integer> getRow(int rowIndex) {
List<Integer> result = new ArrayList<>(rowIndex);
int mid = rowIndex / 2;
for (int k = 0; k <= rowIndex; k++) {
if (k > mid)
result.add(result.get(rowIndex - k));
else
result.add((int)Math.round(calCNK(rowIndex, k)));
}
return result;
}
private double calCNK(int n , int k) {
double result = 1;
long curK = k;
while (curK > 0) {
result *= (n - (k - curK)) * 1.0/ curK;
curK--;
}
return result;
}
}