【BZOJ2004】公交线路（动态规划，状态压缩，矩阵快速幂）

【BZOJ2004】公交线路（动态规划，状态压缩，矩阵快速幂）

题面

BZOJ

题解

看到\(k,p\)这么小

不难想到状态压缩

看到\(n\)这么大，不难想到矩阵快速幂

那么，我们来考虑朴素的\(dp\)

设\(f[i][j]\)表示当前位置为\(i\)，前面的\(P\)个位置的状态为\(j\)

其中，状态的含义是某个公交线路最后的停靠站

如果是最后的停靠站就是\(1\)，否则是\(0\)

那么，任意状态中只存在\(k\)个\(1\)

并且表示\(i\)的二进制位一定是\(1\)

所以状态相当于最多只会有\(120\)个左右

考虑转移，

每次相当于从原来的状态中，

把一个\(1\)放到了当前位置

其他的依次向左挪一个位置

那么，如果两个状态之间满足这个关系，就可以转移

我们发现转移之和后面的\(j\)有关

所以矩阵快速幂就行了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MOD 30031
inline int read()
{
RG int x=0,t=1;RG char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=-1,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return x*t;
}
int n,K,P;
int zt[1<<11],ss,mp[1<<11];
struct Dalao
{
int s[150][150];
void init(){memset(s,0,sizeof(s));}
void pre()
{
init();
for(int i=1;i<=ss;++i)s[i][i]=1;
}
}zsy;
Dalao operator*(Dalao a,Dalao b)
{
Dalao ret;ret.init();
for(int i=1;i<=ss;++i)
for(int j=1;j<=ss;++j)
for(int k=1;k<=ss;++k)
ret.s[i][j]=(ret.s[i][j]+a.s[i][k]*b.s[k][j]%MOD)%MOD;
return ret;
}
Dalao fpow(Dalao a,int b)
{
Dalao s;s.pre();
while(b){if(b&1)s=s*a;a=a*a;b>>=1;}
return s;
}
int ws(int x){int ret=0;while(x)++ret,x-=x&(-x);return ret;}
bool check(int a,int b)
{
if(ws((a<<1)&b)!=K-1)return false;
return true;
}
int main()
{
n=read();K=read();P=read();
for(int i=0;i<(1<<P);++i)
if((i&1)&&ws(i)==K)zt[++ss]=i;
for(int i=1;i<=ss;++i)
for(int j=1;j<=ss;++j)
if(check(zt[i],zt[j]))
zsy.s[j][i]=1;
zsy=fpow(zsy,n-K);
printf("%d\n",zsy.s[1][1]);
return 0;
}