結果

問題 No.1029 JJOOII 3
コンテスト
ユーザー Shibuyap
提出日時 2020-04-19 06:27:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,644 bytes
コンパイル時間 1,401 ms
コンパイル使用メモリ 170,480 KB
実行使用メモリ 30,480 KB
最終ジャッジ日時 2024-10-04 05:44:06
合計ジャッジ時間 6,503 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 1 WA * 1 TLE * 1 -- * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define rrep(i,n) for(int i = 1; i <= (n); ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
#define srep(i,s,t) for (int i = s; i < t; ++i)
using namespace std;
typedef pair<int,int> P;
typedef long long int ll;
#define dame { puts("-1"); return 0;}
#define yn {puts("Yes");}else{puts("No");}

const int MAX_N = 1 << 20;
const int MAX_M = 2;

// セグメント木を持つグローバル配列
int nn;
ll  dat[2 * MAX_N - 1][MAX_M];

// 初期化
void init(int n_){
    // n_が小さいときの変な挙動を防ぐ
    if(n_ < 10)n_ = 10;
    // 簡単のため要素数を2のべき乗に
    nn = 1;
    while(nn < n_)nn *= 2;

    // 全ての値をINT_MAXに
    for(int i = 0; i < 2 * nn - 1; i++)for(int j = 0; j < MAX_M; j++)dat[i][j] = 1001001001001001001;
}

// k番目の値(0-indexed)をaに変更
void update(int k, ll a, int m){
    // 葉の接点
    k += nn - 1;
    dat[k][m] = a;
    // 登りながら更新
    while(k > 0){
        k = (k - 1) / 2;
        dat[k][m] = min(dat[k * 2 + 1][m], dat[k * 2 + 2][m]);
    }
}

// [a,b)の最小値を求める
// 後ろのほうの引数は、計算の簡単のための引数。
// kは節点の番号、l, rはその節点が[l, r)に対応づいていることを表す。
// したがって、外からはquery(a, b, 0, 0, nn)として呼ぶ。
ll query(int a, int b, int k, int l, int r, int m){
    // [a, b)と[l, r)が交差しなければ、INT_MAX
    if (r <= a || b <= l)return (ll)1001001001001001001;

    // [a, b)が[l, r)を完全に含んでいれば、この節点の値
    if (a <= l && r <= b)return dat[k][m];
    else {
        // そうでなければ、2つの子の最小値
        ll vl = query(a, b, k * 2 + 1, l, (l + r) / 2, m);
        ll vr = query(a, b, k * 2 + 2, (l + r) / 2, r, m);
        return min(vl, vr);
    }
}

int main() {
    int n, k;
    cin >> n >> k;
    init(k*3+10);
    string s[n];
    ll c[n];
    rep(i,n)cin >> s[i] >> c[i];
    ll dp[k*3];

    rep(i,k*3)dp[i] = 1001001001001001001;

    int f[n][3] = {};
    int m[n];
    rep(i,n){
        m[i] = s[i].size();
        rep(j,n){
            if(s[i][j] == 'J')f[i][0]++;
            if(s[i][j] == 'O')f[i][1]++;
            if(s[i][j] == 'I')f[i][2]++;
        }
    }

    rep(i,k*3){
        rep(j,n){
            if(k > 200 && 100 <= i && i < k){
                update(i, min(query(i,i+1,0,0,nn,0), query(i-f[j][0],i,0,0,nn,0)+c[j]),0);
                // dp[i] = min(dp[i], dp[i-f[j][0]] + c[j]);
            }else if(k > 200 && k + 100 <= i && i < k*2){
                update(i, min(query(i,i+1,0,0,nn,0), query(i-f[j][1],i,0,0,nn,0)+c[j]),0);
                // dp[i] = min(dp[i], dp[i-f[j][1]] + c[j]);
            }else if(k > 200 && k*2 + 100 <= i && i < k*3){
                update(i, min(query(i,i+1,0,0,nn,0), query(i-f[j][2],i,0,0,nn,0)+c[j]),0);
                // dp[i] = min(dp[i], dp[i-f[j][2]] + c[j]);
            }else{
                int now = i;
                drep(l,m[j]){
                    if(now == -1)break;
                    char x = 'J';
                    if(now >= k)x = 'O';
                    if(now >= k*2)x = 'I';
                    if(s[j][l] == x)now--;
                }
                if(now == -1){
                    update(i, min(query(i,i+1,0,0,nn,0), c[j]),0);
                }
                else update(i, min(query(i,i+1,0,0,nn,0), query(now,i,0,0,nn,0)+c[j]),0);
            }
        }
    }

    ll ans = query(k*3-1,k*3,0,0,nn,0);
    if(ans == 1001001001001001001)ans = -1;
    cout << ans << endl;
    return 0;
}
0