bzoj千题计划161:bzoj1589: [Usaco2008 Dec]Trick or Treat on the Farm 采集糖果

http://www.lydsy.com/JudgeOnline/problem.php?id=1589

tarjan缩环后拓扑排序上DP

#include<cstdio>
#include<iostream>
#include<algorithm> #define N 100001 using namespace std; int to[N]; int m;
int bl[N],siz[N]; int st[N],top;
int id,dfn[N],low[N];
bool vis[N]; int FRONT[N],TO[N],NXT[N],TOT;
int in[N]; int dp[N]; void read(int &x)
{
x=; char c=getchar();
while(!isdigit(c)) c=getchar();
while(isdigit(c)) { x=x*+c-''; c=getchar(); }
} void tarjan(int x)
{
dfn[x]=low[x]=++id;
vis[x]=true;
st[++top]=x;
if(!dfn[to[x]])
{
tarjan(to[x]);
low[x]=min(low[x],low[to[x]]);
}
else if(vis[to[x]]) low[x]=min(low[x],dfn[to[x]]);
if(dfn[x]==low[x])
{
++m;
while(st[top]!=x)
{
vis[st[top]]=false;
bl[st[top]]=m;
siz[m]++;
top--;
}
vis[x]=false;
bl[x]=m;
siz[m]++;
top--;
}
} void add(int u,int v)
{
TO[++TOT]=v; NXT[TOT]=FRONT[u]; FRONT[u]=TOT;
in[v]++;
} void topsort()
{
top=;
for(int i=;i<=m;++i)
{
if(!in[i]) st[++top]=i;
dp[i]=siz[i];
}
int now,t;
while(top)
{
now=st[top--];
for(int i=FRONT[now];i;i=NXT[i])
{
t=TO[i];
dp[t]+=dp[now];
in[t]--;
if(!in[t]) st[++top]=t;
}
}
} int main()
{
int n;
read(n);
for(int i=;i<=n;++i) read(to[i]);
for(int i=;i<=n;++i)
if(!dfn[i]) tarjan(i);
for(int i=;i<=n;++i)
if(bl[i]!=bl[to[i]]) add(bl[to[i]],bl[i]);
topsort();
for(int i=;i<=n;++i) cout<<dp[bl[i]]<<'\n';
}
上一篇:第91讲:Akka第一个案例动手实战架构设计


下一篇:【MySQL】MySQL索引背后的之使用策略及优化【转】