首页 > 代码库 > 【poj3875】 Lights

【poj3875】 Lights

http://poj.org/problem?id=3875 (题目链接)

题意

  有M个N位的不同的二进制数,他们异或起来前v位等于1,求这m个数的不同组合方式(同一组数不同顺序不算)。

Solution

  如果任意两个数可以相同就非常好做了,然而。。事实总是那么令人悲伤。

  考虑dp。假设最后异或出来的数为${V}$,令${f_m}$表示m个数的不同方案(这里我们先算不同顺序算不同方案的,更好处理一些)。那么如果我们先选出其中${m-1}$个数,它的方案就是:

$${(m-1)!×C_{2^n}^{m-1}}$$

  这很好理解,一共有{2^n}个数可供选择,从中任意选取${m-1}$个数,令其为${a_1,a_2,a_3······a_{m-1}}$那么剩下的第${a_m}$个数就可以确定了:

$${\because a_1~XOR~a_2~XOR······XOR~a_m=V}$$

$${\therefore a_m=V~XOR~a_1~XOR~a_2~XOR······XOR~a_{m-1}}$$

  然而问题来了,如果${a_m}$求出来与之前已经选择好了的${m-1}$个数中的某一个相等怎么办?

  假设${a_m=a_1}$,因为相同的数异或起来等于0,则有:

$${a_2~XOR~a_3~XOR······XOR~a_{m-1}=V}$$

  这个式子的方案数代表什么,不就是代表${f_{m-2}}$吗。所以我们可以列出dp方程:

$${f_m=(m-1)!×C_{2^n}^{m-1}-(m-1)×(2^n-(m-2))×f_{m-2}}$$

  其中第一项${(m-1)!×C_{2^n}^{m-1}}$很好理解,就是不考虑${a_m}$与之前已经选好的数相同的方案数。

  第二项中${(m-1)}$表示${a_m}$可以与${m-1}$个数中的任意一个相同,${(2^n-(m-2))}$表示${a_m}$的值的选择方案。

  这里还要注意一个地方:${C_{2^n}^{m-1}}$怎么求。考虑Lucas定理求解组合数取模:

$${Lucas(n,m)=C_{n~mod~p}^{m~mod~p}*Lucas(n/p,m/p)}$$

  因为m最大为1000,恒小于模数p,所以${Lucas(n/p,m/p)=1}$,所以:

$${C_{2^n}^{m-1}=C_{n~mod~p}^{m~mod~p}}$$

  $${ans=\frac{f[m]}{m!}}$$

细节

  此时,问题已经得到了解决。什么?你说过不了样例?

  嘿嘿嘿,→_→。

  显然${f[0]=0,f[1]=1}$,那么${f[2]=?}$。

  考虑${v=0}$的情况,因为两个数不能相同,而只有相同的两个数的异或和等于0,所以${f[2]=0}$。

  而当${v>0}$时,${f[2]=C_{2^n}^{1}}$。

  此时,问题已经得到了解决(真的)

代码

// poj3875#include<algorithm>#include<iostream>#include<cstdlib>#include<cstring>#include<cstdio>#include<cmath>#define LL long long#define inf 2147483640#define MOD 10567201#define Pi acos(-1,0)#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);using namespace std;const int maxn=2000;LL f[maxn],fac[maxn],C[maxn];int n,m,v;LL power(LL a,LL b) {	LL res=1;	while (b) {		if (b&1) res=res*a%MOD;		b>>=1;a=a*a%MOD;	}	return res;}int main() {	fac[0]=1;for (int i=1;i<=1000;i++) fac[i]=fac[i-1]*i%MOD;	while (scanf("%d%d%d",&n,&m,&v)!=EOF) {		if (!n && !m && !v) break;		LL tmp=power(2,n);		C[0]=1;		for (int i=1;i<=m;i++) {			int inv=power(i,MOD-2);			C[i]=C[i-1]*(tmp-i+1)%MOD*inv%MOD;		}		f[1]=1;f[0]=0;if (v==0) f[2]=0;else f[2]=tmp;		for (int i=3;i<=m;i++) {			f[i]=(fac[i-1]*C[i-1]%MOD-(i-1)*(tmp-(i-2))%MOD*f[i-2]%MOD+MOD)%MOD;		}		LL inv=power(fac[m],MOD-2);		printf("%lld\n",inv*f[m]%MOD);	}    return 0;}

  

【poj3875】 Lights