先上链接:点击打开链接
昨天上课偶然发现洛谷上也有题没有过,寒假有限几天写过题的杰作,当时只知道tarjan很厉害,什么求强联通分量怎么做完全不造。当然离上一次看强联通、点连通什么的过了快两周了,其实是有点忘记了,不过怎么写看着lrj大致还是知道的。
题意:每个牛棚有个后继牛棚,每走一个牛棚可以加1,然后求出从每个牛棚出发可以得到多少的值。
当时寒假只会暴力dfs,好吧我连dfs都是抠脚的。反正也不知道什么tarjan直接爆搜40分,然后知道要tarjan就没管它,我这个人怎么这么不负责任呐2333。
这题可以这样看,首先样例还是良心的,有自环,结合做图论题的经验可知这是环加链的模型,自然tarjan缩点嘛(不知道什么是缩点的自己看蓝书)。然后这里next也就是平时的G[i][j]只有一个,也就不用邻接表了,建完图跑完一遍tarjan然后把每个强联通分量用next边连起来,再在这个图上面记忆化搜索就好了。我想也就我这样代码能力极其羸弱的要写40多分钟。这两天想复习了就再更一下tarjan的几个算法。既然tarjan忘了,lca肯定。。。。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<bitset>
#include<vector>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
const int maxn=1e5+7;
int next[maxn],pre[maxn],low[maxn],sccno[maxn],time,cnt,sta[maxn],tot,vis[maxn];
int ans[maxn],sz[maxn],out[maxn];
void dfs(int k)
{
pre[k]=low[k]=++time;
sta[++tot]=k,vis[k]=1;
int v=next[k];
if(!pre[v])
{
dfs(v);
low[k]=min(low[k],low[v]);
}
else if(vis[v]){
low[k]=min(low[k],pre[v]);
}
if(pre[k]==low[k])
{
cnt++;
for(;;)
{
int q=sta[tot--];
sccno[q]=cnt;
vis[q]=0;
sz[cnt]++;
if(q==k) break;
}
}
}
int dfs1(int k)
{
if(!k||ans[k]) return ans[k];
return ans[k]=dfs1(out[k])+sz[k];
}
int main()
{
int n,m;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&next[i]);
}
for(int i=1;i<=n;i++) if(!pre[i])dfs(i);
for(int i=1;i<=n;i++) if(sccno[i]!=sccno[next[i]])out[sccno[i]]=sccno[next[i]];
for(int i=1;i<=n;i++) dfs1(sccno[i]);
for(int i=1;i<=n;i++) printf("%d\n",ans[sccno[i]]);
}