Pascal's Triangle II 帕斯卡三角形系列2

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;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值