[LeetCode 118,119][簡單]楊輝三角/楊輝三角II

118.楊輝三角
題目鏈接
存一個楊輝三角

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        vector<vector<int>>ans;
        for(int i=0;i<numRows;i++){
            vector<int>tmp;
            tmp.emplace_back(1);
            for(int j=1;j<i;j++)tmp.emplace_back(ans[i-1][j-1]+ans[i-1][j]);
            if(i!=0)tmp.emplace_back(1);
            ans.emplace_back(tmp);
        }
        return ans;
    }
};

119.楊輝三角 II
題目鏈接
就暴力寫個逆元吧。。。。

class Solution {
public:
    typedef long long LL;
    const LL mod = INT_MAX;
    LL p[34];
    void init(){
        p[0]=1;
        for(int i=1;i<=33;i++){
            p[i]=(p[i-1]*i)%mod;
        }
    }
    LL qpow(LL a,LL b){
        LL ans = 1;
        while(b){
            if(b&1){
                ans *= a;
                ans %= mod;
            }
            b>>=1;
            a*=a;
            a%=mod;
        }
        return ans;
    }
    LL C(LL m, LL n){
        return p[m]*qpow(p[n],mod-2)%mod*qpow(p[m-n],mod-2)%mod;
    }
    vector<int> getRow(int rowIndex) {
        init();vector<int>ans;
        for(int i=0;i<=rowIndex;i++){
            ans.emplace_back(C(rowIndex,i));
        }
        return ans;
    }
};

ACdreamer的blog很不錯，可以康康