【BZOJ】2004: [Hnoi2010]Bus 公交线路 状压DP+矩阵快速幂

【题意】n个点等距排列在长度为n-1的直线上,初始点1~k都有一辆公车,每辆公车都需要一些停靠点,每个点至多只能被一辆公车停靠,且每辆公车相邻两个停靠点的距离至多为p,所有公车最后会停在n-k+1~n。给定n,k,p,求满足要求的方案数%30031。n<=10^9,k<=p<=10

【算法】状压DP+矩阵快速幂

【题解】开始没看到p<=10,其实很显然p>k的话第一车就不满足要求了。考虑相邻停靠点没有关键信息,只能状压。

因为车都是从头开到尾的,所以直接考虑i~i-p+1的公车停靠状态就行了(i-p和i的距离为p,也就是必须跳到i,因此考虑i-p没有意义)。

设$f[i][S]$表示考虑第i个位置(最快的车可能到i),i~i-p+1的公车停靠状态为S的方案数,并且强制S的最低位为1(从前往后从低到高)。

$$f[i][S]=\sum f[i-1][S']$$

将S右移一位,然后枚举0~p中没有1的位置插入一个1得到的就是合法的S‘,然后把所有状态放在矩阵中就可以快速幂n-k次了。

注意到S是有效状态当且仅当S中含有恰好k个1,所以预处理合法状态数是C(p,k)的,这样复杂度就可以保证了。

总复杂度O(C(p,k)^3*log n)。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=520,MOD=30031;
int n,k,p,tot,c[N][N],q[N*2],A[N][N],ANS[N][N];
void multply(int a[N][N],int b[N][N]){
    for(int i=1;i<=tot;i++){
        for(int j=1;j<=tot;j++){
            c[i][j]=0;
            for(int k=1;k<=tot;k++){
                c[i][j]=(c[i][j]+a[i][k]*b[k][j])%MOD;
            }
        }
    }
    for(int i=1;i<=tot;i++)for(int j=1;j<=tot;j++)b[i][j]=c[i][j];
}
int main(){
    scanf("%d%d%d",&n,&k,&p);
    for(int S=0;S<(1<<p);S++)if(S&1){
        int num=0;
        for(int i=0;i<p;i++)if(S&(1<<i))num++;
        if(num!=k)continue;
        tot++;q[S]=tot;
    }
    for(int S=0;S<(1<<p);S++)if(q[S]){
        int s=S>>1;
        for(int i=0;i<p;i++)if(!(s&(1<<i))&&q[s|(1<<i)])A[q[S]][q[s|(1<<i)]]=1;//
    }
    n-=k;
    int s=(1<<k)-1;
    ANS[q[s]][1]=1;
    while(n){
        if(n&1)multply(A,ANS);
        multply(A,A);
        n>>=1;
    }
    s=(1<<k)-1;
    printf("%d",ANS[q[s]][1]);
    return 0;
}
View Code

 

posted @ 2018-04-10 22:12  ONION_CYC  阅读(...)  评论(...编辑  收藏