yukicoder No.291 黒い文字列
解法
dp[n][k][u][r][o] = (n 文字目までの時点で, "K" のストックが k 個, "KU" のストックが u 個, "KUR" のストックが r 個...) の時の"KUROI" が出来た substring の数
という DP をやります。最高で 100/5 = 20 個しか文字列は完成しないので, 計算量は 100*20^4 でなんとか間に合います。
int dp[101][21][21][21][21]; void chmax(int& x, int y) { x = max(x, y); } int main() { cin.tie(0); ios::sync_with_stdio(false); string s; cin >> s; int n = s.size(); memset(dp, -1, sizeof(dp)); dp[0][0][0][0][0] = 0; for (int i = 0; i < n; i++) for (int k = 0; k <= 20; k++) for (int u = 0; u <= 20; u++) for (int r = 0; r <= 20; r++) for (int o = 0; o <= 20; o++) { chmax(dp[i+1][k][u][r][o], dp[i][k][u][r][o]); if (dp[i][k][u][r][o] == -1) continue; switch (s[i]) { case 'K': chmax(dp[i+1][min(k+1, 20)][u][r][o], dp[i][k][u][r][o]); break; case 'U': if (k > 0) chmax(dp[i+1][k-1][min(u+1, 20)][r][o], dp[i][k][u][r][o]); break; case 'R': if (u > 0) chmax(dp[i+1][k][u-1][min(r+1, 20)][o], dp[i][k][u][r][o]); break; case 'O': if (r > 0) chmax(dp[i+1][k][u][r-1][min(o+1, 20)], dp[i][k][u][r][o]); break; case 'I': if (o > 0) chmax(dp[i+1][k][u][r][o-1], dp[i][k][u][r][o]+1); break; case '?': chmax(dp[i+1][min(k+1, 20)][u][r][o], dp[i][k][u][r][o]); if (k > 0) chmax(dp[i+1][k-1][min(u+1, 20)][r][o], dp[i][k][u][r][o]); if (u > 0) chmax(dp[i+1][k][u-1][min(r+1, 20)][o], dp[i][k][u][r][o]); if (r > 0) chmax(dp[i+1][k][u][r-1][min(o+1, 20)], dp[i][k][u][r][o]); if (o > 0) chmax(dp[i+1][k][u][r][o-1], dp[i][k][u][r][o]+1); break; default: break; } } int ans = 0; for (int k = 0; k <= 20; k++) for (int u = 0; u <= 20; u++) for (int r = 0; r <= 20; r++) for (int o = 0; o <= 20; o++) ans = max(ans, dp[n][k][u][r][o]); cout << ans << endl; return 0; }