爱丽丝和鲍博喜欢玩一维战舰的游戏。他们在一行有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 */