位运算的艺术:POJ2443

题目描述:

给出N个集合,每个集合有M个元素,q次询问,判断两个数字x,y,是否存在一个集合同时包含这两个元素。

大致思路:

首先看到是集合问题,应该会想到并查集,但是这个题显然不能用并查集来做……毕竟不是不相交集合。其次就想到的是用一下高级数据结构,像是set,map之类的,这样时间复杂度是O(nq)的,但是这样显然也是不够的,毕竟Q是10万,N是1000,这样一来不管是什么样的服务器都跑不动1亿的程序。

换一种思路来看,既然不能从集合的角度来记录元素,那么就从元素的角度来记录集合吧。在动态规划里边有一部分叫做状态压缩,想到这里,就离正确做法不远了,过程中一样是利用位运算,一个int类型存成二进制是32位,也就是可以记录这个元素是否在这32个集合中出现,这样的话,复杂度一下子就降到了O(32q),综合一下就是O(q)的复杂度。当然这个复杂度只是询问的过程。

代码:

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <string>
#include <cstring>

using namespace std;

int n,q;
int f[10005][32];

int main() {
	while (cin>>n) {
		memset(f,0,sizeof(f));
		for (int i = 0; i < n; i++) {
			int m;
			scanf("%d",&m);
			for (int j = 0; j < m; j++) {
				int tmp;
				scanf("%d",&tmp);
				f[tmp][i / 32] |= 1 << (i % 32);
			}
		}
		int x,y;
		cin>>q;
		while (q--) {
			scanf("%d%d",&x,&y);
			bool flag = false;
			for (int i = 0; i < 32; i++) {
				if (f[x][i] & f[y][i]) {
					flag = true;
					break;
				}
			}
			if (flag) printf("Yes\n");
			else printf("No\n");
		}
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值