ホームに戻る
行列累乗

// 漸化式の大きな値の項を高速で計算する。
// Xn=a*Xn-1+b*Xn-2+c*Xn-3 のとき、
// | X3 |   | a b c || X2 |
// | X2 | = | 1 0 0 || X1 |
// | X1 |   | 0 1 0 || X0 |
// と行列を作成できる。

// | Xn   |   | a b c |^n| X2 |
// | Xn-1 | = | 1 0 0 |  | X1 |
// | Xn-2 |   | 0 1 0 |  | X0 |
// とし n の累乗計算を先にやる。
// n は 64ビット値まで可能

// どちらの計算も可能
// | ret |   | v1 v1 || v2 |
// | ret | = | v1 v1 || v2 |
// | ret ret |   | v1 v1 || v2 v2 |
// | ret ret | = | v1 v1 || v2 v2 |
vector<vector<long long> > mul(vector<vector<long long> > v1, vector<vector<long long> > v2){
    vector<vector<long long> > ret(v1.size(),vector<long long>(v2[0].size(),0));
    for(int i=0;i<v1.size();i++) {
        for(int j=0;j<v2[0].size();j++) {
            for(int k=0;k<v1.size();k++) {
                ret[i][j]+=v1[i][k]*v2[k][j];
                ret[i][j]%=MOD;
            }
        }
    }
    return ret;
}

vector<vector<long long> > powmatrix(vector<vector<long long> > vv, vector<vector<long long> > v, long long n){
    vector<vector<vector<long long> > > vvv;
    vvv.push_back(vv);
    for(long long i = 0; i < 64; i++){
        vvv.push_back(mul(vvv[i],vvv[i]));
    }
    vector<vector<long long> > v1(vv.size(), vector<long long>(vv.size(), 0));
    for(int i = 0; i < vv.size(); i++)v1[i][i] = 1;
    for(int i = 0; i < 64; i++){
        if((n>>i)&1){
            v1 = mul(v1,vvv[i]);
        }
    }
    v = mul(v1,v);
    return v;
}

long long n = 12;

vector<vector<long long> > v(2, vector<long long>(1, 0));
v[0][0] = 1;
v[1][0] = 1;
vector<vector<long long> > vv(2, vector<long long>(2, 0));
vv[0][0] = 1; vv[0][1] = 1;
vv[1][0] = 1; vv[1][1] = 0;

// フィボナッチの例
// | va0 |   | 1 1 |^n| 1 |
// | va1 | = | 1 0 |  | 1 |
vector<vector<long long> > va = powmatrix(vv,v,n);

rep(i,0,2){
    cout << va[i][0] << endl;
}

inserted by FC2 system