uva Krypton Number System dp+矩阵优化

题意：

规定一种数Krypton Number，有以下几个定义

1.这种数是k进制数（0<=k<=6）

2.不能出现连续相同的数，例如112不合法，但是121可以。

3.没有前导零

4.一个Krypton Number有他的分数，分数等于任意相邻2个数差的平方和。

给出Krypton Number的进制，然后一个分数s，问有多少个数分数为s。s<=1e9 答案对2^32取mod。

分析：

如果s不是很大我们可以进行dp，但是s到了1e9会爆炸。假设s不是很大我们可以有dp【s】【i】【j】表示最后一位为i的j位Krypton Number数的分数为s的情况有多少种。

考虑前2维dp【s】【i】因为k最大是6所以加一个数最大的得分为25分所以可以用一个dp【30】【7】的数组把一个数的情况预处理出来，然后可以构造矩阵来优化第3维的转化。

ACcode：

#include <bits/stdc++.h>
#define ll unsigned long long
#define maxn 152
const ll mod=1ULL<<32;
using namespace std;
ll dp[26][7],A[maxn];
int loop,cnt=1,b,s,m,n;
struct Matrix{
    ll mat[maxn][maxn];
};
inline Matrix mul(Matrix a,Matrix b){
    Matrix ret;
    for(int i=0;i<n;++i)
        for(int j=0;j<n;++j){
            ret.mat[i][j]=0;
            for(int k=0;k<n;++k)
                ret.mat[i][j]=(ret.mat[i][j]+a.mat[i][k]*b.mat[k][j]%mod)%mod;
        }
    return ret;
}
inline Matrix pow_M(Matrix a,int k){
    Matrix ret;
    memset(ret.mat,0,sizeof(ret.mat));
    for(int i=0;i<n;++i)ret.mat[i][i]=1;
    Matrix tmp=a;
    while(k){
        if(k&1)ret=mul(ret,tmp);
        tmp=mul(tmp,tmp);
        k>>=1;
    }
    return ret;
}
int main(){
    scanf("%d",&loop);
    while(loop--){
        scanf("%d%d",&b,&s);
        m=(b-1)*(b-1);
        n=m*b;
        memset(dp,0,sizeof(dp));
        for(int i=0;i<b;++i)dp[0][i]=1;
        for(int i=1;i<=m;++i){
            for(int j=0;j<b;++j){
                for(int k=0;k<b;++k){
                    if(k==j)continue;
                    int tmp=(k-j)*(k-j);
                    if(i-tmp<0)continue;
                    dp[i][j]=(dp[i][j]+dp[i-tmp][k])%mod;
                }
            }
        }
        ll ans=0;
        if(s<=m){
            for(int i=1;i<b;++i)
                ans=(ans+dp[s][i])%mod;
        }else {
            for(int i=1;i<=m;++i)
                for(int j=0;j<b;++j)
                    A[(i-1)*b+j]=dp[i][j];
            Matrix B;
            memset(B.mat,0,sizeof(B.mat));
            for(int i=b;i<n;++i)
                B.mat[i][i-b]=1;
            for(int i=1;i<=m;++i){
                for(int j=0;j<b;++j){
                    for(int k=0;k<b;++k){
                        if(j==k)continue;
                        if(i+(k-j)*(k-j)==m+1)
                            B.mat[(i-1)*b+j][n-b+k]=1;
                    }
                }
            }
            B=pow_M(B,s-m);
            for(int i=n-b+1;i<n;++i)
                for(int j=0;j<n;++j)
                    ans=(ans+A[j]*B.mat[j][i]%mod)%mod;
        }
        cout<<"Case "<<cnt++<<": "<<ans<<'\12';
    }
    return 0;
}