链接:https://cn.vjudge.net/problem/HDU-4694
题意:多组样例,求每个点必经点的编号总和。
思路:建出支配树后,dfs一次统计答案即可。注意有的点不一定能到达,也就是图可能不连通,答案初始化为0。
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 5e4+10;
const int M = 1e5+10;
int n,m;
struct Edge
{
int to,nxt;
};
struct Graph
{
int head[N],cnt;
struct Edge G[M];
inline void init()
{
cnt=0;
for(int i=1;i<=n;i++)
head[i]=-1;
return ;
}
inline void add(int u,int v)
{
G[cnt].to=v;
G[cnt].nxt=head[u];
head[u]=cnt++;
return ;
}
};
struct Graph g,rg,ng,trg;
int fa[N],dfn[N],id[N],dfs_clock;
inline void dfs(int u)
{
int v;
dfn[u]=++dfs_clock;
id[dfs_clock]=u;
for(int i=g.head[u];i!=-1;i=g.G[i].nxt)
{
v=g.G[i].to;
if(dfn[v]) continue;
fa[v]=u;
dfs(v);
}
return ;
}
int semi[N],idom[N];
int f[N],mindfn[N];
inline int getf(int x)
{
if(f[x]==x) return x;
int fx=getf(f[x]);
if(dfn[semi[mindfn[f[x]]]]<dfn[semi[mindfn[x]]]) mindfn[x]=mindfn[f[x]];
return f[x]=fx;
}
inline void tarjan()
{
int u,v;
for(int i=dfs_clock;i>1;i--)
{
u=id[i];
for(int j=rg.head[u];j!=-1;j=rg.G[j].nxt)
{
v=rg.G[j].to;
if(!dfn[v]) continue;
getf(v);
if(dfn[semi[mindfn[v]]]<dfn[semi[u]]) semi[u]=semi[mindfn[v]];
}
ng.add(semi[u],u);
u=f[u]=fa[u];
for(int j=ng.head[u];j!=-1;j=ng.G[j].nxt)
{
v=ng.G[j].to;
getf(v);
if(semi[mindfn[v]]==u) idom[v]=u;
else idom[v]=mindfn[v];
}
}
for(int i=2;i<=dfs_clock;i++)
{
u=id[i];
if(semi[u]!=idom[u]) idom[u]=idom[idom[u]];
trg.add(idom[u],u);
}
return ;
}
ll ans[N];
inline void Init()
{
dfs_clock=0;
for(int i=1;i<=n;i++)
dfn[i]=ans[i]=0,semi[i]=idom[i]=f[i]=mindfn[i]=i;
g.init(); rg.init(); ng.init(); trg.init();
return ;
}
inline void dfs1(int u)
{
int v;
for(int i=trg.head[u];i!=-1;i=trg.G[i].nxt)
{
v=trg.G[i].to;
ans[v]=ans[u]+(ll)v;
dfs1(v);
}
return ;
}
int main(void)
{
int u,v;
while(~scanf("%d%d",&n,&m))
{
Init();
while(m--)
{
scanf("%d%d",&u,&v);
g.add(u,v);
rg.add(v,u);
}
dfs(n);
tarjan();
ans[n]=n;
dfs1(n);
for(int i=1;i<=n;i++)
printf("%lld%c",ans[i],i==n?'\n':' ');
}
return 0;
}