hdu 4565 公式推导+矩阵快速幂

参考：http://blog.csdn.net/crazy______/article/details/9021169

这公式真心难想到……向上取整等价于加上后面一个式子，学习了Orz。

写完快速幂，我还一直WA，最后才发现中途计算的时候没有 (a+MOD)%MOD，因为会变成负数，坑啊。。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
using namespace std;
#define MN 2
#define MM 2
int N;
long long a,b,n,m,MOD;
struct matrix
{
       long long a[MN][MM];
}origin,res,ans;
struct matrix2
{
    long long a[2][1];
    void clear()
    {
        memset(a,0,sizeof(a));
    }
}t;
matrix multiply(matrix &x,matrix &y)
{
       matrix temp;
       for(int i=0;i<N;i++)
       {
           for(int j=0;j<N;j++)
            {
                temp.a[i][j]=0;
                for(int k=0;k<N;k++)
                {
                    temp.a[i][j]=(temp.a[i][j]+(x.a[i][k])*(y.a[k][j])+MOD)%MOD;
                }
            }
       }
       return temp;
}
void init()
{
     origin.a[0][0]=2*a;
     origin.a[0][1]=b-a*a;
     origin.a[1][0]=1;
     origin.a[1][1]=0;
     memset(res.a,0,sizeof(res.a));
     for(int i=0;i<N;i++)
        res.a[i][i]=1;
}

void calc(int n)
{
     while(n)
     {
        if(n&1)
        res=multiply(res,origin);
        n>>=1;
        origin=multiply(origin,origin);
     }
     ans=res;
}
long long mult(matrix2 &b)
{
    matrix2 tmp;
    tmp.clear();
    for(int i=0;i<2;i++)
        for(int j=0;j<1;j++)
            for(int k=0;k<2;k++)
            tmp.a[i][j]+=ans.a[i][k]*b.a[k][j];
    return (tmp.a[1][0]+MOD)%MOD;
}
int main()
{
    N=2;
    while(scanf("%I64d%I64d%I64d%I64d",&a,&b,&n,&m)!=EOF)
    {
        MOD=m;
        t.a[0][0]=2*a;
        t.a[1][0]=2;
        init();
        calc(n);
        printf("%I64d\n",mult(t));

    }
    return 0;
}