【bzoj2004】[Hnoi2010]Bus 公交线路

2004: [Hnoi2010]Bus 公交线路

Time Limit: 20 Sec   Memory Limit: 259 MB
Submit: 768   Solved: 529
[ Submit][ Status][ Discuss]

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<P<=10，K<N，1<K<=P

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

Source

[ Submit][ Status][ Discuss]



开拓了lz对矩阵乘法的认识。。。。

看到n和p的范围应该都能想到是状压和矩乘（然而lz这么菜根本没想到。。。。）

我们考虑设一个f[i][s]表示现在最前面的车开到了i的位置，当前前p个位置的公交车的位置为s

那么显然有f[i][s] = Σf[i - 1][x] (x与s << 1后的后p位只有一个位不同)

为什么只有一个位不同呢？

因为最前面的一辆车肯定是要由上一个状态中的某一辆车开过来的呀

很显然s最高位只能为1，那么同一阶段状态最多只会有C(9,4)种，离散一下状态，空间上就可以接受了

接下来考虑一下矩乘：

很容易知道如果f[i][s]能由f[i - 1][x]转移过来，那么不论i为多少s总是能由x转移过来

那么我们可以先预处理好s由哪些x转移来，用一个矩阵存好

然后直接矩乘就好了

代码：

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;

const int mod = 30031;

struct data{
	int r,c;
	int m[200][200];
	data(){r = c = 0; memset(m,0,sizeof(m));}
	data operator * (data b) const
	{
		data ret;
		ret.r = r; ret.c = b.c;
		for (int i = 1; i <= r; i++)
			for (int j = 1; j <= c; j++)
				for (int k = 1; k <= b.c; k++)
					ret.m[i][k] = (ret.m[i][k] + m[i][j] * b.m[j][k]) % mod;
		return ret;
	}
};

data b,o,frd;
int id[2000],n,k,p,tot;

inline data maxtrix_pow(data a,int b)
{
	data ret = o,t = a;
	while (b)
	{
		if (b & 1) ret = ret * t;
		t = t * t;
		b >>= 1;
	}
	return ret;
}

inline int lowbit(int x)
{
	return x & -x;
}

inline int getint()
{
	int ret = 0;
	char c = getchar();
	while (c < '0' || c > '9') c = getchar();
	while (c >= '0' && c <= '9')
		ret = ret * 10 + c - '0',c = getchar();
	return ret;
}

int main()
{
	n = getint(); k = getint(); p = getint();
	for (int i = 1; i <= (1 << p) - 1; i++)
	{
		int cnt = 0;
		for (int j = i; j; j -= lowbit(j)) cnt++;
		if (cnt != k) continue;
		if (!(i & (1 << p - 1))) continue;
		id[i] = ++tot;
	}
	for (int i = 1; i <= (1 << p) - 1; i++)
	{
		if (!id[i]) continue;
		for (int j = 1; j <= (1 << p) - 1; j++)
		{
			if (!id[j]) continue;
			int si = (i << 1) ^ (1 << p),sj = j;
			if (lowbit(si ^ sj) == (si ^ sj)) frd.m[id[i]][id[j]] = 1;
		}
	}
	frd.r = frd.c = tot;
	int bit = (1 << p - 1);
	int from = 0;
	for (int i = 1; i <= k; i++)
	{
		from ^= bit;
		bit >>= 1;
	}
	data ans;
	ans.m[id[from]][1] = 1; ans.r = tot; ans.c = 1;
	
	o.r = o.c = tot;
	for (int i = 1; i <= tot; i++) o.m[i][i] = 1;
	data p = maxtrix_pow(frd,n - k);
	ans = p * ans;
	printf("%d",ans.m[id[from]][1]);
	return 0;
}