ACM-数论-矩阵快速幂 HDU5015&HDU2276

HDU2276题目链接
HDU5015题目链接
2276
矩阵：（只有左边和本身有关联（1），其他都无关（0））
第一个样例

1 0 0 0 0 0 1
1 1 0 0 0 0 0
0 1 1 0 0 0 0
0 0 1 1 0 0 0
0 0 0 1 1 0 0
0 0 0 0 1 1 0
0 0 0 0 0 1 1

代码：

#include<iostream>
#include<string.h>
#include<string>
#include<cmath>
#include<algorithm>
#include<cstdio>
#define mem(x,y) memset(x,y,sizeof(x)) 
using namespace std;
typedef long long ll;
int n;
ll m;
struct mx{
    int v[102][102];
}a;
mx mult(mx x,mx y,int n){
    mx res;
    mem(res.v,0);
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            for(int k=0;k<n;k++){
                res.v[i][j]+=(x.v[i][k]&y.v[k][j]);
                //(1+1)%2和1^1结果一样，1*1和1&1结果一样，所以相乘函数改下就行了
                res.v[i][j]=res.v[i][j]&1;
            }
        }
    }
    return res;
}
mx power(mx a,ll b,int n){
    mx res;
    mem(res.v,0);
    for(int i=0;i<n;i++)res.v[i][i]=1;
    while(b>0){
        if(b&1)res=mult(res,a,n);
        b=b>>1;
        a=mult(a,a,n);
    }
    return res;
}
int main(){
    int k[102];
    string s;
    while(scanf("%lld",&m)!=EOF){
        cin>>s;
        int n=s.size();
        for(int i=0;i<n;i++)k[i]=(int)(s[i]-'0');
        mem(a.v,0);
        a.v[0][n-1]=1;a.v[0][0]=1;
        for(int i=1;i<n;i++){
            a.v[i][i-1]=1;a.v[i][i]=1;
        }
        mx res=power(a,m,n);
        int y=0;
        /*for(int i=0;i<n;i++){
            for(int j=0;j<n;j++)cout<<res.v[i][j]<<" ";
            cout<<endl;
        }*/
        for(int i=0;i<n;i++){
            y=0;
            for(int j=0;j<n;j++)y+=k[j]*res.v[i][j];
            printf("%d",y&1);
        }
        printf("\n");
    }
    return 0;
}

5015
矩阵：（比如样例第二个）

10 0 0 1
10 1 0 1
10 1 1 1
0  0 0 1

代码：

#include<iostream>
#include<string.h>
#include<string>
#include<cmath>
#include<algorithm>
#include<cstdio>
#define mem(x,y) memset(x,y,sizeof(x)) 
using namespace std;
typedef long long ll;
const int u=23;
int n;
ll m;
const int mod=10000007;
struct mx{
    ll v[12][12];
}a;
mx mult(mx x,mx y,int n){
    mx res;
    mem(res.v,0);
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            for(int k=0;k<n;k++){
                res.v[i][j]+=(x.v[i][k]*y.v[k][j])%mod;
                res.v[i][j]%=mod;
            }
        }
    }
    return res;
}
mx power(mx a,ll b,int n){
    mx res;
    mem(res.v,0);
    for(int i=0;i<n;i++)res.v[i][i]=1;
    while(b>0){
        if(b&1)res=mult(res,a,n);
        b=b>>1;
        a=mult(a,a,n);
    }
    return res;
}
int main(){
    ll k[12];
    while(scanf("%d%lld",&n,&m)!=EOF){
        for(int i=1;i<=n;i++){
            cin>>k[i];
            k[i]%=mod;
        }
        mem(a.v,0);
        for(int i=0;i<n+1;i++){
            a.v[i][0]=10;
            for(int j=1;j<=i;j++)a.v[i][j]=1;
            a.v[i][n+1]=1;
        }
        a.v[n+1][n+1]=1;
        /*for(int i=0;i<n+2;i++){
            for(int j=0;j<n+2;j++)cout<<a.v[i][j]<<" ";
            cout<<endl; 
        }*/
        mx res=power(a,m,n+2);
        /*cout<<"print\n";
        for(int i=0;i<n+2;i++){
            for(int j=0;j<n+2;j++)cout<<res.v[i][j]<<" ";
            cout<<endl; 
        }*/
        ll ans=(res.v[n][0]*u%mod+3*res.v[n][n+1])%mod;
        //printf("first=%lld\n",ans);
        for(int i=1;i<=n;i++){
            ans=(ans%mod+res.v[n][i]*k[i]%mod)%mod;
        }
        printf("%lld\n",ans);//cout<<ans<<endl;
    }
    return 0;
}