bzoj3198 spring hash&容斥原理

       首先可以用容斥,这样就只用统计制定的i列相同的行的对数了(而不用恰好)。

       那么可以把这几行对应的列抽出来,然后用hash值(不妨自然溢出)来统计,可以得到hash值相同的行的个数,然后一边更新一边统计答案即可。注意hash值很大需要用hash统计(此hash非彼hash)。

       时间复杂度O(2^6*N)。

AC代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#define p 2150527
#define N 100005
#define ll long long
using namespace std;

int n,m,tot,a[N][6],fst[p],len[N],nxt[N],bin[15],vis[p],cbn[15][15]; ll pnt[N];
int read(){
	int x=0; char ch=getchar();
	while (ch<'0' || ch>'9') ch=getchar();
	while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
	return x;
}
ll solve(int x){
	int i,j,k; ll tmp,ans=0; tot=0;
	for (i=1; i<=n; i++){
		tmp=0;
		for (j=0; j<6; j++) if (x&bin[j]) tmp=tmp*1000003+a[i][j];
		j=tmp%p; if (j<0) j+=p;
		if (vis[j]!=x){ vis[j]=x; fst[j]=0; }
		for (k=fst[j]; k; k=nxt[k])
			if (pnt[k]==tmp){
				ans+=len[k]; len[k]++; break;
			}
		if (!k){
			pnt[++tot]=tmp; len[tot]=1; nxt[tot]=fst[j]; fst[j]=tot;
		}
	}
	return ans;
}
int main(){
	n=read(); m=read(); int i,j; ll ans=0;
	for (i=1; i<=n; i++)
		for (j=0; j<6; j++) a[i][j]=read();
	bin[0]=1; for (i=1; i<=6; i++) bin[i]=bin[i-1]<<1;
	cbn[0][0]=1;
	for (i=1; i<=6; i++)
		for (cbn[i][0]=1,j=1; j<=i; j++) cbn[i][j]=cbn[i-1][j]+cbn[i-1][j-1];
	for (i=0; i<64; i++){
		int cnt=0;
		for (j=0; j<6; j++) if (i&bin[j]) cnt++;
		if (cnt>=m){
			ll tmp=solve(i)*cbn[cnt][m];
			if ((cnt^m)&1) ans-=tmp; else ans+=tmp;
		}
	}
	printf("%lld\n",ans);
	return 0;
}


by lych

2016.3.29

你可能感兴趣的:(hash,容斥原理)