【bzoj2004】[Hnoi2010]Bus 公交线路

题目链接

Description

小Z所在的城市有N个公交车站，排列在一条长(N-1)km的直线上，从左到右依次编号为1到N，相邻公交车站间的距离均为1km。 作为公交车线路的规划者，小Z调查了市民的需求，决定按下述规则设计线路：
1.设共K辆公交车，则1到K号站作为始发站，N-K+1到N号台作为终点站。
2.每个车站必须被一辆且仅一辆公交车经过（始发站和终点站也算被经过）。
3.公交车只能从编号较小的站台驶往编号较大的站台。
4.一辆公交车经过的相邻两个站台间距离不得超过Pkm。 在最终设计线路之前，小Z想知道有多少种满足要求的方案。由于答案可能很大，你只需求出答案对30031取模的结果。

Input

仅一行包含三个正整数N K P，分别表示公交车站数，公交车数，相邻站台的距离限制。
N<=10^9，1

Output

仅包含一个整数，表示满足要求的方案数对30031取模的结果。

Sample Input

样例一：10 3 3

样例二：5 2 3

样例三：10 2 4

Sample Output

1

3

81

HINT

【样例说明】

样例一的可行方案如下： (1,4,7,10)，(2,5,8)，(3,6,9)

样例二的可行方案如下： (1,3,5)，(2,4) (1,3,4)，(2,5) (1,4)，(2,3,5)

P<=10 , K <=8

题解

我们用一个p位二进制数来表示连续p个位置中k辆车的停车情况，1表示有车停，0表示没有。所以这个二进制数中有且仅有k个1，同时为了不重复表示状态，所以规定每个状态的最高位都为1。
那么每个状态在仅移动最靠前的车的情况下可以转移到右侧的其他状态（为了不重复规定只可以移动最靠前的车），而x状态可以转移到y状态，当且仅当x去掉最高位整体左移移位后与y仅有一位。构造出转移矩阵（与DNA Sequence差不多），该矩阵的n-k次方就是最终答案矩阵。

#include<bits/stdc++.h>
using namespace std;

#define rep(i, l, r) for(int i = (l); i <= (r); i++)

inline int read(){
    int x = 0, f = 1; char c = getchar();
    while(!isdigit(c)) { if(c == '-') f = -1; c = getchar(); }
    while(isdigit(c)) { x = x * 10 + c - '0'; c = getchar(); }
    return x * f;
}

const int mod = 30031, N = 150;
int n, k, p;

struct Mat{
    int x, y, dat[N][N];
    Mat(){memset(dat, 0, sizeof(dat));}

    friend Mat operator * (const Mat a, const Mat b){
        Mat mul; mul.x = a.x; mul.y = b.y;
        rep(i, 1, mul.x) rep(j, 1, mul.y) rep(k, 1, a.y){
            mul.dat[i][j] += a.dat[i][k] * b.dat[k][j];
            mul.dat[i][j] %= mod;
        }
        return mul;
    }

    friend Mat operator ^ (const Mat a, int n){
        Mat b = a, ret;
        ret.x = b.x, ret.y = b.y;
        rep(i, 1, b.x) ret.dat[i][i] = 1;
        while(n){
            if(n & 1) ret = ret * b;
            b = b * b;
            n >>= 1;
        }
        return ret;
    }
}ans, uni;

int stk[N], top;

void init(){
    n = read(); k = read(); p = read();
}

int cal(int x){
    int cnt = 0;
    while(x){
        cnt ++;
        x -= x&-x;
    }
    return cnt;
}

bool check(int x, int y){
    x = (x << 1) - (1 << p);
    int t = x ^ y;
    if(cal(t) == 1) return true;
    else return false;
}

void work(){
    int res;
    for(int i = 1 << p - 1; i <= (1 << p) - 1; i++){
        if(cal(i) == k) stk[++top] = i;
        if(i == (1 << p) - (1 << p - k)) res = top;
    }
    uni.x = uni.y = top;
    ans.x = 1; ans.y = top;
    ans.dat[1][res] = 1;
    rep(i, 1, top) rep(j, 1, top)
        if(check(stk[i], stk[j])) uni.dat[i][j] = 1;
    uni = uni ^ (n - k);
    ans = ans * uni;
    printf("%d\n", ans.dat[1][res]);
}

int main(){
    init();
    work();
    return 0;
}