結果

問題 No.291 黒い文字列
コンテスト
ユーザー kimiyuki
提出日時 2016-06-22 03:59:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 249 ms / 2,000 ms
コード長 2,302 bytes
コンパイル時間 1,115 ms
コンパイル使用メモリ 86,144 KB
実行使用メモリ 115,584 KB
最終ジャッジ日時 2024-10-11 19:13:36
合計ジャッジ時間 3,811 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#define whole(f,x,...) ([&](decltype((x)) y) { return (f)(begin(y), end(y), ## __VA_ARGS__); })(x)
template <class T> bool setmax(T & l, T const & r) { if (not (l < r)) return false; l = r; return true; }
using namespace std;
template <typename T, typename X> auto vectors(T a, X x) { return vector<T>(x, a); }
template <typename T, typename X, typename Y, typename... Zs> auto vectors(T a, X x, Y y, Zs... zs) { auto cont = vectors(a, y, zs...); return vector<decltype(cont)>(x, cont); }

int n;
string s;
vector<vector<vector<vector<vector<int> > > > > memo;
int go(int i, int k, int ku, int kur, int kuro) {
    if (i == n) return 0;
    if (memo[i][k][ku][kur][kuro] != -1) return memo[i][k][ku][kur][kuro];
    bool nk     =          k    + 1 <= n/5;
    bool nku    = k    and ku   + 1 <= n/5;
    bool nkur   = ku   and kur  + 1 <= n/5;
    bool nkuro  = kur  and kuro + 1 <= n/5;
    bool nkuroi = kuro                    ;
    if (s[i] == 'K') return memo[i][k][ku][kur][kuro] = go(i+1, k+nk,  ku,      kur,       kuro);
    if (s[i] == 'U') return memo[i][k][ku][kur][kuro] = go(i+1, k-nku, ku+nku,  kur,       kuro);
    if (s[i] == 'R') return memo[i][k][ku][kur][kuro] = go(i+1, k,     ku-nkur, kur+nkur,  kuro);
    if (s[i] == 'O') return memo[i][k][ku][kur][kuro] = go(i+1, k,     ku,      kur-nkuro, kuro+nkuro);
    if (s[i] == 'I') return memo[i][k][ku][kur][kuro] = go(i+1, k,     ku,      kur,       kuro-nkuroi) + nkuroi;
    int acc = -1;
    if (nk)        setmax(acc, go(i+1, k+nk,  ku,      kur,       kuro));
    if (nku)       setmax(acc, go(i+1, k-nku, ku+nku,  kur,       kuro));
    if (nkur)      setmax(acc, go(i+1, k,     ku-nkur, kur+nkur,  kuro));
    if (nkuro)     setmax(acc, go(i+1, k,     ku,      kur-nkuro, kuro+nkuro));
    if (nkuroi)    setmax(acc, go(i+1, k,     ku,      kur,       kuro-nkuroi) + nkuroi);
    if (acc == -1) setmax(acc, go(i+1, k,     ku,      kur,       kuro));
    return memo[i][k][ku][kur][kuro] = acc;
};
int main() {
    cin >> s;
    s.erase(whole(remove_if, s, [&](char c) { return string("KUROI?").find(c) == string::npos; }), s.end());
    n = s.length();
    memo = vectors(-1, n+1, n/5+1, n/5+1, n/5+1, n/5+1);
    cout << go(0, 0, 0, 0, 0) << endl;
    return 0;
}
0