IOI2016Day2. paint

---

题目链接：http://uoj.ac/problem/238

题目大意：

有一个长度为n的黑白序列，告诉你所以k个极长连续黑段长度和顺序。有一些位置的颜色已知，需要判断剩下未知的位置哪

些颜色

一定是白或一定是黑。保证至少存在一组解。

---

分析:

先猜测，根据数据范围，虽然n很大，但是k却很小，可以考虑复杂度为O(n*k)的算法。(*￣︶￣)

我们再来具体看看，可以设状态：

f0[i][j] 表示前i个位置能匹配j段并且i位置为白是否合法。

f1[i][j] 表示前i个位置能匹配j段并且i位置为黑是否合法，可以O(n*k)求出。

但这样却又漏洞，我们只是保证了从前往后i,j合法，但是我们还不能保证在这种情况满足后面i+1到n-1是否合法，大概就是我们不能保证

剩下的位置够不够。

为了解决这个问题我们再定义两个状态，g0[i][j],g1[i][j]从后往前来记录。

这样状态转移就是：

if(a[i]!=X) f[i][j][0]=f[i-1][j][0]|f[i-1][j][1] (可以与前面一个组成一组，也可以接在一个黑色后面，有一种满足都合法)

if(i-b[j]>=0&&sum[i]-sum[i-b[j]]==0)f[i][j][1]=f[i-b[j]][j-1][0];(在i-b[j]+1-i直接没有X&&i-b[j]位置为白)

g的转移一样，只是从后往前处理。

---

答案：

如果f[i][j][0]==1&&g[i][j+1][0](表示从前往后第i个位置满足j段并且当前为白色时合法&&从i到n-1满足j+1到m段的块，并且当前

为白色合法)

两个都合法说明i就可以为白色，用y记录是否为白色。

如果f[i][j][1]==1&&g[i+1][j+1][0](上面类似，当前为黑，那么下一位就要为白）

两个都合法说明i就可以为黑色，用个x记录一个数能为黑色的数量，如果最后x>0说明可以为黑色。

---

统计：

如果y[i]&&x[i]，两个都可以输出?，其他x,y输出即可。

---

附上交互代码：

//g为从后往前 f为从前往后 是否合法
#include "paint.h"
#include<bits/stdc++.h>
using namespace std;

const int N=2e5+;
const int K=;
bool g[N][K][],f[N][K][];//第i位 满足j段  并现在为0/1 白/黑
int x[N],y[N],last;
int tag[N];
char a[N];int b[N],sum1[N],sum2[N];
string res;
string solve_puzzle(std::string s, std::vector<int> c)
{
    int n=s.length(),m=c.size();
    for(int i=;i<=n;i++) a[i]=s[i-],sum1[i]+=sum1[i-]+(a[i]=='X'),sum2[i]+=sum2[i-]+(a[i]=='_');
    for(int i=;i<=m;i++) b[i]=c[i-];
    g[n+][m+][]=;
    for(int i=n;i>=;i--)
    {
        for(int j=m+;j>=;j--)
        {
            if(a[i]^'X') g[i][j][]=g[i+][j][]|g[i+][j][];
            if(j!=m+&&n-i+>=b[j]&&sum2[i+b[j]-]-sum2[i-]==) g[i][j][]=g[i+b[j]][j+][];
        }
    }
    f[][][]=;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            if(a[i]!='X') f[i][j][]=f[i-][j][]|f[i-][j][];
            if(j&&i>=b[j]&&sum2[i]-sum2[i-b[j]]==) f[i][j][]=f[i-b[j]][j-][];
            if(f[i][j][]&&g[i][j+][]) y[i]=;
            if(f[i][j][]&&g[i+][j+][]) x[i-b[j]+]++,x[i+]--;
        }
    }
    for(int i=;i<=n;i++)
    {
        x[i]+=x[i-];
        if(a[i]=='.')
        {
            if(x[i]&&y[i]) res+='?';
            else if(x[i]) res+='X';
            else res+='_';
        }else res+=a[i];
    }
    return res;
}

---

转载请声明！！！

---