爆零赛T2(矩阵加速递推式)

61 篇文章 0 订阅
24 篇文章 0 订阅

前言:
曲神:今天中午做了个噩梦,梦见给DP讲题,讲了半个小时。。。

这里写图片描述

分析:
终于在曲神,ZYXZYXZYX和舒老师的共同努力下,zz博主终于学会了70%做法

动态规划
我们首先就要设计状态:

f[i][j]:表示到第i位为止,[i-j+1,i]这j位每一位上的颜色都不同(一共有j种颜色),这种情况下的方案数

我们需要从f[i-1]转移到f[i]
有两种情况:
(色差盲福利)
这里写图片描述

得到的状态转移:
这里写图片描述

要注意,不能有连续m种不同颜色,所以第二维只能到m-1

这样我们就可以转移了
为了提高效率,我们考虑用矩阵加速:
这里写图片描述

我们处理出f[1]~f[m-1]
之后的n-m+1项,我们用矩阵加速

//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long

using namespace std;

const ll p=1000000007;
ll n,f[1001][201],ans[1001];
int m;

struct node{
    ll H[101][101];
    node operator *(const node &a) const
    {
        node ans;
        for (int i=1;i<m;i++)
            for (int j=1;j<m;j++)
            {
                ans.H[i][j]=0;
                for (int k=1;k<m;k++)
                    ans.H[i][j]=(ans.H[i][j]%p+H[i][k]*a.H[k][j]%p)%p;
            }
        return ans;
    }
    node clear()
    {
        memset(H,0,sizeof(H));
    }
    node KSM(ll p)
    {
        node a=(* this);
        node ans=(* this);
        p--;
        while (p)
        {
            if (p&1) ans=ans*a;
            p>>=1;
            a=a*a;
        }
        return ans;
    }
};
node H;

ll KSM(ll a,ll b)
{
    ll t=1LL;
    a%=p;
    while (b)
    {
        if (b&1)
           t=(t%p*a%p)%p;
        b>>=1;
        a=(a%p*a%p)%p;
    }
    return t%p;
}

int main()
{
    freopen("herzlich.in","r",stdin);  
    freopen("herzlich.out","w",stdout);
    scanf("%lld%d",&n,&m);
    if (m>n)
    {
        printf("%lld",KSM(m,n)%p);
        return 0;
    }

    for (int i=1;i<m;i++)
        for (int j=1;j<=i;j++)
            H.H[i][j]=1;
    for (int i=1;i<m-1;i++) H.H[i][i+1]=m-i;

    f[1][1]=m;
    for (int i=2;i<m;i++)      //递推式 
    {
        for (int j=2;j<m;j++)
        {
            f[i][j]=f[i-1][j-1]*(m-j+1)%p;
            for (int k=j;k<m;k++)
                f[i][j]=(f[i][j]+f[i-1][k])%p;
        }
        for (int k=1;k<m;k++) f[i][1]=(f[i][1]+f[i-1][k])%p;
    }

    node an=H.KSM(n-(ll)(m-1));
    for (int i=1;i<m;i++)
        for (int k=1;k<m;k++)
            ans[i]=(ans[i]+f[m-1][k]*an.H[k][i])%p;

    ll sum=0;
    for (int i=1;i<m;i++) sum=(sum+ans[i])%p;
    printf("%lld\n",sum);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值