51nod 1521 一维战舰【set】

爱丽丝和鲍博喜欢玩一维战舰的游戏。他们在一行有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
思路：
思路不难想，但是每插入一个值，集合就要改变且集合为非递减，想了想，居然数组模拟没超时。

//有点意外，这个复杂度也能过
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
const int maxn = 2 * 1e5  + 10;
typedef long long LL;
using namespace std;
bool flag[maxn];

int main() {
    int n, k, a, m, cnt;
    scanf("%d %d %d", &n, &k, &a);
    scanf("%d", &m);
    int id = 0;
    int sum = (n + 1) / (a + 1);
    for(int i = 1; i <= m; i++) {
        scanf("%d", &cnt);
        flag[cnt] = 1;
        int L = cnt - 1, R = cnt + 1;
        while(L >= 1 && !flag[L]) L--;
        while(R <= n && !flag[R]) R++;
        sum -= (R - L) / (a + 1) - (R - cnt) / (a + 1) - (cnt - L) / (a + 1);
        if(sum < k && !id) {
            id = i;
            break;
        } 
    }
    if(!id) printf("-1\n");
    else printf("%d\n", id);
    return 0;
} 

看到网上有set的做法：

#include <cstdio>
#include <iostream>
#include <set>
#include <algorithm>
const int maxn = 2 * 1e5  + 10;
using namespace std;
bool flag[maxn];
set<int> s;
set<int>::iterator L;
set<int>::iterator R;

int main() {
    int n, k, a, m, cnt;
    scanf("%d %d %d", &n, &k, &a);
    scanf("%d", &m);
    int id = 0;
    int sum = (n + 1) / (a + 1);
    s.insert(0); //下界 
    s.insert(n + 1); //上界 
    for(int i = 1; i <= m; i++) {
        scanf("%d", &cnt);
        s.insert(cnt);
        L = s.lower_bound(cnt), L--;
        R = s.upper_bound(cnt);
        sum -= (*R - *L) / (a + 1) - (*R - cnt) / (a + 1) - (cnt - *L) / (a + 1);
        if(sum < k && !id) {
            id = i;
            break;
        } 
    }
    if(!id) printf("-1\n");
    else printf("%d\n", id);
    return 0;
}