题目来源: 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有这么方便的用法,以后处理类似的区间问题可以考虑下用set,就不用闭门造车啦
#include<bits/stdc++.h>
using namespace std;
int main()
{
int n,k,a;
scanf("%d%d%d",&n,&k,&a);
int m;
scanf("%d",&m);
set<int>st;
st.insert(0);st.insert(n+1);
int ans=-1;int total=(n+1)/(a+1);//初始时能放入的战舰数量
int pos;int left,right;
for(int i=1;i<=m;i++)
{
scanf("%d",&pos);
st.insert(pos);
left=*--st.find(pos);
right=*++st.find(pos);
total-=((pos-left-1)+(right-pos-1)+2)/(a+1)-((pos-left)/(a+1)+(right-pos)/(a+1));
if(total<k){ans=i;break;}
}
printf("%d
",ans);
return 0;
}