bzoj 3495: PA2010 Riddle（2-SAT）

3495: PA2010 Riddle

Time Limit: 30 Sec   Memory Limit: 512 MB
Submit: 327   Solved: 115
[ Submit][ Status][ Discuss]

Description

k个国家，几个城市，m条边。
要求每个国家有且仅有一个首都，每条边两端的城市至少要有一个首都。
判断是否有解, 有解输出“TAK”，无解输出"NIE"
1  < =  k, N ,M ， < =1000000。

Input

Output

Sample Input

6 5 2

1 2

3 1

1 4

5 2

6 2

3 3 4 2

3 1 6 5

Sample Output

TAK

原题：http://main.edu.pl/en/archive/pa/2010/zag

输入部分缺了，第一行输入的三个数分别是n, m, k，之后m条边，再之后每行代表每个国家有多少个城市，然后跟着若干个城市编号，题目中也少了一句话：每个城市都一定属于一个国家

这题就是判断2-SAT是否有解，将每个点x拆成两个点x和x'，分别表示x点是不是首都

对于一条边(u, v)，u'和v连一条边，v'和u连一条边

对于每个国家，如果有k个城市，就再新建2k个点(q1, q1')(q2, q2')…，分别表示该国首都是否在前qi(q<=k)个城市中，即前缀

这样对于一个属于该国家的点x，连接以下六条边（x是前缀q中的最后一个点）

①x到pre(q')  ---  x点若是首都，那么x前面所有点都不能是首都；

②q'到x'；③x到q

④pre(q)到x'  ---  如果x前面已经有首都了，那么x点不能是首都

⑤pre(q)到q；⑥q'到pre(q')

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stack>
#include<vector>
using namespace std;
vector<int> G[4000005];
stack<int> st;
int t, num, time[4000005], low[4000005], vis[4000005], fa[4000005];
int Read()
{
	int x = 0, f = 1;
	char ch;
	ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch=='-')  f = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
		x = x*10+ch-'0', ch = getchar();
	return x*f;
}
void tarjan(int u)
{
	int v, i;
	vis[u] = 1;
	st.push(u);
	time[u] = low[u] = ++t;
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i];
		if(time[v]==0)
		{
			tarjan(v);
			low[u] = min(low[u], low[v]);
		}
		else if(vis[v])
			low[u] = min(low[u], time[v]);
	}
	if(time[u]==low[u])
	{
		num++;
		while(1)
		{  
			v = st.top();
			st.pop();
			vis[v] = 0;
			fa[v] = num;
			if(v==u)
				break;
		}
	}
}
int main(void)  
{
	int i, j, ans, n, m, k, x, y, temp;
	scanf("%d%d%d", &n, &m, &k);
	temp = n;
	for(i=1;i<=m;i++)
	{
		x = Read(), y = Read();
		G[x+n].push_back(y);
		G[y+n].push_back(x);
	}
	n *= 2;
	for(i=1;i<=k;i++)
	{
		x = Read();
		for(j=1;j<=x;j++)
		{
			y = Read();
			G[y].push_back(n+1);
			G[n+2].push_back(y+temp);
			if(j!=1)
			{
				G[n+2].push_back(n);
				G[y].push_back(n);
				G[n-1].push_back(n+1);
				G[n-1].push_back(y+temp);
			}
			n += 2;
		}
	}
	for(i=1;i<=n;i++)
	{
		if(time[i]==0)
			tarjan(i);
	}
	ans = 1;
	for(i=1;i<=temp;i++)
	{
        if(fa[i]==fa[i+temp])
			ans = 0;
		if(ans==0)
			break;
	}
	for(i=temp*2+1;i<=n-1;i+=2)
	{
		if(fa[i]==fa[i+1])
			ans = 0;
		if(ans==0)
			break;
	}
	if(ans)
		printf("TAK\n");
	else
		printf("NIE\n");
	return 0;
}
/*
3 2 1
1 2
2 3
3 1 2 3
*/