Submission #4063792

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int MOD = 1000000007LL;

signed main(){

    int n, x; cin >> n >> x;
    string s; cin >> s;
    //番兵
    s.push_back('m');
    n = (int)s.size();
    vector<vector<vector<int> > > dp(n + 2, vector<vector<int> > (n * n + 1, vector<int> (n + 1, 0LL)));
    dp[0][0][0] = 1LL;

    for(int i = 0; i <= n; i++){

        //cout << s[i] << endl;
        for(int j = 0; j <= x; j++){
            for(int k = 0; k <= n; k++){
                
                //cout << "dp[" << i << "][" << j << "][" << k << "] = " << dp[i][j][k] << endl; 
                if(i == n) continue;
                if(dp[i][j][k] == 0) continue;

                if(s[i] == '2' || s[i] == '?'){
                    if(k % 2 == 0){
                        dp[i + 1][j][k + 1] = (dp[i + 1][j][k + 1] + dp[i][j][k]) % MOD;
                        dp[i + 1][j][k + 1] %= MOD;
                    }else{
                        int num = k / 2;
                        int sum = ((1 + num) * num) / 2;
                        if(j + sum >= x) sum = x - j;
                        dp[i + 1][j + sum][1] = (dp[i + 1][j + sum][1] + dp[i][j][k]) % MOD;
                        dp[i + 1][j + sum][1] %= MOD;
                        
                    }
                }
                
                if(s[i] == '5' || s[i] == '?'){
                    if(k % 2 == 1){
                        dp[i + 1][j][k + 1] = (dp[i + 1][j][k + 1] + dp[i][j][k]) % MOD;
                        dp[i + 1][j][k + 1] %= MOD;
                    }else{
                        int num = k / 2;
                        int sum = ((1 + num) * num) / 2;
                        if(j + sum >= x) sum = x - j;
                        dp[i + 1][j + sum][0] = (dp[i + 1][j + sum][0] + dp[i][j][k]) % MOD;
                        dp[i + 1][j + sum][0] %= MOD;
                        
                    }
                }
                
                if((s[i] != '2' && s[i] != '5') || s[i] == '?'){
                    int val = 1LL;
                    if(s[i] == '?') val = 8LL;
                    
                    int num = k / 2;
                    int sum = ((1 + num) * num) / 2;
                    if(j + sum >= x) sum = x - j;
                    dp[i + 1][j + sum][0] = (dp[i + 1][j + sum][0] + dp[i][j][k] * val) % MOD;
                    dp[i + 1][j + sum][0] %= MOD;
                }
            }
        }
    }

    int ans = 0LL;
    for(int j = x; j <= n * n; j++){
        ans += dp[n][j][0];
        ans %= MOD;
    }

    cout << ans % MOD << endl;
    return 0;
}

Submission Info

Judge Result