一个长度为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