Submission #592788

#	Time	Username	Problem	Language	Result	Execution time	Memory
592788		jasmin	Paint By Numbers (IOI16_paint)	C++14	90 / 100	2096 ms	228516 KiB

paint

#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++){
        for(int l=0, r=c[j]; r<=n; l++, r++){
            bool ok=true;
            for(int i=l; i<r; i++){
                if(s[i]=='_'){
                    ok=false;
                }
            }

            if(ok && dp_front[l][j][0] && dp_back[r][j+1][0]){
                for(int i=l; i<r; i++){
                    black[i]=true;
                }
            }
        }

    }
    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;
}

• Subtask 1: 7 / 7
• Subtask 2: 3 / 3
• Subtask 3: 22 / 22
• Subtask 4: 27 / 27
• Subtask 5: 21 / 21
• Subtask 6: 10 / 10
• Subtask 7: 0 / 10