hdu4686 简单的矩阵快速幂求前n项和

HDU4686

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4686

题意：题目说的很清楚了，英语不好的猜也该猜懂了，就是求一个表达式的前n项和，矩阵快速幂一般多加一行一列来完成这个加的操作。具体看代码吧。比较简单，唯一有一点坑的地方，就是ax和bx可能比较大，在求ax*bx的时候，要考虑溢出的问题，需要先mod。其他没有了，直接看代码吧！

//Author: xiaowuga
#include <bits/stdc++.h>
#define maxx INT_MAX
#define minn INT_MIN
#define inf 0x3f3f3f3f
#define n 5
#define mod 1000000007
using namespace std;
typedef long long ll;
ll a0,b0,ax,bx,ay,by;
struct Matrix{
    ll mat[][];
    Matrix operator * (const Matrix & m) const{
        Matrix tmp;
        for(int i=;i<n;i++)
            for(int j=;j<n;j++){
                tmp.mat[i][j]=;
                for(int k=;k<n;k++){
                    tmp.mat[i][j]+=mat[i][k]*m.mat[k][j]%mod;
                    tmp.mat[i][j]%=mod;
                }
            }
        return tmp;
    }
};
Matrix q_power(Matrix a,ll k){
    Matrix ans;
    memset(ans.mat,,sizeof(ans.mat));
    for(int i=;i<n;i++) ans.mat[i][i]=;
    while(k){
        if(k&) ans=ans*a;
        k/=;
        a=a*a;
    }
    return ans;

}
int main() {
    ios::sync_with_stdio(false);cin.tie();
    ll T;
    while(cin>>T>>a0>>ax>>ay>>b0>>bx>>by){
        if(T==){cout<<<<endl;continue;}
        Matrix m;
        memset(m.mat,,sizeof(m.mat));
        m.mat[][]=m.mat[][]=ax*bx%mod;
        m.mat[][]=ax*by%mod;m.mat[][]=ax%mod;m.mat[][]=ax*by%mod;
        m.mat[][]=bx*ay%mod;m.mat[][]=bx%mod; m.mat[][]=bx*ay%mod;
        m.mat[][]=by*ay%mod;m.mat[][]=ay%mod;m.mat[][]=by%mod;m.mat[][]=;m.mat[][]=by*ay%mod;
        m.mat[][]=;
        Matrix p=q_power(m,T-);
        Matrix f;
        memset(f.mat,,sizeof(f.mat));
        f.mat[][]=f.mat[][]=a0*b0%mod;f.mat[][]=a0%mod;f.mat[][]=b0%mod;f.mat[][]=;
        f=f*p;
        cout<<f.mat[][]<<endl;
    }
    return ;
}

矩阵有点难画，我就不画了，根据我的对矩阵m的赋值，把矩阵画出来，你就会明白了为什么这个构造矩阵了