A1258. 稳定婚姻(魏铭) tarjan

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

const int N=32005;
int head
,next
,list
,dfn
,low
,q
,num
,ret
,belong
;
bool inset
;
int n,m,cnt,top,tot,scc,yzy;
int ch
[52];
char s[10];

inline void insert(int x,int y)
{
next[++cnt]=head[x];
head[x]=cnt;
list[cnt]=y;
}

inline int get(char c)
{
if (c>='a'&&c<='z') return c-'a';
return c-'A'+26;
}

inline int add()
{
int len=strlen(s),c,now=0;
for (int i=0;i<len;i++)
{
c=get(s[i]);
if (ch[now][c]) now=ch[now][c];
else now=ch[now][c]=++yzy;
}
return now;
}

void dfs(int x)
{
dfn[x]=low[x]=++tot;
inset[x]=true;
q[++top]=x;
for (int i=head[x];i;i=next[i])
if (!dfn[list[i]])
{
dfs(list[i]);
low[x]=min(low[x],low[list[i]]);
}
else if (inset[list[i]]) low[x]=min(low[x],dfn[list[i]]);
if (dfn[x]==low[x])
{
int i=-1; scc++;
while (i!=x)
{
i=q[top--];
inset[i]=false;
num[scc]++;
ret[i]=scc;
}
}
}

int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
scanf("%s",s);
belong[add()]=i;
scanf("%s",s);
belong[add()]=i;
}
scanf("%d",&m);
for (int i=1;i<=m;i++)
{
scanf("%s",s);
int u=belong[add()];
scanf("%s",s);
int v=belong[add()];
insert(u,v);
}
for (int i=1;i<=n;i++)
if (!dfn[i]) dfs(i);
for (int i=1;i<=n;i++)
if (num[ret[i]]>1) puts("Unsafe"); else puts("Safe");
return 0;
}