[CodeForces 567D] One-Dimensional Battle Ships （区间隔断计数）

CodeForces - 567D

给定一个序列，上面有 K艘长度为 A的船互相不相邻地摆放在一起
Bob 每次都射击一个位置，Alice每次都告诉他没中
问到第几次射击的时候，可以判断出 Alice在说谎

Alice在说谎，说明射击点将区间分割成的若干个区间放不下 K艘船
一开始想着用单调队列维护了一个 DP，然后搞了些奇怪的做法，然后就 WA了

事实上这题很简单，用一个 set维护一下每次操作的位置
每次操作都将原来的一个区间隔断成了两个
然后新增一个操作的时候，在 set里upper_bound找这个位置左右的隔断点
将总的可放船数减去左右隔断点内的可放船数
再加上新增的隔断点所分成的两个区间内的可放船数
当总的可放船数小于 K*(A+1)时矛盾

注意一下这题船与船是不能相邻的

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <queue>
using namespace std;
typedef pair<int,int> Pii;
typedef long long LL;
typedef unsigned long long ULL;
typedef double DBL;
typedef long double LDBL;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Pow2(a) (a*a)

const int maxn=2e5+10;
int N,K,A,M;
int posi[maxn];
set<int> list;

int Get(int);

int main()
{
    while(~scanf("%d%d%d%d", &N, &K, &A, &M))
    {
        int now=Get(N),ans=-1;
        for(int i=1; i<=M; i++) scanf("%d", &posi[i]);
        list.clear();
        list.insert(0);list.insert(N+1);
        for(int i=1; i<=M; i++)
        {
            set<int>::iterator it=list.upper_bound(posi[i]);
            int aft=*it;it--;
            int pre=*it;
            now=now-Get(aft-pre-1)+Get(posi[i]-pre-1)+Get(aft-posi[i]-1);
            if(now<K){ans=i;break;}
            list.insert(posi[i]);
        }
        printf("%d\n", ans);
    }
    return 0;
}

int Get(int len)
{
    return (len+1)/(A+1);
}