結果

問題 No.1288 yuki collection
ユーザー NachiaViviasNachiaVivias
提出日時 2020-11-22 20:45:19
言語 C++17(1z)
(gcc 10.2.0 + boost 1.73.0)
結果
AC  
実行時間 387 ms / 5,000 ms
コード長 2,159 Byte
コンパイル時間 2,537 ms
使用メモリ 6,880 KB
最終ジャッジ日時 2020-11-22 20:45:32
合計ジャッジ時間 13,550 ms
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 2 ms
6,872 KB
testcase_01 AC 2 ms
4,828 KB
testcase_02 AC 2 ms
4,828 KB
testcase_03 AC 2 ms
6,876 KB
testcase_04 AC 3 ms
6,876 KB
testcase_05 AC 2 ms
6,876 KB
testcase_06 AC 1 ms
6,872 KB
testcase_07 AC 2 ms
4,824 KB
testcase_08 AC 2 ms
6,872 KB
testcase_09 AC 2 ms
4,828 KB
testcase_10 AC 2 ms
4,828 KB
testcase_11 AC 2 ms
6,876 KB
testcase_12 AC 2 ms
6,872 KB
testcase_13 AC 358 ms
4,824 KB
testcase_14 AC 347 ms
6,876 KB
testcase_15 AC 274 ms
4,828 KB
testcase_16 AC 272 ms
4,832 KB
testcase_17 AC 347 ms
6,876 KB
testcase_18 AC 347 ms
4,824 KB
testcase_19 AC 356 ms
4,824 KB
testcase_20 AC 354 ms
4,828 KB
testcase_21 AC 354 ms
6,872 KB
testcase_22 AC 367 ms
6,872 KB
testcase_23 AC 354 ms
4,824 KB
testcase_24 AC 365 ms
4,828 KB
testcase_25 AC 344 ms
4,828 KB
testcase_26 AC 377 ms
4,828 KB
testcase_27 AC 152 ms
4,824 KB
testcase_28 AC 180 ms
4,828 KB
testcase_29 AC 157 ms
4,828 KB
testcase_30 AC 10 ms
4,828 KB
testcase_31 AC 15 ms
4,832 KB
testcase_32 AC 16 ms
6,876 KB
testcase_33 AC 271 ms
6,872 KB
testcase_34 AC 353 ms
4,828 KB
testcase_35 AC 337 ms
4,828 KB
testcase_36 AC 373 ms
6,876 KB
testcase_37 AC 387 ms
4,828 KB
testcase_38 AC 324 ms
6,872 KB
testcase_39 AC 220 ms
4,828 KB
testcase_40 AC 4 ms
4,828 KB
testcase_41 AC 2 ms
6,876 KB
testcase_42 AC 2 ms
4,824 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
using LL = long long;
using ULL = unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)


struct mcf{
static const int xN = 10000;

template<class T>
using nega_queue = priority_queue<T,vector<T>,greater<T>>;

struct Edge{ int u,v,r; LL c; LL d; };

int N;
int fSrc=0;
int fSnk=1;
vector<Edge> J;
vector<int> E[xN];

LL F;
LL fAns = 0;
LL fD[xN];
LL fPE[xN];
LL fP[xN]={};

void addEdge(int u,int v,LL c,LL d){
  E[u].push_back(J.size());
  J.push_back({u,v,(int)J.size()+1,c,d});
  E[v].push_back(J.size());
  J.push_back({v,u,(int)J.size()-1,0,-d});
}

void fDijkstra(){
  rep(i,N) fD[i]=fPE[i]=-1;
  nega_queue<pair<LL,pair<int,LL>>> Q; Q.push({0,{fSrc,-1}});
  while(Q.size()){
    LL d=Q.top().first;
    int p=Q.top().second.first;
    LL pre=Q.top().second.second;
    Q.pop();
    if(fD[p]!=-1) continue;
    fD[p]=d;
    fPE[p]=pre;
    for(int j:E[p]){
      if(J[j].c==0)continue;
      LL nxd = d+J[j].d+fP[p]-fP[J[j].v];
      if(fD[J[j].v] >= nxd) continue;
      Q.push({nxd,{J[j].v,j}});
    }
  }
  rep(i,N) fP[i]+=fD[i];
}

void flow(){
  fDijkstra();
  if(fD[fSnk]==-1){ F=0; fAns=-1; return; }
  int p=fSnk;
  LL c=F;
  while(p!=fSrc){
    c=min(c,J[fPE[p]].c);
    p=J[fPE[p]].u;
  }
  p=fSnk;
  while(p!=fSrc){
    J[fPE[p]].c-=c;
    J[J[fPE[p]].r].c+=c;
    fAns += (int)c * J[fPE[p]].d;
    p=J[fPE[p]].u;
  }
  F-=c;
}
} G;

int main(){
  int N; cin>>N;
  string S; cin>>S;
  vector<int> S2(N); rep(i,N){ S2[i]=string("yuki").find(S[i]); }
  vector<LL> V(N); rep(i,N) cin>>V[i];

  G.fSrc=N*2; G.fSnk=N*2+1;
  G.N = N*2+2;

  static const LL maxCost = 1000000000;

  int preP[4]; rep(i,4) preP[i]=-1;
  rep(i,N){
    int ch = S2[i];

    G.addEdge(i,i+N,1,maxCost-V[i]);
    if(preP[ch]!=-1) G.addEdge(preP[ch]+N,i+N,1000000,0);
    if(ch!=0) if(preP[ch-1]!=-1) G.addEdge(preP[ch-1]+N,i,1000000,0);
    if(ch==0) G.addEdge(G.fSrc,i,1,0);
    if(ch==3) G.addEdge(i+N,G.fSnk,1,0);
    
    preP[ch]=i;
  }

  LL ans=0;
  for(LL f=1; ; f++){
    G.F=1;
    while(G.F) G.flow();

    if(G.fAns==-1) break;
    ans = max(ans,maxCost*4*f-G.fAns);
  }

  cout<<ans<<endl;

  return 0;
}
0