【yLOI2018】锦鲤抄

题面

https://www.luogu.org/problem/P5008

题解

为了带入氛围还把$QQ$音乐里的《锦鲤抄》点开听了听。

让我们想想删除的顺序是什么样子的：最优的顺序一定是从拓扑序最大的点倒着删，删到拓扑序最小的点。

其中有入度的强连通分量可以全都删完（最后通过“入度”退回拓扑序小的强连通分量）如果最后一个了，就必须留一个点。

贪心即可。

#include<stack>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 500500
#define M 2000500
#define mod 1000000007
#define ri register int

using namespace std;

inline int read() {
  int ret=0; char ch=getchar();
  while (ch<'0' || ch>'9') ch=getchar();
  while (ch>='0' && ch<='9') ret*=10,ret+=(ch-'0'),ch=getchar();
  return ret;
}

stack<int> s; vector<int> to[N];
int dfn[N],low[N],a[N],bel[N],id[N];
int n,m,k,u[M],v[M],sc=0,vc=0,ins[N],siz[N];

void tarjan(int x) {
  dfn[x]=low[x]=++sc; s.push(x); ins[x]=1;
  for (ri i=0;i<to[x].size();i++) {
    int y=to[x][i];
    if (!dfn[y]) {
      tarjan(y);
      low[x]=min(low[x],low[y]);
    }
    else {
      if (ins[y]) low[x]=min(low[x],dfn[y]);
    }
  }
  if (low[x]==dfn[x]) {
    ++vc;
    while (1) {
      int t=s.top(); s.pop(); ins[t]=0;
      bel[t]=vc;
      siz[vc]++;
      if (t==x) break;
    }
  }
}

bool cmp(int x,int y) {
  return a[x]>a[y];
}

int main() {
  n=read(); m=read(); k=read();
  for (ri i=1;i<=n;i++) a[i]=read();
  for (ri i=1;i<=m;i++) {
    u[i]=read(); v[i]=read();
    to[u[i]].push_back(v[i]);
  }
  for (ri i=1;i<=n;i++) if (!dfn[i]) tarjan(i);
  for (ri i=1;i<=m;i++) {
    if (bel[u[i]]!=bel[v[i]]) siz[bel[v[i]]]++;
  }
  for (ri i=1;i<=n;i++) id[i]=i;
  sort(id+1,id+n+1,cmp);
  int ans=0,cnt=0;
  for (ri i=1;i<=n;i++) {
    int cur=id[i];
    if (siz[bel[cur]]>1) {
      siz[bel[cur]]--;
      ans+=a[cur];
      if (++cnt==k) {
        printf("%d
",ans);
        return 0;
      }
    }
  }
  printf("%d
",ans);
  return 0;
}