码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ2004: [Hnoi2010]Bus 公交线路

时间:2017-09-25 22:04:07      阅读:110      评论:0      收藏:0      [点我收藏+]

标签:include   简单   tmp   register   clu   zoj   ons   bsp   排列   

2004: [Hnoi2010]Bus 公交线路

Time Limit: 20 Sec  Memory Limit: 259 MB
Submit: 747  Solved: 512
[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

思路{
  P,K小得不行,想到了状态压缩。n的范围如此之大,考虑矩阵快速幂!
  先从简单的状态压缩开始。设$dp[i][j]$为强制选第i号停车,从第i号开始的后面p个的停车状态。转移。
  发现一个问题(1->3,2->4)和(2->4,1->3)属于同一种情况,所以我们只需要考虑dp[i-1]->dp[i]。
  关键就是后面状态j的转移关系了。发现dp[i-1][j‘]->dp[i][j]满足j‘变成j时实际位置对应的状态中的二进制数的位数+1
  那么先删去i-1位置的数,j‘-=(1<<(p-1)),然后j‘<<1,变成对应的位置,由于j‘->j时只有一个相对位置发生了变化,那么j‘^=j,判断j‘的二进制数中1的个数是否为1即可。
  那么转移方程就出来了。考虑构造矩阵。
  暴力构造矩阵的话大小是(2^p)^2的显然不行,我们只需考虑后面的p位中有k位有车的状态,不妨先DFS找出所有的状态,然后两两枚举即可。注意转移关系在矩阵中对应的位置
  不难发现构造的矩阵大小是(C(p-1,k-1))的,那这个就很对了。
  所以总的复杂度是O(logn*C(p-1,k-1)^3+p!)的。
}

#include<bits/stdc++.h>
#define ll long long
#define RG register
#define il inline
#define N 200
#define mod 30031
using namespace std;
int n,k,p,que[N];
struct matrix{
  int n,m;
  ll ma[N][N];
  void clear(){memset(ma,0,sizeof(ma));n=m=0;}
  matrix operator *(const matrix & a)const {
    matrix c;c.clear();
    c.n=n,c.m=a.m;
    for(int i=1;i<=c.n;++i){
      for(int j=1;j<=c.m;++j){
	for(int k=1;k<=m;++k){
	  c.ma[i][j]+=ma[i][k]*a.ma[k][j];
	}
	(c.ma[i][j]+=mod)%=mod;
      }
    }
    return c;
  }
};
matrix qp(matrix a,ll b){
  if(b==1)return a;if(b==2)return a*a;
  matrix tmp=qp(a,(b>>1));
  tmp=tmp*tmp;
  if(b&1)tmp=tmp*a;
  return tmp;
}
void dfs(int pos,int num,int cnt){
  if(cnt==k){que[++que[0]]=num;return;}
  for(int i=pos-1;i!=-1;i--)
    dfs(i,num+(1<<i),cnt+1);
}
#define lowbit(o) ( (o) & (-o) )
bool check(int x,int y){
  int Temp=x-(1<<(p-1));
  Temp<<=1;Temp^=y;
  if(Temp==lowbit(Temp))return 1;
  return 0;
}
int main(){
  scanf("%d%d%d",&n,&k,&p);
  dfs(p-1,(1<<(p-1)),1);
  matrix temp;temp.clear();temp.n=temp.m=que[0];
  matrix ans;ans.clear();ans.ma[1][1]=1;ans.n=que[0],ans.m=1;
  for(int i=1;i<=que[0];++i){
    for(int j=1;j<=que[0];++j){
      if(check(que[i],que[j]))temp.ma[j][i]=1;
    }
  }
  temp=qp(temp,n-k);
  ans=temp*ans;
  cout<<ans.ma[1][1];
  return 0;
}

 

BZOJ2004: [Hnoi2010]Bus 公交线路

标签:include   简单   tmp   register   clu   zoj   ons   bsp   排列   

原文地址:http://www.cnblogs.com/zzmmm/p/7593834.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!