bzoj 2326 矩阵快速幂

思路：矩阵快速幂搞一搞。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PII pair<int, int>
#define PLI pair<LL, int>
#define ull unsigned long long
using namespace std;

const int N = 3e5 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int Mod = 1e9 + ;

LL n;
int mod;

struct Matrix {
    int a[][];
    Matrix() {
        memset(a, , sizeof(a));
    }
    void init() {
        for(int i = ; i < ; i++)
            a[i][i] = ;
    }
    Matrix operator * (const Matrix &B) const {
        Matrix C;
        for(int i = ; i < ; i++)
            for(int j = ; j < ; j++)
                for(int k = ; k < ; k++)
                    C.a[i][j] = (C.a[i][j] + 1ll * a[i][k] * B.a[k][j]) % mod;
        return C;
    }
    Matrix operator ^ (LL b) {
        Matrix C; C.init();
        Matrix A = (*this);
        while(b) {
            if(b & ) C = C * A;
            A = A * A; b >>= ;
        }
        return C;
    }
} M;

int main() {
    int Mat[][] = {
        {, , },
        {, , },
        {, , }
    };
    for(int i = ; i < ; i++)
        for(int j = ; j < ; j++)
            M.a[i][j] = Mat[i][j];

    scanf("%lld%d", &n, &mod);

    Matrix A; A.init();
    bool flag = true;
    for(LL i = ; i; i /= ) {
        LL p = ;
        if(n >= i) {
            if(flag) p = n - i + ;
            else p = i *  - i;
            flag = false;
        }
        M.a[][] = i % mod *  % mod;
        A = A * (M ^ p);
    }
    printf("%d\n", (A.a[][] + A.a[][]) % mod);
    return ;
}

/*
*/