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

P1407

对于数据读入的处理就是用map将字符串映射为一个数值即可。

我们称n对为夫妇， m对为情人， 如果把这n+m对关系都用边相连，我们可以得到很多个环，在环中如果一对夫妇离婚，那么两个人都可以和环中的情人分别再进行配对，所以我们可以用tarjan来判断一对夫妇是否都在同一个环中，如果在那么这对婚姻就不安全，反之则安全。

tarjan找强联通的对象是有向图，我们可以这样处理：在夫妇中女向男连边， 情人中男向女连边，这样能保证图中男女交替相连。

剩下的就是tarjan模板了。

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
int n, m, t, tot, idx;
int dfn[N], low[N], vis[N], bel[N];
int head[N], to[N << 1], nxt[N << 1];
stack<int> s;
map<string, int> mp;
void add(int a, int b) {
    nxt[++ tot] = head[a]; head[a] = tot; to[tot] = b;
}
void tarjan(int x) {
    dfn[x] = low[x] = ++ t;
    s.push(x), vis[x] = 1;
    for (int i = head[x]; i; i = nxt[i]) {
        int y = to[i];
        if (!dfn[y]) {
            tarjan(y);
            low[x] = min(low[x], low[y]);
        }
        else if (vis[y]) 
            low[x] = min(low[x], dfn[y]);
    }
    if (dfn[x] == low[x]) {
        idx ++;
        int v;
        do {
            v = s.top();
            s.pop();
            vis[v] = 0;
            bel[v] = idx;
        } while (v != x);
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    cin >> n; string a, b;
    for (int i = 1; i <= n; i ++) {
        cin >> a >> b;
        mp[a] = i, mp[b] = i + n;
        add(i, i + n);
    }
    cin >> m;
    for (int i = 1; i <= m; i ++) {
        cin >> a >> b;
        add(mp[b], mp[a]);
    }
    m = n << 1;
    for (int i = 1; i <= m; i ++)
        if (!dfn[i]) tarjan(i); 
    for (int i = 1; i <= n; i ++)
        if (bel[i] == bel[i + n]) printf("Unsafe\n");
        else printf("Safe\n");
    return 0;
}