Submission #592549

#TimeUsernameProblemLanguageResultExecution timeMemory
592549jasminPaint By Numbers (IOI16_paint)C++14
32 / 100
1 ms852 KiB
#include<bits/stdc++.h>
#include<paint.h>
using namespace std;

void make_dp_front(string& s, vector<int>& c, vector<vector<vector<int> > >& dp){
    int n=s.size();
    int k=c.size();

    vector<int> lastwhite(n);
    int last=-10;
    for(int i=0; i<n; i++){
        if(s[i]=='_'){
            last=i;
        }
        lastwhite[i]=last;
    }

    dp[0][0][0]=true;

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

            if(s[i-1]=='_'){//white
                dp[i][j][0]=dp[i-1][j][0] || dp[i-1][j][1];
                dp[i][j][1]=false;
            }
            else if(s[i-1]=='X'){ //black
                dp[i][j][0]=false;
                if(0<j && 0<=i-c[j-1] && lastwhite[i-1]+1<=i-c[j-1]){
                    dp[i][j][1]=dp[i-c[j-1]][j-1][0];
                }
            }
            else{
                dp[i][j][0]=dp[i-1][j][0] || dp[i-1][j][1];
                if(0<j && 0<=i-c[j-1] && lastwhite[i-1]+1<=i-c[j-1]){
                    dp[i][j][1]=dp[i-c[j-1]][j-1][0];
                }
            }
            
        }
    }
}
void make_dp_back(string& s, vector<int>& c, vector<vector<vector<int> > >& dp){
    int n=s.size();
    int k=c.size();

    vector<int> lastwhite(n);
    int last=n+10;
    for(int i=n-1; i>=0; i--){
        if(s[i]=='_'){
            last=i;
        }
        lastwhite[i]=last;
    }
    dp[n][k][0]=true;

    for(int i=n-1; i>=0; i--){
        for(int j=k; j>=0; j--){

            if(s[i]=='_'){//white
                dp[i][j][0]=dp[i+1][j][0] || dp[i+1][j][1];
                dp[i][j][1]=false;
            }
            else if(s[i]=='X'){ //black
                dp[i][j][0]=false;
                if(j<k && i+c[j]<=n && i+c[j]<=lastwhite[i]){
                    dp[i][j][1]=dp[i+c[j]][j+1][0];
                }
            }
            else{
                dp[i][j][0]=(dp[i+1][j][0] || dp[i+1][j][1]);
                if(j<k && i+c[j]<=n && i+c[j]<=lastwhite[i]){
                    dp[i][j][1]=dp[i+c[j]][j+1][0];
                }
            }
            
        }
    }
}


string solve_puzzle(string s, vector<int> c){
    int n=s.size(); 
    int k=c.size();

    vector<vector<vector<int> > > dp_front(n+1, vector<vector<int> > (k+1, vector<int> (2, false)));
    make_dp_front(s, c, dp_front);
    vector<vector<vector<int> > > dp_back(n+1, vector<vector<int> > (k+1, vector<int> (2, false)));
    make_dp_back(s, c, dp_back);

    vector<bool> black(n);
    vector<bool> white(n);
    for(int j=0; j<k; j++){
        int r=1; //exklusiv
        for(int l=0; l<n; l++, r=min(r, l+1)){
            while(r<=n && r-l<c[j]){
                if(s[r]=='_'){
                    l=r;
                    r=r+1;
                    continue;
                }
                r++;
            }
            if(r>n) break;

            //cout << l << " " << r << "\n" << flush;

            if(r-l==c[j] && dp_front[l][j][0] && dp_back[r][j+1][0]){
                for(int i=l; i<r; i++){
                    black[i]=true;
                }
            }
            //cout << "uh?\n" << flush;
        }

    }
    for(int j=0; j<=k; j++){
        for(int i=0; i<n; i++){
            if((dp_front[i][j][0] || dp_front[i][j][1])
                && (dp_back[i+1][j][0] || dp_back[i+1][j][1])){

                    white[i]=true;
                }
        }
    }

    string ans=s;
    for(int i=0; i<n; i++){
        assert(black[i] || white[i]);
        if(s[i]=='_' || !black[i]){
            ans[i]='_';
        }
        else if(s[i]=='X' || !white[i]){
            ans[i]='X';
        }
        else{
            ans[i]='?';
        }
    }
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...