51nod-1287 加农炮

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

思路：可用线段树维护区间最大值，对于B[i],用二分查找找到要修改的点坐标，修改其值即可。

Code :

#include<iostream>
using namespace std;

const int MAX_N=50005;
int n,m;
int a[MAX_N];
int tree[MAX_N*4];

void PushUp(int id);
void Build(int l,int r,int id);
void Update(int t,int x,int L,int R,int id);
int Query(int l,int r,int L,int R,int id);
int Lower_bound(int l,int r,int x);
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;++i)
		cin>>a[i];
	Build(1,n,1);
	for(int i=0,x;i<m;++i)
	{
		cin>>x;
		int k=Lower_bound(1,n,x);
		if(k>=1&&k!=n)	Update(k,a[k]+1,1,n,1);
	}
	for(int i=1;i<=n;++i)
		cout<<a[i]<<endl;
	return 0;
}

void PushUp(int id)
{
	tree[id]=max(tree[id<<1],tree[id<<1|1]);
}

void Build(int l,int r,int id)
{
	if(l==r){
		tree[id]=a[r];
		return;
	}
	int h=(l+r)>>1;
	Build(l,h,id<<1);
	Build(h+1,r,id<<1|1);
	PushUp(id);
}

void Update(int t,int x,int l,int r,int id)
{
	if(l==r){
		tree[id]=a[t]=x;
		return;
	}
	int h=(l+r)>>1;
	if(t<=h){
		Update(t,x,l,h,id<<1);
	}else	Update(t,x,h+1,r,id<<1|1);
	PushUp(id);
}

int Query(int l,int r,int L,int R,int id)
{
	if(l<=L&&r>=R){
		return tree[id];
	}
	int h=(L+R)>>1;
	int s1=0,s2=0;
	if(h>=l)	s1=Query(l,r,L,h,id<<1);
	if(h+1<=r)	s2=Query(l,r,h+1,R,id<<1|1);
	return max(s1,s2);
}

int Lower_bound(int l,int r,int x)
{
	while(l<=r){
		int h=(l+r)>>1;
		if(Query(1,h,1,n,1)>=x)	r=h-1;
		else	l=h+1;
	}
	return l-1;
}s