【codeforces上紫】图论

cfedu100 E. Plan of Lectures

题意

拓扑排序加强版。n个数，每个数指定一个祖先，这个祖先一定要出现在当前数之前。形成一颗树。然后有k个额外条件，每个条件给出x和y，要求给所有点进行拓扑排序，且x一定刚好在y之前一位。如果这样的拓扑排序不存在，输出0，否则输出排序。

思路

• dsu
• toposorting(bfs)
• constructive algorithm

先找一种可能解，再判断这个解是否成立。若这个解不成立，则排序不存在。
找到一种改图的方式，使之能进行拓扑排序。
把x与y的紧邻关系看作无向边，把他们看做一个连通分量，缩成一个点。
构造一个新的缩点图。如果两个分量中分别存在两个点，他们相连，那么两个分量是连通的。在新的缩点图中进行拓扑排序，能得到一个整体的排序。然后找到缩点内部的排序。
写答案：拓扑排序的顺序遍历第一个图的缩点，对每一个缩点，用第二个图的拓扑排序记录此连通分量内部点的拓扑排序。检查是否每个节点的祖先都在这之前的答案中出现过。这就是一个可能解。

代码

#include<bits/stdc++.h>
using namespace std;
vector<vector<int> >G;
const int maxn = 3e5+5;
int fa[maxn],in[maxn],pos[maxn],nex[maxn];
vector<int> a;  // tree edge
vector<int> ans;
// dsu
int find(int x){
    return fa[x]=fa[x]==x?x:find(fa[x]);
}

void unite(int x,int y){
    x = find(x),y = find(y);
    if(x==y) return;
    fa[y] = x;
}

void bfs(int x){
    queue<int> q;
    q.push(x);
    ans.push_back(x);
    while(!q.empty()){
        int u = q.front();q.pop();
        for(auto v:G[u]){
            if(!(--in[v])){
                q.push(v);
                ans.push_back(v);
            }
        }
    }
}

int main(){
    int n,m;scanf("%d%d",&n,&m);
    int rt = 0;
    G.resize(n+1);
    for(int i=1;i<=n;++i){
        int x;scanf("%d",&x);
        fa[i] = i;
        if(x==0) rt = i;
        a.push_back(x);
    }
    for(int i=1;i<=m;++i){
        int x,y;scanf("%d%d",&x,&y);
        nex[x] = y;
        unite(x,y);
    }
    for(int i=1;i<=n;++i){
        int x = find(a[i-1]),y = find(i);
        if(x==y) continue;
        G[x].push_back(y);
        in[y]++;
    }
    int cnt = 0; // number of component
    for(int i=1;i<=n;++i) if(i==find(i)) cnt++;
    rt = find(rt);
    bfs(rt);
    vector<int> res;
    for(auto u:ans){
        for(int j=u;j;j=nex[j]){
            res.push_back(j);
            if(res.size()>n) {puts("0");return 0;} // 缩点内部有环
        }
    }
    for(int i=0;i<res.size();++i) pos[res[i]] = i;
    for(int i=1;i<=n;++i) {
        int x = a[i-1],y = i;
        if(x==0) continue;
        if(pos[x] >= pos[y]){
            puts("0");return 0;  
            // 出现不符合第二种关系的排序！（缩点外部有环，导致bfs提前终止。导致有些缩点内部pos一样）
        }
    }
    for(auto i:res) cout<<i<<" ";
    cout<<endl;
}