一维战舰 51Nod - 1521 【set二分】

题目来源:  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

思路:

可以用一个set容器存储每次的miss位置,用一个sum记录当前最多能放多少战舰。每插入一个值,都会使一个区间一分为二,更新sum,直到sum<k。可以先插入0和n+1。需要注意的是战舰不能相邻,即中间必须空一格。考虑边界的情况,每次计算sum时需要(R-L)/(a+1)。可以用二分查找。

并且用vector+lower_bound会超时,set不会。

代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<stdlib.h>
#include<map>
#include<iostream>
#include<set>
using namespace std;
typedef long long ll;
int n,k,r;
set<int> vec;
set<int>::iterator it;
int main()
{
    while(~scanf("%d%d%d",&n,&k,&r))
    {
        vec.clear();
        int sum=(n+1)/(r+1),ans=-1;
        vec.insert(0);
        vec.insert(n+1);
        int m;
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            int d;
            scanf("%d",&d);
            it=vec.lower_bound(d);
            if(ans!=-1 || *it==d) continue;
            int rr=*it;
            it--;
            int lr=*it;
            sum-=(rr-lr)/(r+1);
            sum+=(rr-d)/(r+1);
            sum+=(d-lr)/(r+1);
            vec.insert(it,d);
            if(sum<k && ans==-1) ans=i;
        }
        printf("%d\n",ans);
    }
    return 0;
}
/*
5000 1660 2
20
1 100 18 102 300 81 19 25 44 88 1337 4999 1054 1203 91 16 164 914 1419 1487
*/

猜你喜欢

转载自blog.csdn.net/u013852115/article/details/80269851
今日推荐