結果
問題 |
No.291 黒い文字列
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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; }