結果

問題 No.910 素数部分列
ユーザー NOSS
提出日時 2019-10-18 22:16:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,070 bytes
コンパイル時間 1,887 ms
コンパイル使用メモリ 169,252 KB
実行使用メモリ 32,000 KB
最終ジャッジ日時 2024-06-25 17:11:41
合計ジャッジ時間 4,009 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using ll = long long int;
using ull = unsigned long long int;
using P = pair<ll, ll>;
using P3 = pair<ll, P>;
using PP = pair<P ,P>;
constexpr ll MOD = ll(1e9+7);
constexpr int IINF = INT_MAX;
constexpr ll LLINF = LLONG_MAX;
constexpr int MAX_N = int(2e5) + 5;
constexpr double EPS = 1e-8;
constexpr int di[] = {0, 1, 0, -1}, dj[] = {1, 0, -1, 0};
#define REP(i, n) for (int i = 0; i < n; i++)
#define REPR(i, n) for (int i = n; i >= 0; i--)
#define SORT(v) sort((v).begin(), (v).end())
#define SORTR(v) sort((v).rbegin(), (v).rend())
#define ALL(v) (v).begin(), (v).end()

void chmax(int &a, int b){
    a = max(a, b);
}

int dp[MAX_N][6][6];

int main(){
    int n, ans = 0;
    string s, t = "";
    cin >> n >> s;
    for(int i=0;i<n;i++){
        if(s[i] == '3' || s[i] == '5' || s[i] == '7') ans++;
        else if(s[i] == '1'){
            t.push_back('1');
        }
        else{
            t.push_back('9');
        }
    }
    int m = t.size();
    for(int i=0;i<m;i++){
        for(int j=0;j<6;j++){
            for(int k=0;k<6;k++){
                dp[i][j][k] = -IINF;
            }
        }
    }
    dp[0][0][0] = 0;
    for(int i=0;i<m;i++){
        for(int c1=0;c1<5;c1++){
            for(int c9=0;c9<5;c9++){
                chmax(dp[i+1][c1][c9], dp[i][c1][c9]);
                if(t[i] == '1'){
                    chmax(dp[i+1][c1+1][c9], dp[i][c1][c9]);
                    // 11
                    if(c1 >= 1) chmax(dp[i+1][c1-1][c9], dp[i][c1][c9]+1);
                    // 991
                    if(c9 >= 2) chmax(dp[i+1][c1][c9-2], dp[i][c1][c9]+1);
                }
                else{
                    chmax(dp[i+1][c1][c9+1], dp[i][c1][c9]);
                    // 19
                    if(c1 >= 1) chmax(dp[i+1][c1-1][c9], dp[i][c1][c9]+1);
                }
            }
        }
    }
    int maxi = 0;
    for(int i=0;i<6;i++){
        for(int j=0;j<6;j++){
            maxi = max(maxi, dp[m][i][j]);
        }
    }
    ans += maxi;
    cout << ans << endl;
    return 0;
}
0