计蒜客 31721- ACM-ICPC 2018 焦作赛区网络预赛 - L. Poor God Water - 矩阵快速幂

题目链接https://nanti.jisuanke.com/t/31721

题意:有鱼,肉,巧克力三种食物,有N个小时,现在要求每小时吃一种食物,而且下边的吃法不合法:

  1. 续三小时吃一种食物
  2. 连续三小时三种食物都吃并且中间吃的巧克力
  3. 连续三小时,两边吃巧克力,中间吃了鱼或者肉

问N小时内共有几种吃法。

解析:假设鱼,巧克力,肉分别用字符X,Y,Z表示,那么合法的长度为N的字符串的组成方案数即是答案。

          考虑由前面时间的状态推出后面时间的状态,但是当加一个新字符时,我们要考虑旧字符串最后两个字符的状态,这样是不好推的。我们可以考虑每次添加两个新字符(共9种),然后用旧字符串最后两个字符(共9种)的状态进行转移。生成一个如下9*9的矩阵。

其中1代表两个字符串能连接,0表示不能,比如左边列"ZY"与上面行"ZX",表示""ZY"之后加上"ZX"其值为1表示可以加。

那么这个初始矩阵中1的个数就表示所有长度为4的合法字符串的方案数,那么对于两个这种矩阵相乘就表示在原来长度为4的字符串上再加2个字符,矩阵中所有元素和就是长度为6的合法字符串的方案数。

那么对于N为偶数直接对初始矩阵求(N-2)/2次幂,然后最终矩阵中元素的和就是方案总数。N为奇数时,那么N-1为偶数,先生成长度为N-1对应的矩阵(初始矩阵求(N-2-1)/2次幂),然后在乘以下矩阵,表示加一个字符,最终矩阵的元素和就是总方案数:

此矩阵与上面矩阵原理相同。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const ll M=9;

ll N;

struct Matrix
{
    Matrix(){memset(a,0,sizeof(a));}
    ll a[M][M];
    void init()
    {
        for(ll i=0;i<M;i++)
        for(ll j=0;j<M;j++)
            a[i][j]=0;
        for(ll i=0;i<M;i++)
            a[i][i]=1;
    }
}A,B,C;
Matrix mul(Matrix a,Matrix b) //(a*b)%mod
{
    Matrix ans;
    ll i,j,k;
    for(i=0;i<M;i++)
        for(j=0;j<M;j++)
    {
        for(k=0;k<M;k++)
            ans.a[i][j]+=a.a[i][k]*b.a[k][j];
        ans.a[i][j]%=mod;
    }
    return ans;
}
Matrix pow(Matrix a,ll n) //(a^n)%mod
{
    Matrix ans;
    ans.init();
    while(n)
    {
        if(n&1)
            ans=mul(ans,a);
        n>>=1;
        a=mul(a,a);
    }
    return ans;
}
ll sum(Matrix a,int col)
{
    ll res=0;
    for(int i=0;i<9;i++)
        for(int j=0;j<col;j++)
            res=(res+a.a[i][j])%mod;
    return res;
}

int main()
{
    A.a[0][0]=0; A.a[0][1]=0; A.a[0][2]=0; A.a[0][3]=1; A.a[0][4]=1; A.a[0][5]=0; A.a[0][6]=1; A.a[0][7]=1; A.a[0][8]=1;
    A.a[1][0]=1; A.a[1][1]=0; A.a[1][2]=1; A.a[1][3]=1; A.a[1][4]=0; A.a[1][5]=1; A.a[1][6]=0; A.a[1][7]=0; A.a[1][8]=0;
    A.a[2][0]=1; A.a[2][1]=1; A.a[2][2]=1; A.a[2][3]=1; A.a[2][4]=1; A.a[2][5]=0; A.a[2][6]=1; A.a[2][7]=1; A.a[2][8]=0;
    A.a[3][0]=0; A.a[3][1]=1; A.a[3][2]=1; A.a[3][3]=0; A.a[3][4]=0; A.a[3][5]=0; A.a[3][6]=1; A.a[3][7]=1; A.a[3][8]=1;
    A.a[4][0]=1; A.a[4][1]=0; A.a[4][2]=1; A.a[4][3]=0; A.a[4][4]=0; A.a[4][5]=0; A.a[4][6]=1; A.a[4][7]=0; A.a[4][8]=1;
    A.a[5][0]=1; A.a[5][1]=1; A.a[5][2]=1; A.a[5][3]=0; A.a[5][4]=0; A.a[5][5]=0; A.a[5][6]=1; A.a[5][7]=1; A.a[5][8]=0;
    A.a[6][0]=0; A.a[6][1]=1; A.a[6][2]=1; A.a[6][3]=0; A.a[6][4]=1; A.a[6][5]=1; A.a[6][6]=1; A.a[6][7]=1; A.a[6][8]=1;
    A.a[7][0]=0; A.a[7][1]=0; A.a[7][2]=0; A.a[7][3]=1; A.a[7][4]=0; A.a[7][5]=1; A.a[7][6]=1; A.a[7][7]=0; A.a[7][8]=1;
    A.a[8][0]=1; A.a[8][1]=1; A.a[8][2]=1; A.a[8][3]=0; A.a[8][4]=1; A.a[8][5]=1; A.a[8][6]=0; A.a[8][7]=0; A.a[8][8]=0;

    C.a[0][0]=0; C.a[0][1]=1; C.a[0][2]=1;
    C.a[1][0]=1; C.a[1][1]=1; C.a[1][2]=0;
    C.a[2][0]=1; C.a[2][1]=1; C.a[2][2]=1;
    C.a[3][0]=1; C.a[3][1]=0; C.a[3][2]=1;
    C.a[4][0]=1; C.a[4][1]=0; C.a[4][2]=1;
    C.a[5][0]=1; C.a[5][1]=0; C.a[5][2]=1;
    C.a[6][0]=1; C.a[6][1]=1; C.a[6][2]=1;
    C.a[7][0]=0; C.a[7][1]=1; C.a[7][2]=1;
    C.a[8][0]=1; C.a[8][1]=1; C.a[8][2]=0;

    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld",&N);
        if(N==1)
        {
            printf("3\n");
        }else if(N==2){
            printf("9\n");
        }else if(N==3){
            printf("20\n");
        }else if((N&1)==0)
        {
            B=pow(A,(N-2)/2);
            printf("%lld\n",sum(B,9));
        }else{
            B=pow(A,(N-2-1)/2);
            Matrix D;
            for(int i=0;i<9;i++)
            {
                for(int j=0;j<3;j++)
                {
                    for(int k=0;k<9;k++)
                        D.a[i][j]+=B.a[i][k]*C.a[k][j];
                    D.a[i][j]%=mod;
                }
            }
            printf("%lld\n",sum(D,3));
        }
    }
    return 0;
}

/*
1 3
2 9
3 20
4 46
5 106
6 244
7 560
8 1286
9 2956
10 6794
11 15610
12 35866
13 82416
14 189384
15 435170
16 999936
*/

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值