結果

問題 No.1029 JJOOII 3
ユーザー QCFium
提出日時 2020-04-17 22:11:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 288 ms / 2,000 ms
コード長 2,172 bytes
コンパイル時間 1,571 ms
コンパイル使用メモリ 173,564 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-03 13:23:10
合計ジャッジ時間 6,387 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
int ri() {
    int n;
    scanf("%d", &n);
    return n;
}
int main() {
    int n = ri();
    int need = ri();
    std::vector<int> s[n];
    int cost[n];
    for (int i = 0; i < n; i++) {
        std::string tmp;
        std::cin >> tmp;
        for (auto j : tmp) s[i].push_back(j == 'J' ? 0 : j == 'O' ? 1 : 2);
        cost[i] = ri();
    }
    
    int64_t dp[3][100001];
    for (auto &i : dp) for (auto &j : i) j = 1000000000000000000;
    
    for (int i = 0; i < 3; i++) {
        dp[i][0] = 0;
        for (int j = 0; j < n; j++) {
            int cnt = std::count(s[j].begin(), s[j].end(), i);
            for (int k = 0; k <= 100000; k++)
                dp[i][std::min(100000, k + cnt)] = std::min(dp[i][std::min(100000, k + cnt)], dp[i][k] + cost[j]);
        }
        for (int j = 100000; j; j--) dp[i][j - 1] = std::min(dp[i][j - 1], dp[i][j]);
    }
    
    // two
    int64_t res = 1000000000000000000;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j <= (int) s[i].size(); j++) {
            int left0 = std::count(s[i].begin(), s[i].begin() + j, 0);
            int right0 = std::count(s[i].begin() + j, s[i].end(), 1);
            for (int k = 0; k < n; k++) {
                int left1 = 0;
                int right1 = std::count(s[k].begin(), s[k].end(), 2);
                for (int l = 0; l <= (int) s[k].size(); l++) {
                    int need0 = std::max(0, need - left0);
                    int need1 = std::max(0, need - right0 - left1);
                    int need2 = std::max(0, need - right1);
                    res = std::min(res, dp[0][need0] + dp[1][need1] + dp[2][need2] + cost[i] + cost[k]);
                    if (l < (int) s[k].size()) {
                        if (s[k][l] == 1) left1++;
                        if (s[k][l] == 2) right1--;
                    }
                }
            }
        }
    }
    // one
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < (int) s[i].size(); j++) {
            int left = std::count(s[i].begin(), s[i].begin() + j, 0);
            int head = s[i].size();
            int cnt = 0;
            for (int k = j; k < (int) s[i].size(); k++) {
                if (s[i][k] == 1) cnt++;
                if (cnt >= need) {
                    head = k;
                    break;
                }
            }
            if (head == (int) s[i].size()) break;
            head++;
            int right = std::count(s[i].begin() + head, s[i].end(), 2);
            res = std::min(res, cost[i] + dp[0][std::max(0, need - left)] + dp[2][std::max(0, need - right)]);
        }
    }
    printf("%" PRId64 "\n", res == 1000000000000000000 ? -1 : res);
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0