51nod 1521 一维战舰

1521 一维战舰

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

题解：这一题主要是做减法。 对于刚开始n个格子，我们知道最多可以放 Max = (n+1)/(a+1) 个战舰。下面每填进一个

temp位置不能放战舰，我们就需要计算填进来的这个位置造成了多少个可填放的战舰数的减少。 也就是 

num = (right-left)/(a+1) - (right-temp)/(a+1) - (temp-left)/(a+1) 。（left，right表示填数分割出来的区间）当前能放的战

舰最大数就是 Max =  Max - num。 当Max<k时就表示放不下了。

在这个题目的实现上面有很多方式，set+二分，数组+二分，直接用数组找边界。 重要的还是理解上面的做减法的思想。

代码如下：

#include <cstdio>
#include <cstring>
#include <set>
#include <algorithm>
using namespace std;
const int maxn = 2e5+10;
bool flag[maxn];
set<int> s;
int main()
{
	int n,k,a,m;
	while(scanf("%d%d%d%d",&n,&k,&a,&m)!=EOF)
	{
		int Max = (n+1)/(a+1);
		memset(flag,0,sizeof(flag));
		s.clear();
		s.insert(0);//注意添加边界 
		s.insert(n+1);
		int temp;
		for(int i=1;i<=m;++i)
		{
			scanf("%d",&temp);
			if(flag[temp])
				continue;
			flag[temp]=true;
			s.insert(temp);
			set<int>::iterator left = s.find(temp);
			left--; 
			set<int>::iterator right = s.upper_bound(temp);  
			Max -= (*right - *left)/(a+1) - (*right - temp)/(a+1) - (temp - *left)/(a+1);
			if(Max<k && Max>=0)
			{
				printf("%d\n",i);
				Max=-1;
			}
		}
		if(Max>=k)
			printf("-1\n");
	}
	return 0;
}