【BZOJ】3495: PA2010 Riddle

题意

\(n(1 \le n \le 1000000)\)个城市，\(k(1 \le k \le n)\)个国家，\(m(1 \le m \le 1000000)\)条边。要求每个国家有且仅有一个首都，每条边两端的城市至少要有一个首都。判断是否有解。

分析

满足性问题。而且每个城市只有两种情况，首都or不是首都。所以考虑2-sat

题解

对于每一个点，拆点为\(i\)和\(i'\)，表示有首都和无首都。
对于每一个国家（假设有\(a\)个城市），由于只有一个首都，也就是说，假设这个国家的第\(j\)城市是首都，则前\(j-1\)个城市和后\(a-j\)个城市都不是首都！对应着前缀和和后缀和为0！
所以我们对每个国家建立前缀和和后缀和的结点，由于只有两种情况，0和1，所以照样用2-sat可以解决。
至于怎么连边，自己yy一下，很简单的。

#include <bits/stdc++.h>
using namespace std;
const int N=1000005*6, M=1000005*12;
int ihead[N], cnt, FF[N], LL[N], p[N], scc, tot;
struct E {
	int next, to;
}e[M];
void add(int x, int y) {
	e[++cnt]=(E){ihead[x], y}; ihead[x]=cnt;
}
void dfs(int x) {
	static int tid=0, s[N], vis[N], top=0;
	s[++top]=x;
	vis[x]=1;
	FF[x]=LL[x]=++tid;
	for(int i=ihead[x]; i; i=e[i].next) {
		int y=e[i].to;
		if(!FF[y]) {
			dfs(y);
			LL[x]=min(LL[x], LL[y]);
		}
		else if(vis[y]) {
			LL[x]=min(LL[x], FF[y]);
		}
	}
	if(FF[x]==LL[x]) {
		++scc;
		int y;
		do {
			y=s[top--];
			vis[y]=0;
			p[y]=scc;
		} while(x!=y);
	}
}
bool check() {
	for(int i=1, all=tot<<1|1; i<=all; ++i) {
		if(!FF[i]) {
			dfs(i);
		}
	}
	for(int i=1; i<=tot; ++i) {
		if(p[i<<1]==p[i<<1|1]) {
			return 0;
		}
	}
	return 1;
}
int main() {
	int n, m, k;
	scanf("%d%d%d", &n, &m, &k);
	for(int i=1; i<=m; ++i) {
		int x, y;
		scanf("%d%d", &x, &y);
		add(x<<1|1, y<<1);
		add(y<<1|1, x<<1);
	}
	tot=n;
	for(int j=1; j<=k; ++j) {
		scanf("%d", &m);
		for(int i=1; i<=m; ++i) {
			int a;
			scanf("%d", &a);
			int now=tot+i;
			if(i!=1) {
				add(now<<1|1, (now-1)<<1|1);
				add((now-1)<<1, now<<1);
				add(a<<1, (now-1)<<1|1);
			}
			else {
				add(a<<1|1, now<<1|1);
			}
			add(now<<1|1, a<<1|1);
			add(a<<1, now<<1);

			now=tot+m+i;
			if(i!=m) {
				add(now<<1|1, (now+1)<<1|1);
				add((now+1)<<1, now<<1);
				add(a<<1, (now+1)<<1|1);
			}
			else {
				add(a<<1|1, now<<1|1);
			}
			add(now<<1|1, a<<1|1);
			add(a<<1, now<<1);
		}
		tot+=2*m;
	}
	puts(check()?"TAK":"NIE");
	return 0;
}