P1407 [国家集训队] 稳定婚姻

[国家集训队] 稳定婚姻

题目描述

我们已知 \(n\) 对夫妻的婚姻状况，称第 \(i\) 对夫妻的男方为 \(B_i\)，女方为 \(G_i\)。若某男 \(B_i\) 与某女 \(G_j\) 曾经交往过（无论是大学，高中，亦或是幼儿园阶段，\(i \le j\)），则当某方与其配偶（即 \(B_i\) 与 \(G_i\) 或 \(B_j\) 与 \(G_j\)）感情出现问题时，他们有私奔的可能性。不妨设 \(B_i\) 和其配偶 \(G_i\) 感情不和，于是 \(B_i\) 和 \(G_j\) 旧情复燃，进而 \(B_j\) 因被戴绿帽而感到不爽，联系上了他的初恋情人 \(G_k\) ……一串串的离婚事件像多米诺骨牌一般接踵而至。若在 \(B_i\) 和 \(G_i\) 离婚的前提下，这 \(2n\) 个人最终依然能够结合成 \(n\) 对情侣，那么我们称婚姻 \(i\) 为不安全的，否则婚姻 \(i\) 就是安全的。

给定所需信息，你的任务是判断每对婚姻是否安全。

输入格式

第一行为一个正整数 \(n\)，表示夫妻的对数；

以下 \(n\) 行，每行包含两个字符串，表示这 \(n\) 对夫妻的姓名（先女后男），由一个空格隔开；

第 \(n+2\) 行包含一个正整数 \(m\)，表示曾经相互喜欢过的情侣对数；

以下 \(m\) 行，每行包含两个字符串，表示这 \(m\) 对相互喜欢过的情侣姓名（先女后男），由一个空格隔开。

输出格式

输出文件共包含 \(n\) 行，第 \(i\) 行为 Safe（如果婚姻 \(i\) 是安全的）或 Unsafe（如果婚姻 \(i\) 是不安全的）。

样例 #1

样例输入 #1

2
Melanie Ashley
Scarlett Charles
1
Scarlett Ashley

样例输出 #1

Safe
Safe

样例 #2

样例输入 #2

2
Melanie Ashley
Scarlett Charles
2
Scarlett Ashley
Melanie Charles

样例输出 #2

Unsafe
Unsafe

提示

对于 \(20\%\) 的数据，\(n \le 20\)；

对于 \(40\%\) 的数据，\(n \le 100\)，\(m \le 400\)；

对于 \(100\%\) 的数据，所有姓名字符串中只包含英文大小写字母，大小写敏感，长度不大于 \(8\)，保证每对关系只在输入文件中出现一次，输入文件的最后 \(m\) 行不会出现未在之前出现过的姓名，这 \(2n\) 个人的姓名各不相同，\(1 \le n \le 4000\)，\(0 \le m \le 20000\)。

分析

对于n对夫妻 $ (u,v) $ 和m对前情侣 $ (u',v') $ ，建边 $ u->v $ 和 $ v'->u' $ 。

跑一边Tarjan缩点，如果 $ (u,v) $ 在一个SCC中则不安全。因为此时 $ (u,v) $ 在长度大于3的偶环上，可以使得 $ (u,v') $ 和 $ (v,u') $ 配对。

如图

#include<bits/stdc++.h>
using namespace std;
const int N=2e4+100,M=1e5+100;
int n,m;
int head[N],cnt;
int dfn[N],low[N],tot,dd;
int sta[N],top,id,scc[N];
int pa[N][2];
bool inc[N];
map<string,int>mp;
struct edge{int y,n;}e[M<<1];
string s1,s2;
void ad(int x,int y)
{
    e[++cnt].n=head[x];
    e[cnt].y=y;
    head[x]=cnt;
}

void go(int u)
{
    sta[++top]=u;
    inc[u]=1;
    low[u]=dfn[u]=++tot;
    for(int i=head[u];i;i=e[i].n)
    {
        int v=e[i].y;
        if(!dfn[v])
        {
            go(v);
            low[u]=min(low[u],low[v]);
        }
        else if(inc[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(low[u]==dfn[u])
    {
        ++id;
        while(sta[top]!=u)
        {
            scc[sta[top]]=id;
            inc[sta[top]]=0;
            --top;
        }
        scc[u]=id;
        inc[u]=0;
        --top;
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        cin>>s1>>s2;
        int id1=mp[s1],id2=mp[s2];
        if(!id1)id1=mp[s1]=++dd;
        if(!id2)id2=mp[s2]=++dd;
        pa[i][0]=id1;
        pa[i][1]=id2;
        ad(id1,id2);

    }
    scanf("%d",&m);
    for(int i=1;i<=m;++i)
    {
        cin>>s1>>s2;
        int x=mp[s2],y=mp[s1];
        ad(x,y);
    }
    for(int i=1;i<=dd;++i)
        if(!dfn[i])go(i);
    for(int i=1;i<=n;++i)
    {
        if(scc[pa[i][0]]==scc[pa[i][1]])printf("Unsafe");
        else printf("Safe");
        putchar('\n');
    }
    return 0;
}