# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1276395 | k12_khoi | Paint By Numbers (IOI16_paint) | C++17 | 0 ms | 0 KiB |
#include "paint.h"
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,k,c[N],sz[N];
bool f[N],g[N];
string s;
string solve_puzzle(string s,int c[])
{
string ans=s;
int k;
for (int i=0;i<=1000;i++)
if (c[i]==0)
{
k=i;
break;
}
int n=s.size();
string t=s;
function <void(int)> ql = [&] (int i) -> void
{
if (i==n)
{
int cur=0;
for (int i=0;i<n;i++)
sz[i]=0;
for (int i=0;i<n;i++)
{
if (t[i]=='_')
{
if (sz[cur]) cur++;
}
else sz[cur]++;
}
if (sz[cur]) cur++;
if (cur==k)
{
for (int i=0;i<k;i++)
if (c[i]!=sz[i]) return;
for (int i=0;i<n;i++)
if (t[i]=='X') f[i]=true;
else g[i]=true;
}
return;
}
if (s[i]=='.')
{
t[i]='X';
ql(i+1);
t[i]='_';
ql(i+1);
}
else ql(i+1);
};
ql(0);
for (int i=0;i<n;i++)
if (ans[i]=='.')
{
if (f[i] and g[i]) ans[i]='?';
else if (f[i]) ans[i]='X';
else ans[i]='_';
}
return ans;
}