51nod 1287: 加农炮 好题啊好题

题目来源: Codility
基准时间限制:1 秒 空间限制:131072 KB 分值: 40 难度:4级算法题
 收藏
 关注
一个长度为M的正整数数组A,表示从左向右的地形高度。测试一种加农炮,炮弹平行于地面从左向右飞行,高度为H,如果某处地形的高度大于等于炮弹飞行的高度H(A[i] >= H),炮弹会被挡住并落在i - 1处,则A[i - 1] + 1。如果H <= A[0],则这个炮弹无效,如果H > 所有的A[i],这个炮弹也无效。现在给定N个整数的数组B代表炮弹高度,计算出最后地形的样子。
例如:地形高度A = {1, 2, 0, 4, 3, 2, 1, 5, 7}, 炮弹高度B = {2, 8, 0, 7, 6, 5, 3, 4, 5, 6, 5},最终得到的地形高度为:{2, 2, 2, 4, 3, 3, 5, 6, 7}。
Input
第1行:2个数M, N中间用空格分隔,分别为数组A和B的长度(1 <= m, n <= 50000)
第2至M + 1行:每行1个数,表示对应的地形高度(0 <= A[i] <= 1000000)。
第M + 2至N + M + 1行,每行1个数,表示炮弹的高度(0 <= B[i] <= 1000000)。
Output
输出共M行,每行一个数,对应最终的地形高度。
Input示例
9 11
1
2
0
4
3
2
1
5
7
2
8
0
7
6
5
3
4
5
6
5
Output示例
2
2
2
4
3
3
5
6
7

最开始自己的思路是用刚学的线段树来做,但是问题在于更新这块自己还不是很熟练,就是一个炮弹打过来,某地高度+1,这样的话我自己现在的方法只能是更新整个的线段树,一直到最小节点上。所以估计时间会超时,这个思路out。

然后自己的思路是记录整个地方的高地。就是 {1, 2, 0, 4, 3, 2, 1, 5, 7},记录1 2 4 5 7。然后炮弹打过来不断更新这个高地数组。过了19个用例,其中一个死活过不了。。。后来想自己更新这个也不对啊,中间会产生高地,比方说2 2 5,一个炮弹4的打过来,高地原来是2 5,现在就得是 2 3 5了。。。还是麻烦。

最后,还是老实用二分找炮弹打过来的高地,将其之前的土地高度+1,然后更新高地高度。

代码:

#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#pragma warning(disable:4996)
using namespace std;

int canno[1000003];
int highet[1000003];
int A,B;

int main()
{
	int i,j,h_max,gun,pos;
	memset(canno,0,sizeof(canno));

	scanf("%d%d",&A,&B);
	h_max=-1;
	for(i=1;i<=A;i++)
	{
		scanf("%d",highet+i);
		h_max=max(h_max,highet[i]);
		canno[i]=h_max;
	}
	int count=0;
	for(i=1;i<=B;i++)
	{
		scanf("%d",&gun);
		if(gun<=canno[0]||gun>canno[A])
			continue;
		pos=lower_bound(canno+1,canno+1+A,gun)-canno;
		highet[pos-1]++;
		canno[pos-1]=max(canno[pos-1],highet[pos-1]);
	}
	for(i=1;i<=A;i++)
	{
		printf("%d
",highet[i]);
	}
	return 0;
}





版权声明:本文为博主原创文章,未经博主允许不得转载。

原文地址:https://www.cnblogs.com/lightspeedsmallson/p/4899563.html