【二分/尺取】Educational Codeforces Round 11_C. Hard Process

Educational Codeforces Round 11_C. Hard Process

题意：

• 长为n的一个数组，只有0 和 1，问任意填补 k 个0，能使得1连续的最大长度是多少。

思路1_尺取法：

• 先使得“尺”的左右端点为l = 1, r = 1
• 左右端点在n的范围内的时候，保证我们“尺”上全是1，并且“尺”中的添加的0的个数在k的范围之内，同时更新答案。
• 如果在保证“尺”上全是1的时候，需要添加的0的个数在k之内，就使得右端点++，反之，左端点++，并且如果左端点是0的话，那么就更新已经用掉的0的个数。
• 另外，如果l > r 了，就使得r = l，然后继续使我们的“尺”往后移。直到跑到最后一个元素为止。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define INF 0x3f3f3f3f
#define MID l + r >> 1
#define lsn rt << 1
#define rsn rt << 1 | 1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
using namespace std;
typedef long long ll;
const int maxN = 3e5 + 7;
const int maxM = 1e7;

int n, k, a[maxN] = {0};

int main()
{
    while(~scanf("%d%d", &n, &k))
    {
        for(int i = 1; i <= n; i ++ )
            scanf("%d", &a[i]);
        int l = 1, r = 1, zero = 0;
        int ans = 0, ansL = 0, ansR = 0;
        while(l <= n && r <= n)
        {
            if(r < l)
                r = l;
            if(a[r] == 1)
            {
                if(ans < r - l + 1)
                {
                    ans =  r - l + 1;
                    ansL = l; ansR = r;
                }
                r ++;
            }
            else if(zero < k)
            {
                if(ans < r - l + 1)
                {
                    ans =  r - l + 1;
                    ansL = l; ansR = r;
                }
                zero ++;
                r ++ ;
            }
            else
            {
                if(a[l] == 0 && k)
                    zero -- ;
                l ++ ;
            }
        }
        printf("%d\n", ans);
        for(int i = 1; i <= n; i ++ )
        {
            if(i >= ansL && i <= ansR)
                printf("1");
            else
                printf("%d", a[i]);
            printf("%c", " \n"[i == n]);
        }
    }
    return 0;
}

思路2_二分法

• 我们用数组的1 - n来存储0 和 1
• 我们开一个num_one数组，num_one[ i ]: 表示从开始位置到现在为止一共有多少个1。
• 这个数组的用处是 i - num_one[ i ] 就是开始到 i 位置需要添多少个0能使得 i 之前全是1
• 我们两个位置i j ( j < i )之间需要多少个0呢？ zero1 = i - num_one[ i ]，zero2 = j - num_one[ j ]。如果a [ j ] == 0 ,那么就是zero1 - zero2 + 1，否则为zero1 - zero2
• 知道了这些之后我们就只需要二分答案即可。
• 二分复杂度为log，我们判断是不是存在这个长度可行的时候的复杂度为O(n)，因为必须是连续的，所以假设我们要找长度为len的一段，那么就只有n - len种可能。
• 所以整体复杂度为O(nlogn)
• AC

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define INF 0x3f3f3f3f
#define MID l + r >> 1
#define lsn rt << 1
#define rsn rt << 1 | 1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
using namespace std;
typedef long long ll;
const int maxN = 3e5 + 7;
const int maxM = 1e7;

int n, k;
int a[maxN];
int num_one[maxN];
int ansL, ansR;

bool judge(int x)
{
    for(int i = 1; i <=n && x <= n; i ++ , x ++ )
    {
        int zeror = x - num_one[x];
        int zerol = i - num_one[i];
        if(a[i] == 0)
        {
            if(zeror - zerol + 1 <= k)
            {
                ansL = i;
                ansR = x;
                return true;
            }
        }
        else
        {
            if(zeror - zerol <= k)
            {
                ansL = i;
                ansR = x;
                return true;
            }
        }
    }
    return false;
}

int main()
{
    while(~scanf("%d%d", &n, &k))
    {
        num_one[0] = 0;
        for(int i = 1; i <= n; i ++ )
        {
            scanf("%d", &a[i]);
            num_one[i] = num_one[i - 1] + a[i];
        }
        int l = 1, r = n;
        int ans = 0;
        while(r - l >= 0)
        {
            int mid = (l + r) / 2;
            if(judge(mid))
            {
                ans = mid;
                l = mid + 1;
            }
            else
                r = mid - 1;
        }
        printf("%d\n", ans);
        for(int i = 1; i <= n; i ++ )
        {
            if(i >= ansL && i <= ansR)
                printf("1");
            else
                printf("%d", a[i]);
            printf("%c", " \n"[i == n]);
        }
    }
    return 0;
}