洛谷 P1407 [国家集训队]稳定婚姻 题解

题目：P1407 稳定婚姻

Tarjan - 强连通分量

此意利用一种拆点的思想，把一对夫妻想象为一个点，女方负责接收入边，男方负责发出出边。当然夫妻之间也要连一条 $girl$ $-->$ $boy$的边
对于情侣关系，我们连接两人所在的点，也就是$boy-->girl$

最后，若一对夫妻在同一个强连通分量内，则婚姻是不安全的，反之亦然

#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<algorithm>
#include<stack>
using namespace std;
const int Maxn=8020,inf=0x3f3f3f3f;
string a[Maxn>>1],b[Maxn>>1];
map <string , int> vis;
vector <int> e[Maxn];
stack <int > s;
int dfn[Maxn],low[Maxn],f[Maxn];
bool flag[Maxn];
int n,m,idcnt,timecnt;
inline int read()
{
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
inline int id(string s)
{
	if(vis[s]==0)vis[s]=++idcnt;
	return vis[s];
}
void tarjan(int x)
{
	dfn[x]=low[x]=++timecnt;
	s.push(x),flag[x]=1;
	for(int i=0;i<e[x].size();++i)
	{
		int y=e[x][i];
		if(!dfn[y])
		{
			tarjan(y);
			low[x]=min(low[x],low[y]);
		}
		else if(flag[y])low[x]=min(low[x],dfn[y]);
	}
	if(dfn[x]==low[x])
	{
		while(s.size())
		{
			int y=s.top();
			s.pop();
			f[y]=x,flag[y]=0;
			if(x==y)break;
		}
	}
}
int main()
{
//	freopen("in.txt","r",stdin);
	n=read();
	for(int i=1;i<=n;++i)
	{
		cin>>a[i]>>b[i];
		int x=id(a[i]),y=id(b[i]);
		e[x].push_back(y);
	}
	
	m=read();
	for(int i=1;i<=m;++i)
	{
		string u,v;
		cin>>u>>v;
		int x=id(u),y=id(v);
		e[y].push_back(x);
	}
	
	for(int i=1;i<=idcnt;++i)
	if(!dfn[i])tarjan(i);
	
	for(int i=1;i<=n;++i)
	{
		int x=id(a[i]),y=id(b[i]);
		
		if(f[x]==f[y])puts("Unsafe");
		else puts("Safe");
	}
	return 0;
}