码迷,mamicode.com
首页 > 其他好文 > 详细

【yLOI2018】锦鲤抄

时间:2019-08-30 23:28:23      阅读:125      评论:0      收藏:0      [点我收藏+]

标签:har   read   namespace   stack   code   i++   class   reg   fine   

题面

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\n",ans);
        return 0;
      }
    }
  }
  printf("%d\n",ans);
  return 0;
}

 

【yLOI2018】锦鲤抄

标签:har   read   namespace   stack   code   i++   class   reg   fine   

原文地址:https://www.cnblogs.com/shxnb666/p/11437503.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!