Codeforces 401C 题解

题意简述

构造一个01序列，使得这个序列中连续的0长度小于2，连续的1长度小于3，并且有n个0，m个1。（有SPJ，构造一个即可）。没有解输出-1。$n,m<=1000000$

数据

输入
1 2
输出
101
输入
4 8
输出
110110110101
输入
4 10
输出
11011011011011
输入
1 5
输出
-1

思路

如果您数学有一定水平，那么首先想到的必定是插入0的方法。先固定下来$m$个$1$，然后往这些$1$中插入$0$。拿第2组样例举例子。我们先确定
$$\begin{array}{cccccccc}1& 1& 1& 1& 1& 1& 1& 1\end{array}$$
为了使连续的$1$数量小于$3$个，我们每隔两个$1$就要插入一个$0$。变成：
$$\begin{array}{ccccccccccc}1& 1& 0& 1& 1& 0& 1& 1& 0& 1& 1\end{array}$$
然后我们就会发现：如果$n$不够插入这些$0$，就无解了。经过很多次$WA$的提交，我们发现这个关系是$n>=\lfloor \frac{(m+1)}{2}\rfloor -1$。除此之外还要满足$n<=m+1$，不然$0$太多也会不合法，放不下。
回到这个样例。已经放了$3$个$0$，接下来还需要放$4-3=1$个$0$。我们只需要从左往右找空位插入即可。原序列变成：
$$\begin{array}{cccccccccccc}0& 1& 1& 0& 1& 1& 0& 1& 1& 0& 1& 1\end{array}$$
那么问题来了：如何快速的插入呢？
我们会发现，每个$1$前面最多有连续的$1$个$0$。所以，我们记录一个$pre0$数组，$pre0[i]$表示位置$i$前是否有$0$。但是，有一种情况：
$$\begin{array}{ccccc}0& 1& 0& 1& 0\end{array}$$
即$n=m+1$的情况。此时有一个$0$后面没有$1$，不就无法记录了？
没有问题，只要特判就珂以了，因为只有这一种情况是无法记录的。
代码：

#include<bits/stdc++.h>
#define N 2001000
using namespace std;
string s;
bool pre0[N];
int n,m;

void Construct()
{
    if (n==m+1)//特判n=m+1
    {
        for(int i=0;i<n+m;i++)
        {
            putchar(i%2+'0');
        }//01交替输出
        putchar('\n');
        return;
    }
    if (n>=(m+1)/2-1 and n<=m+1)//有解
    {
        for(int i=0;i<m;i++)//先放m个1
        {
            s+='1';//准确来讲这个s只是为了骗分（骗0分）用的，下面没用
        }


        int cnt0=0;//放了多少0
        for(int i=2;i<m;i+=2)//隔两个插入0
        {
            pre0[i]=1;++cnt0;
        }

        int rest=n-cnt0;//还要放多少
        int put=0/*接下来放了多少0*/,pos=0/*第一个能放0(即pre0[i]=0)的位置*/;
        while(put<rest)
        {
            while(pre0[pos]) ++pos;//找一个能放的位置
            pre0[pos]=1;++put;//放上
        }

        for(int i=0;i<m;i++)
        {
            if (pre0[i])//如果前有0就输出
            {
                putchar('0');
            }
            putchar('1');//输出1
        }
    }
    else//无解
    {
        puts("-1");
    }
}
main()
{
    scanf("%d%d",&n,&m);
    Construct();
    return 0;
}