bzoj 1231 混乱的奶牛
Description
混乱的奶牛 [Don Piele, 2007] Farmer John的N(4 <= N <= 16)头奶牛中的每一头都有一个唯一的编号S_i (1 <= S_i <= 25,000). 奶牛为她们的编号感到骄傲, 所以每一头奶牛都把她的编号刻在一个金牌上, 并且把金牌挂在她们宽大的脖子上. 奶牛们对在挤奶的时候被排成一支"混乱"的队伍非常反感. 如果一个队伍里任意两头相邻的奶牛的编号相差超过K (1 <= K <= 3400), 它就被称为是混乱的. 比如说,当N = 6, K = 1时, 1, 3, 5, 2, 6, 4 就是一支"混乱"的队伍, 但是 1, 3, 6, 5, 2, 4 不是(因为5和6只相差1). 那么, 有多少种能够使奶牛排成"混乱"的队伍的方案呢?
Input
* 第 1 行: 用空格隔开的两个整数N和K
* 第 2..N+1 行: 第i+1行包含了一个用来表示第i头奶牛的编号的整数: S_i
Output
第 1 行: 只有一个整数, 表示有多少种能够使奶牛排成"混乱"的队伍的方案. 答案保证是 一个在64位范围内的整数.
Sample Input
4 1
3
4
2
1
3
4
2
1
Sample Output
2
输出解释:
两种方法分别是:
3 1 4 2
2 4 1 3
输出解释:
两种方法分别是:
3 1 4 2
2 4 1 3
思路: 状压dp,dp[j][i]表示当前已经被选的数的情况是j,最后一个被选的数是i。
1 #include<bits/stdc++.h> 2 using namespace std; 3 #define re register 4 #define R re int 5 #define rep(i,a,b) for(R i=a;i<=b;i++) 6 #define Rep(i,a,b) for(R i=a;i>=b;i--) 7 #define ms(i,a) memset(a,i,sizeof(a)) 8 #define LL long long 9 int const N=16; 10 LL dp[1<<N][N]; 11 int a[N],n,k; 12 int main(){ 13 scanf("%d%d",&n,&k); 14 rep(i,0,n-1) scanf("%d",&a[i]); 15 rep(i,0,n-1) dp[1<<i][i]=1; 16 rep(j,0,(1<<n)-1) rep(i,0,n-1) if(dp[j][i]){ 17 rep(p,0,n-1) if(!(j&(1<<p)) && abs(a[i]-a[p])>k) 18 dp[j|(1<<p)][p]+=dp[j][i]; 19 } 20 LL ans=0; 21 rep(i,0,n-1) ans+=dp[(1<<n)-1][i]; 22 printf("%lld\n",ans); 23 return 0; 24 }