結果

問題 No.1288 yuki collection
ユーザー lorent_kyopro
提出日時 2020-11-14 01:06:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,313 bytes
コンパイル時間 3,758 ms
コンパイル使用メモリ 217,376 KB
最終ジャッジ日時 2025-01-16 00:18:44
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/mincostflow>
using namespace std;
using namespace atcoder;

__attribute__((constructor))
void fast_io() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
}

int main() {
    int n;
    string s;
    cin >> n >> s;
    vector<int> v(n);
    for (auto& vi : v) cin >> vi;

    mcf_graph<int, long long> g(n + 2);
    int a = n, b = n + 1;
    int Y = -1, U = -1, K = -1, I = -1;
    for (int i = n-1; i >= 0; --i) {
        if (s[i] == 'i') {
            g.add_edge(i, b, 1, INT_MAX - v[i]);
            if (I != -1) g.add_edge(i, I, INT_MAX, 0);
            I = i;
        } else if (s[i] == 'k') {
            if (K != -1) g.add_edge(i, K, INT_MAX, 0);
            if (I != -1) g.add_edge(i, I, 1, INT_MAX - v[i]);
            K = i;
        } else if (s[i] == 'u') {
            if (U != -1) g.add_edge(i, U, INT_MAX, 0);
            if (K != -1) g.add_edge(i, K, 1, INT_MAX - v[i]);
            U = i;
        } else if (s[i] == 'y') {
            if (Y != -1) g.add_edge(i, Y, INT_MAX, 0);
            if (U != -1) g.add_edge(i, U, 1, INT_MAX - v[i]);
            Y = i;
        }
    }
    if (Y != -1) g.add_edge(a, Y, INT_MAX, 0);

    auto [flow, cost] = g.flow(a, b);
    long long ans = -(cost - (long long)4 * INT_MAX * flow);
    cout << ans << '\n';
}
0