一个长度为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
#include <iostream>
#include <cstdio>
using namespace std;
const int Length = 50005;
typedef struct node
{
int left,right,maxn;
}Tree;
Tree tree[Length*4];
int p,ans;
int addr[Length];
int f_max(int a,int b)
{
return a>b?a:b;
}
void BTree(int left,int right,int rt)
{
tree[rt].left = left;
tree[rt].right = right;
if(left == right)
{
addr[left] = rt;
scanf("%d",&tree[rt].maxn);
return;
}
int mid = (tree[rt].left + tree[rt].right) >> 1;
BTree(left,mid,rt<<1);
BTree(mid+1,right,rt<<1|1);
tree[rt].maxn = f_max(tree[rt<<1].maxn,tree[rt<<1|1].maxn);
}
void Query(int left,int right,int rt,int v)
{
if(left == right)
{
if(v>tree[rt].maxn)
{
p = left;
ans = tree[rt].maxn + 1;
}
return;
}
if(v>tree[rt<<1].maxn)
{
p = tree[rt<<1].right;
ans = tree[addr[p]].maxn + 1;
Query(tree[rt<<1|1].left,tree[rt<<1|1].right,rt<<1|1,v);
}
else
Query(tree[rt<<1].left,tree[rt<<1].right,rt<<1,v);
}
void Update(int left,int right,int rt,int pos,int v)
{
if(left == right)
{
tree[rt].maxn = v;
return;
}
int mid = (tree[rt].left + tree[rt].right) >> 1;
if(pos>mid)
Update(mid+1,right,rt<<1|1,pos,v);
else
Update(left,mid,rt<<1,pos,v);
tree[rt].maxn = f_max(tree[rt<<1].maxn,tree[rt<<1|1].maxn);
}
void Print(int n)
{
for(int i=1;i<=n;i++)
printf("%d\n",tree[addr[i]].maxn);
}
int main()
{
int n,m,h;
scanf("%d%d",&n,&m);
BTree(1,n,1);
while(m--)
{
scanf("%d",&h);
if(h>tree[addr[1]].maxn&&h<=tree[1].maxn)
{
Query(1,n,1,h);
Update(1,n,1,p,ans);
}
}
Print(n);
return 0;
}