51nod1287 加农炮 线段树

Description

---

一个长度为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行，每行一个数，对应最终的地形高度。

Solution

---

刷水题有益身心健康，要是这题还能交我就前五十了

一个很显然的做法就是线段树维护区间最大值，这样是最好想的

另一个思路就是在原高度数组中找一个最长上升的子序列，然后每次二分，这样比较好写

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define max(x,y) ((x)>(y)?(x):(y))
#define N 60005
int mx[N<<2|1];
int n,m,st;
int read() {
    int x=0; char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x;
}
void buildTree(int now,int l,int r) {
    if (l==r) {
        mx[now]=read();
        if (l==1) st=now;
        return ;
    }
    int mid=(l+r)>>1;
    buildTree(now<<1,l,mid);
    buildTree(now<<1|1,mid+1,r);
    mx[now]=max(mx[now<<1],mx[now<<1|1]);
}
void writeln(int now,int l,int r) {
    if (l==r) {
        printf("%d\n",mx[now]);
        return ;
    }
    int mid=(l+r)>>1;
    writeln(now<<1,l,mid);
    writeln(now<<1|1,mid+1,r);
}
void modify(int now,int tl,int tr,int x) {
    if (tl==tr) {
        ++mx[now];
        return ;
    }
    int mid=(tl+tr)>>1;
    if (x<=mid) modify(now<<1,tl,mid,x);
    else modify(now<<1|1,mid+1,tr,x);
    mx[now]=max(mx[now<<1],mx[now<<1|1]);
}
int query(int now,int tl,int tr,int v) {
    if (tl==tr) return tl;
    int mid=(tl+tr)>>1;
    if (mx[now<<1]>=v) return query(now<<1,tl,mid,v);
    return query(now<<1|1,mid+1,tr,v);
}
int main(void) {
    n=read();
    m=read();
    buildTree(1,1,n);
    rep(i,1,m) {
        int x=read();
        if (x>mx[1]||x<=mx[st]) continue;
        int pos=query(1,1,n,x);
        modify(1,1,n,pos-1);
    }
    writeln(1,1,n);
    return 0;
}