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编号。

样例1

11 3 3
5
4 8 6 1 11

样例2

5 1 3
2
1 5

样例输出1

3

样例输出2

-1

思路

题目是中文的。

我们可以用set来维护一个每次插进去的点的值,首先这么多长度能放下的最大战舰数量一定是 n + 1 a + 1 个,然后刚开始往set里面插入0n+1,然后每一次进行插点的时候找到大于点的第一个值和小于点的第一个值,计算出这个点右边能放下的战舰数量和左边能放下的战舰数量,看是否能放下k个战舰,如果放不下就证明找到答案了

代码

#include <cstdio>
#include <cstring>
#include <cctype>
#include <stdlib.h>
#include <string>
#include <map>
#include <iostream>
#include <sstream>
#include <set>
#include <stack>
#include <cmath>
#include <queue>
#include <vector>
#include <algorithm>
#include<list>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define inf 0x3f3f3f3f
typedef long long ll;
set<int>s;
int n,k,a,x,m;
int f(int q)
{
    return (q+1)/(a+1);
}
int main()
{
    int ans=-1;
    scanf("%d%d%d",&n,&k,&a);
    scanf("%d",&m);
    int len=(n+1)/(a+1);
    s.insert(0);
    s.insert(n+1);
    int flag=0;
    if(len<k)
    {
        flag=1;
        ans=1;
    }
    for(int i=1; i<=m; i++)
    {
        scanf("%d",&x);
        if(flag) continue;
        auto it=s.upper_bound(x);
        auto it2=s.lower_bound(x);
        it2--;
        int a1=f((*it)-(*it2)-1);
        int a2=f((*it)-x-1);
        int a3=f(x-(*it2)-1);
        len=len-(a1-a2-a3);
        if(len<k)
        {
            flag=1;
            ans=i;
        }
        s.insert(x);
    }
    printf("%d\n",ans);
    return 0;
}
/*
14 3 3
5
4 5 6 12 1
*/

猜你喜欢

转载自blog.csdn.net/riba2534/article/details/80159966