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]);
}