把博客园图标替换成自己的图标
把博客园图标替换成自己的图标end

qzezoj 1545 疫病救援

题面传送门
话说这不是一张真正的图啊,只是一根链表后面挂着一个可有可无的环儿。
一眼扫去,这道题如果没有环,那么就是记忆化搜索或\(topo\),但有环,就不好办。
根据亡羊补牢原则,我们把这个环拆了不就好了。
接下来就可以缩点。即把一个环变成一个权值为这个环的权值之和的点。
我们可以把那些绕在环上的链拆了,然后对环跑dfs求出权值和。
然后就是正常\(topo\)
代码实现:

#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
int n,m,f[100039],tot,now,flag,a[100039],s[100039],ans[100039],in[100039];
queue<int> q;
vector<int> fs[100039];
inline void dfs(int x) {
    if(x==now&&tot) return;
    tot++;
    dfs(a[x]);
    ans[x]=tot;
}
int main() {
    register int i;
    scanf("%d",&n);
    for(i=1; i<=n; i++) {
        scanf("%d",&a[i]);
        fs[a[i]].push_back(i);
        in[a[i]]++;
    }
    for(i=1;i<=n;i++) if(!in[i]) q.push(i);
    while(!q.empty()){
        now=q.front();
        q.pop();
        in[a[now]]--;
        if(!in[a[now]]) q.push(a[now]);
    }
    for(i=1;i<=n;i++) if(in[i]) s[i]=1;
    for(i=1; i<=n; i++) {
        tot=flag=0;
        if(s[i]&&!ans[i]) now=i,dfs(i);
    }
    //for(i=1;i<=n;i++) printf("%d\n",s[i]);
    for(i=1; i<=n; i++) {
        if(s[a[i]]&&a[i]!=i&&!s[i]) q.push(i),ans[i]=ans[a[i]]+1;
    }
    while(!q.empty()) {
        now=q.front();
        q.pop();
        for(i=0; i<fs[now].size(); i++) {
            if(!ans[fs[now][i]]) ans[fs[now][i]]=ans[now]+1,q.push(fs[now][i]);
        }
    }
    for(i=1; i<=n; i++) printf("%d\n",ans[i]);
}
posted @ 2020-03-26 13:32  275307894a  阅读(54)  评论(0编辑  收藏  举报
浏览器标题切换
浏览器标题切换end