bzoj 3198 [Sdoi2013]spring(容斥原理+Hash)

Description

bzoj 3198 [Sdoi2013]spring(容斥原理+Hash)

Input

bzoj 3198 [Sdoi2013]spring(容斥原理+Hash)

Output

bzoj 3198 [Sdoi2013]spring(容斥原理+Hash)

Sample Input

3 3
1 2 3 4 5 6
1 2 3 0 0 0
0 0 0 4 5 6

Sample Output

2

HINT

bzoj 3198 [Sdoi2013]spring(容斥原理+Hash)

【思路】

容斥原理+Hash

恰有k个元素相同的对数=至少k+1个相同*C(k+1,k) - 至少k+2个相同*C(k+2,k) + ……

枚举状态i,如果是101表示至少1和3两个相同,把n个年份关于i构造一个hash,然后放入hash中统计。这里只是关于位是1的构造hash,其他位都忽略了,所以得到的是至少有多少个相同的数目。

学了个hash表的写法,新姿势get :)

【代码】

 #include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std; typedef unsigned long long ull;
typedef long long ll;
const int N = ;
const int B = ; namespace Hash_set
{
struct node {
node* nxt;
ull H; int v;
node(){}
node(node* _,ull __) :
nxt(_),H(__),v() {}
}*head[N],mempool[N],*C=mempool;
int vis[N],kase;
void init() {
kase++; C=mempool;
}
int& insert(ull st) {
int pos=st%;
if(vis[pos]!=kase) {
vis[pos]=kase; head[pos]=0x0;
}
for(node* p=head[pos];p;p=p->nxt)
if(p->H==st) return p->v;
head[pos]=new (C++) node(head[pos],st);
return head[pos]->v;
}
} int n,K,a[N][],C[][]; ll calc(int st)
{
using namespace Hash_set;
ll ans=;
init();
for(int i=;i<n;i++) {
ull hash=;
for(int j=;j<;j++)
if( st&(<<j) )
hash=(hash*B+a[i][j]);
int& val=insert(hash);
ans+=(val++);
}
return ans;
} void get_C()
{
for(int i=;i<=;i++) {
C[i][]=C[i][i]=;
for(int j=;j<i;j++)
C[i][j]=C[i-][j-]+C[i-][j];
}
} void read(int& x) {
char c=getchar(); int f=; x=;
while(!isdigit(c)){if(c=='-')f=-; c=getchar();}
while(isdigit(c)) x=x*+c-'',c=getchar();
} int main()
{
get_C();
read(n),read(K);
for(int i=;i<n;i++)
for(int j=;j<;j++) read(a[i][j]);
ll ans=;
for(int i=;i<;i++) {
int cnt=;
for(int j=;j<;j++)
if(i&(<<j)) cnt++;
if(cnt>=K) ans+=((cnt-K)&?-:)*calc(i)*C[cnt][K];
}
printf("%lld",ans);
return ;
}
上一篇:C#问题


下一篇:poj 2262【素数表的应用---判断素数】【哈希】