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

最新推荐文章于 2021-02-28 21:13:39 发布

原创最新推荐文章于 2021-02-28 21:13:39 发布 · 352 阅读

1 ·

CC 4.0 BY-SA版权

容斥原理同时被 2 个专栏收录

9 篇文章

订阅专栏

Hash

6 篇文章

订阅专栏

探讨了在给定条件下，如何使用hash容斥原理解决6维坐标中点对匹配问题，通过双hash技巧和广义容斥原理，实现高效算法。

题目描述

给出 $n$ 个 $6$ 维坐标,求有多少对点对满足恰好 $m$ 个位置相等
$1<=n<=10^5$
$0 < = k < = 6$
坐标数值在 $2^{30}$ 以内

题目分析

这道题一看就是hash容斥原理,用 $m$ 个位置对应相等 $- (m + 1)$ 个位置对应相等 $+ (m + 2)$ 个位置对应相等的…
但是不能简简单单直接 $+ / -$ ,根据广义容斥,还要乘上容斥系数 $C_{k}^{m}$
双 $H a s h$ ,过程中遇到 $H a s h 1$ 相同但 $H a s h 2$ 不同的就往后平移,用数组存一下 $H a s h 1$ 为 $k$ 时的 $H a s h 2$ 值与 $C n t$ 值
注意此处 $M o d$ 数要大于 $n$

考试时没用双Hash,想到了做法,奈何代码太丑,这题爆0了…

AC code

#include <bits/stdc++.h>
using namespace std;
#define LL long long
const int MAXN = 100005;
const int P1 = 137, Mod1 = 9999997;
const int P2 = 167, Mod2 = 7394895;
int num[MAXN][6], c[7][7], kase, n, m;
struct MyHash
{
	LL y; int flag, cnt; //y存的是Hash1为当前下标i时的Hash2值
	//flag是用int打标记,就不用每次清零了
	bool Exist() { return flag == kase; }
}h[Mod1];
LL Ans;
void init()
{
	for(int i = 0; i < 7; ++i)
	{
		c[i][0] = c[i][i] = 1;
		for(int j = 1; j < i; ++j)
			c[i][j] = c[i-1][j-1] + c[i-1][j];
	}
}

inline void MyUnique(LL &x, LL o)
{
	while(h[x].Exist() && h[x].y != o) (++x) %= Mod1;
}

bool used[7];
void dfs(int pos, int tot)//枚举当前是求哪几个位置
{
	if(pos == 6)
	{
		if(tot < m) return; //小于m的不用处理
		LL sum = 0; ++kase;
		for(int i = 1; i <= n; ++i)
		{
			LL hsh1 = 0, hsh2 = 0;
			for(int j = 0; j < 6; ++j) if(used[j])
				hsh1 = (hsh1 * P1 % Mod1 + num[i][j]) % Mod1,
				hsh2 = (hsh2 * P2 % Mod2 + num[i][j]) % Mod2;
			MyUnique(hsh1, hsh2);
			if(h[hsh1].flag < kase)
				h[hsh1].cnt = 0, h[hsh1].flag = kase;
			h[hsh1].y = hsh2, sum += (h[hsh1].cnt++);
		}
		Ans += sum * (((tot-m)&1) ? -1 : 1) * c[tot][m]; //容斥
		return;
	}
	used[pos] = 1;
	dfs(pos+1, tot+1);
	used[pos] = 0;
	dfs(pos+1, tot);
}

int main ()
{
	scanf("%d%d", &n, &m); init();
	for(int i = 1; i <= n; ++i)
		for(int j = 0; j < 6; ++j)
			scanf("%d", &num[i][j]);
	dfs(0, 0);
	printf("%lld\n", Ans);
}