加农炮

一个长度为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<algorithm>
#include<stdio.h>
using namespace std;
const int Maxn=50005;
int a[Maxn],b[Maxn],c[Maxn*20];
int main(){
    int n,m;scanf("%d%d",&n,&m);
    int Max=0;
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    for(int j=0;j<m;j++) scanf("%d",&b[j]),Max=max(Max,b[j]);
    int j=0;
    for(int i=1;i<=Max;i++){
        while(a[j]<i&&j<n) j++;
        c[i]=j;
    }
    for(int i=0;i<m;i++){
        int index=c[b[i]]-1;
        if(index<0||index>=n-1) continue;
        a[index]++;
        if(c[a[index]]>index) c[a[index]]=index;
    }
    for(int i=0;i<n;i++) printf("%d\n",a[i]);
}
自己用线段树做：
#include<iostream>
#include<stdio.h>
#include<algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define root 1,n,1
using namespace std;
const int Maxn=50005;
int Max[Maxn<<2],ans[Maxn];
void PushUp(int rt){
    Max[rt]=max(Max[rt<<1],Max[rt<<1|1]);
}
void Build(int l,int r,int rt){
    if(l==r){
        scanf("%d",&Max[rt]);
        ans[l]=Max[rt];
        return;
    }
    int m=(l+r)>>1;
    Build(lson);
    Build(rson);
    PushUp(rt);
}
void Updata(int p,int add,int l,int r,int rt){
    if(l==r){
        Max[rt]+=add;
        ans[l]+=add;
        return;
    }
    int m=(l+r)>>1;
    if(p<=m)
        Updata(p,add,lson);
    else
        Updata(p,add,rson);
    PushUp(rt);
}
int Query(int val,int l,int r,int rt){
    if(l==r) return l;
    int m=(l+r)>>1;
    if(Max[rt<<1]>=val) return Query(val,lson);
    return Query(val,rson);
}
int main(){
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        Build(root);
        for(int h,i=0;i<m;i++){
            scanf("%d",&h);
            if(Max[1]<h) continue;
            int index=Query(h,root);
            if(index==1) continue;
            Updata(index-1,1,root);
        }
        for(int i=1;i<=n;i++){
            printf("%d\n",ans[i]);
        }
    }
}