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

样例输出2

-1

思路:一开始写了一个暴力，复杂度是O(n2)，果不其然TLE，然后几何维护，二分查找，复杂度降到O(nlogn)

#include <bits/stdc++.h>

using namespace std;
const int MAXN=200005;

bool vis[MAXN];
int n,k,a;
int m;


set<int>s;
set<int>::iterator it1;
set<int>::iterator it2;


int main()
{
    cin>>n>>k>>a;
    cin>>m;
    int x;


    s.insert(0);
    s.insert(n+1);
    int  maxLen=(n+1)/(a+1);


    memset(vis,0,sizeof(vis));


    for(int i=0;i<m;i++)
    {
        cin>>x;
        if(vis[x])
        {
            continue;
        }
        vis[x]=true;

        s.insert(x);
        it1=s.lower_bound(x);
        it2=s.upper_bound(x);
        it1--;
        //cout<<(*it1)<<" "<<(*it2)<<endl;
        maxLen-=(*it2-*it1)/(a+1)-(*it2-x)/(a+1)-(x-*it1)/(a+1);
        if(maxLen<k)
        {
            cout<<i+1<<endl;
            return 0;
        }

    }
    cout<<-1<<endl;



    return 0;
}