codeforces 567D One-Dimensional Battle Ships (set)

/*
题意：
在1~n的格子里，有k艘，每艘长为a的船，并且这些船不相交且不能相邻。
现在有m个射击，每个射击给你一个位置（1~n），并且不会重复。
由于对方每次都说没打中，因此你想找到至少要射击几次才能证明对方说谎。
若不能证明对方说谎，输出-1。
思路：
n个格子，船长为a，则能放（n+1）/（a+1)艘船。（由于要求不能相邻，因此a+1）。

每次在set中插入射击点，二分上下界 原来可以放的船数减去因这个点而不能放的与k进行比较
*/
# include<stdio.h>
# include<algorithm>
# include<string.h>
# include<set>
using namespace std;
set<int>s;
set<int>::iterator it1;
set<int>::iterator it2;
int vis[200010];
int num[200010];
int main()
{
    int n,k,a,i,m;
    while(~scanf("%d%d%d",&n,&k,&a))
    {
        s.clear();
        scanf("%d",&m);
        for(i=1; i<=m; i++)
            scanf("%d",&num[i]);
        int tmp=(n+1)/(a+1);///原来可以放这么多
        s.insert(0);//上下边界
        s.insert(n+1);
        memset(vis,0,sizeof(vis));
        for(i=1; i<=m; i++)
        {
            if(vis[num[i]])
                continue;
            vis[num[i]]=1;
            s.insert(num[i]);
            it1=s.lower_bound(num[i]);
            it1--;
            it2=s.upper_bound(num[i]);
            tmp-=((*it2)-(*it1))/(a+1)-((*it2)-num[i])/(a+1)-(num[i]-(*it1))/(a+1);
            if(tmp<k)
            {
                printf("%d\n",i);
                break;
            }
            //  printf("%d %d\n",*it1,*it2);
        }
        if(i>m)
            printf("-1\n");
    }
    return 0;
}