51NOD 1521 一维战舰

1521 一维战舰
题目来源： CodeForces
基准时间限制：1 秒 空间限制：131072 KB 分值: 10 难度：2级算法题
收藏
关注

爱丽丝和鲍博喜欢玩一维战舰的游戏。他们在一行有n个方格的纸上玩这个游戏（也就是1×n的表格）。

在游戏开始的时候，爱丽丝放k个战舰在这个表格中，并不把具体位置告诉鲍博。每一只战舰的形状是 1×a 的长方形（也就是说，战舰会占据a个连续的方格）。这些战舰不能相互重叠，也不能相接触。

然后鲍博会做一系列的点名。当他点到某个格子的时候，爱丽丝会告诉他那个格子是否被某只战舰占据。如果是，就说hit，否则就说miss。

但是这儿有一个问题！爱丽丝喜欢撒谎。他每次都会告诉鲍博miss。

请你帮助鲍博证明爱丽丝撒谎了，请找出哪一步之后爱丽丝肯定撒谎了。

Input

单组测试数据。
第一行有三个整数n,k和a(1≤n,k,a≤2*10^5)，表示表格的大小，战舰的数目，还有战舰的大小。输入的n,k,a保证是能够在1×n的表格中放入k只大小为a的战舰，并且他们之间不重叠也不接触。
第二行是一个整数m(1≤m≤n)，表示鲍博的点名次数。
第三行有m个不同的整数x1,x2,...,xm，xi是鲍博第i次点名的格子编号。格子从左到右按照1到n编号。

Output

输出一个整数，表示最早一次能够证明爱丽丝一定撒谎的点名编号。如果不能证明，输出-1。点名的编号依次从1到m编号。

Input示例

样例1
11 3 3
5
4 8 6 1 11

样例2
5 1 3
2
1 5

Output示例

样例输出1
3

样例输出2
-1

不难发现 当有一片连续的格子 长度为
2a+1 ->能放2艘战舰
3a+2 ->3艘
4a+3 ->4艘
所以 (L+1)/(a+1)能判断长度为L的连续格子能放几个战舰
维护一个canPlace 表示最多能放多少艘战舰
如果加入一次询问 就把询问的位置所在的连续区域断开
长度为L的区域变为 L1+L2的2段 而且L1+L2+1=L
(L+1)/(a+1)-(L1+1)/(a+1)-(L2+1)/(a+1)就得到canPlace减少的数量
用set维护已经查询过的点 每次找到L的2个端点只需要O(log(m))
复杂度为m*log(m)

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string>
#include<vector>
#include<deque>
#include<queue>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<math.h>
#include<list>
#include<cstring>
#include<fstream>
//#include<memory.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define INF 1000000007
#define pll pair<ll,ll>
#define pid pair<int,double>

set<int>exist;
int canPlace;
int n,k,a;

bool canInsert(int pos){
    if(exist.find(pos)!=exist.end())//已经询问过pos了
        return true;
    exist.insert(pos);
    auto it=exist.find(pos);
    int L1,L2;
    auto before=it;
    auto after=it;
    ++after;
    if(it==exist.begin())
        L1=pos-1;
    else{
        --before;
        L1=pos-*before-1;
    }
    if(after==exist.cend())
        L2=n-pos;
    else{
        L2=*after-pos-1;
    }
    canPlace-=(L1+L2+1+1)/(a+1)-(L1+1)/(a+1)-(L2+1)/(a+1);
    return canPlace>=k;
}

int main()
{
    //freopen("/home/lu/文档/r.txt","r",stdin);
    //freopen("/home/lu/文档/w.txt","w",stdout);
    int m,pos,i;
    bool flag;
    while(~scanf("%d%d%d",&n,&k,&a)){
        exist.clear();
        canPlace=(n+1)/(a+1);
        scanf("%d",&m);
        flag=false;//标记是否已经发现撒谎
        for(i=1;i<=m;++i){
            scanf("%d",&pos);
            if((!flag)&&(!canInsert(pos))){
                printf("%d\n",i);
                flag=true;
            }
        }
        if(!flag)
            printf("-1\n");
    }
    return 0;
}