UVa 10655

祭奠下我又一个wa了20+次的题，（都是泪啊。~~），这个题的也其实就是一个矩阵快速幂，但是写的时候有一些小坑，比如数据都要取long long 的，改了后我还是 wa，其实是结构体的里的矩阵也要ll，然而这个 一直忘记改了，就这样一直wa。。。。。。。。。。。还有就是在重载 * 运算的时候的行列大小要搞对，m*n 的与n*p的相乘，可以得到m*p然后中间的变量k的值就是最大到第一个矩阵的列数或者第二个矩阵的行数。

最后还有一个地方有小坑，就是当p=q=0的时候，不要急着就退出了，因为这个时候a=b=0，那么输出可以是0，不要退出（虽然题目也告诉我要退出，但是就改了这么一个地方测试着交了一发，就会wa掉，= =！）（题目的意思应该是只包含两个0的，如果是0 0 k 这样三个的就继续正常进行咯？！= =？，感觉以后还是好好读题把。）

下面包含着两种写法（另外一个是递归的）

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>

using namespace std;
typedef long long ll;
ll p,q,n;
struct Mat {
    ll mat[5][5];
    int r,c;
    //Mat(int _r,int _c):r(_r),c(_c) {};
};

Mat operator * (Mat a, Mat b) {
    Mat m;
    memset(m.mat, 0, sizeof(m.mat));
    m.r=a.r;
    m.c=b.c;
    int i, j, k;
        for(i = 0; i < m.r; ++i) {
           // if(a.mat[i][k] <= 0)  continue;    //***
            for(j = 0; j < m.c; ++j) {
               // if(b.mat[k][j] <= 0)    continue;    //***
               for(k=0;k<a.c;++k)
                m.mat[i][j] = m.mat[i][j]+a.mat[i][k] * b.mat[k][j];
            }
        }
    return m;
}

Mat operator ^ (Mat a, ll k) {
    Mat m;
    m.r=a.r;
    m.c=a.c;
    int i,j;
    for(i = 0; i < m.r; ++i)
        for(j = 0; j < m.c; ++j)
            m.mat[i][j] = (i == j);
    for(; k; k >>= 1) {
        if(k&1) m = m*a;
        a = a*a;
    }
    return m;
}

Mat multiply(Mat a,Mat b)
 {
     Mat m;
     m.r=a.r;m.c=a.c;
     for(int i=0;i<m.r;++i){
       for(int j=0;j<m.c;++j){
            m.mat[i][j]=0;
                for(int k=0;k<a.c;++k)
                 m.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
         }
     }
     return m;
 }

Mat matrix_pow(Mat m,int n)
 {
    if(n==0){
         m.mat[0][0]=m.mat[1][1]=1;
         m.mat[0][1]=m.mat[1][0]=0;
         return m;
     }
     if(n==1)
         return m;
     Mat res=matrix_pow(m,n/2);
     res=multiply(res,res);
     if(n&1)
         res=multiply(res,m);
     return res;
 }

int main()
{
    while(scanf("%lld%lld%lld",&p,&q,&n)==3)
    {
        if(n==0)
        {
            printf("2\n");
            continue;
        }
         if(n==1)
        {
           printf("%d\n",p);
             continue;
        }
        Mat a;
      a.mat[0][0]=p;
      a.mat[0][1]=-q;
      a.mat[1][0]=1;
      a.mat[1][1]=0;
      a.r=2;a.c=2;
      Mat b;
      b.mat[0][0]=p*p-2*q;
      b.mat[1][0]=p;
      b.r=2;b.c=1;
        ll k=n-2;
        Mat tmp=a^k;
        tmp=tmp*b;
        ll sum=tmp.mat[0][0];
       cout<<sum<<endl;
    }
    return 0;
}