結果

問題 No.1014 competitive fighting
ユーザー beet
提出日時 2020-03-20 22:09:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 633 ms / 2,000 ms
コード長 2,737 bytes
コンパイル時間 3,139 ms
コンパイル使用メモリ 224,784 KB
最終ジャッジ日時 2025-01-09 08:37:10
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}
using Int = long long;
const char newl = '\n';


template<typename V>
V compress(V v){
  sort(v.begin(),v.end());
  v.erase(unique(v.begin(),v.end()),v.end());
  return v;
}
template<typename T>
map<T, Int> dict(const vector<T> &v){
  map<T, Int> res;
  for(Int i=0;i<(Int)v.size();i++)
    res[v[i]]=i;
  return res;
}
map<char, Int> dict(const string &v){
  return dict(vector<char>(v.begin(),v.end()));
}


struct SCC{
  vector< vector<Int> > G,R,T,C;
  vector<Int> vs,used,blg;
  SCC(){}
  SCC(Int n):G(n),R(n),used(n),blg(n){}

  void add_edge(Int u,Int v){
    G[u].emplace_back(v);
    R[v].emplace_back(u);
  }

  void dfs(Int v){
    used[v]=1;
    for(Int u:G[v])
      if(!used[u]) dfs(u);
    vs.emplace_back(v);
  }

  void rdfs(Int v,Int k){
    used[v]=1;
    blg[v]=k;
    C[k].emplace_back(v);
    for(Int u:R[v])
      if(!used[u]) rdfs(u,k);
  }

  Int build(){
    Int n=G.size();
    for(Int v=0;v<n;v++)
      if(!used[v]) dfs(v);

    fill(used.begin(),used.end(),0);
    Int k=0;
    for(Int i=n-1;i>=0;i--){
      if(!used[vs[i]]){
        T.emplace_back();
        C.emplace_back();
        rdfs(vs[i],k++);
      }
    }

    for(Int v=0;v<n;v++)
      for(Int u:G[v])
        if(blg[v]!=blg[u])
          T[blg[v]].push_back(blg[u]);

    for(Int i=0;i<k;i++){
      sort(T[i].begin(),T[i].end());
      T[i].erase(unique(T[i].begin(),T[i].end()),T[i].end());
    }
    return k;
  }

  Int operator[](Int k) const{return blg[k];}
};


template<typename T> void drop(const T &x){cout<<x<<endl;exit(0);}

//INSERT ABOVE HERE
signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  Int n;
  cin>>n;

  vector<Int> as(n),bs(n),cs(n);
  for(Int i=0;i<n;i++) cin>>as[i]>>bs[i]>>cs[i];

  vector<Int> vs(as);
  for(Int i=0;i<n;i++)
    vs.emplace_back(bs[i]-cs[i]);
  vs=compress(vs);
  auto dc=dict(vs);
  Int m=vs.size();

  SCC scc(n+m);
  for(Int i=0;i+1<m;i++)
    scc.add_edge(n+i+1,n+i);
  for(Int i=0;i<n;i++){
    scc.add_edge(n+dc[as[i]],i);
    scc.add_edge(i,n+dc[bs[i]-cs[i]]);
  }

  Int k=scc.build();

  auto blg=scc.blg;
  auto C=scc.C;
  auto T=scc.T;

  vector<Int> cnt(k,0);
  for(Int i=0;i<n;i++) cnt[blg[i]]++;

  const Int INF = 1e18;
  vector<Int> dp(k,0);
  for(Int i=k-1;i>=0;i--){
    if(cnt[i]>1){
      dp[i]=INF;
      continue;
    }
    for(Int j:T[i])
      chmax(dp[i],dp[j]);

    for(Int v:C[i])
      if(v<n) dp[i]+=bs[v];
  }

  vector<Int> ans(n);
  for(Int i=0;i<n;i++) ans[i]=dp[blg[i]];
  for(Int a:ans) if(a>=INF) cout<<"BAN"<<newl; else cout<<a<<newl;
  return 0;
}
0