题目链接:gmoj
分析
上代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
queue<int> q;
const int N=100010;
int n,m,k,top,ans1,tim,tt;
int tot,hd[N];
int u[10*N],v[10*N],w[N],ans[N];
int dfn[N],low[N],du[N],st[N],c[N],sum[N],vis[N];
int fa[N];
int d[N];
struct lwx
{
int to,next;
}e[2000010];
void add(int x,int y)
{
e[++tot]=(lwx){y,hd[x]};
hd[x]=tot;
}
int father(int x)
{
if(fa[x]==x) return x;
else return fa[x]=father(fa[x]);
}
void tarjan(int x)
{
dfn[x]=low[x]=++tim;
st[++top]=x;
for(int i=hd[x];i>0;i=e[i].next)
{
int lv=e[i].to;
if(!dfn[lv])
{
tarjan(lv);
low[x]=min(low[x],low[lv]);
}
else if(!c[lv]) low[x]=min(low[x],low[lv]);
}
if(dfn[x]==low[x])
{
c[x]=++tt;
sum[tt]+=w[x];
while(st[top]!=x)
{
sum[tt]+=w[st[top]];
c[st[top]]=tt;
top--;
}
top--;
}
}
void work()
{
for(int i=1;i<=tt;i++)
{
if(!du[i]) q.push(i);
}
while(!q.empty())
{
int x=q.front();
q.pop();
d[x]+=sum[x];
int fx=father(x);
ans[fx]=max(ans[fx],d[x]);
for(int i=hd[x];i>0;i=e[i].next)
{
int y=e[i].to;
du[y]--;
d[y]=max(d[y],d[x]);
if(du[y]==0) q.push(y);
}
}
}
int main()
{
freopen("azeroth.in","r",stdin);
freopen("azeroth.out","w",stdout);
cin>>n>>m;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&u[i],&v[i]);
add(u[i],v[i]);
}
for(int i=1;i<=n;i++)
{
scanf("%d",&w[i]);
}
for(int i=1;i<=n;i++)
{
if(!dfn[i]) tarjan(i);
}
for(int i=1;i<=tt;i++) fa[i]=i;
memset(e,0,sizeof(e));
memset(hd,0,sizeof(hd));
for(int i=1;i<=m;i++)
{
if(c[u[i]]!=c[v[i]])
{
add(c[u[i]],c[v[i]]);
du[c[v[i]]]++;
int fx=father(c[u[i]]);
int fy=father(c[v[i]]);
if(fx!=fy)
{
fa[fy]=fx;
}
}
}
work();
sort(ans+1,ans+n+1);
cin>>k;
for(int i=n;i>=max(n-k,1);i--)
{
ans1+=ans[i];
}
cout<<ans1;
return 0;
}