51Nod 1287 加农炮 （线段数单点更新单点查询）

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=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

暴力可以过 不过用线段数能够满足题目所给数据范围

单点更新单点查询 记录线段里的最大值

#include <bits/stdc++.h>
using namespace std;
#define maxn 50000+5
struct xjy
{
    int num;
    int left;
    int right;
    int mmax;
    int pre;
};
xjy tree[maxn<<2];
int a[maxn];
int maxh=0;
void build(int i,int left ,int right)
{
    if(left==right)
    {
        tree[i].num=a[left];
        tree[i].left=left;
        tree[i].right=right;
        tree[i].mmax=tree[i].num;
        tree[i].pre=left-1;
        return ;
    }
    int mid=(left+right)>>1;
    build(i<<1,left,mid);
    build (i<<1|1,mid+1,right);
    tree[i].mmax=max(tree[i<<1].mmax,tree[i<<1|1].mmax);
    tree[i].left=left;
    tree[i].right=right;
}
void updatepoint(int i,int pos,int val)
{
    if(pos==tree[i].left&&pos==tree[i].right)
    {
        tree[i].num+=val;
        tree[i].mmax+=val;
        maxh=max(maxh,tree[i].num);
        return ;
    }
    int mid=(tree[i].left+tree[i].right)>>1;
    if(pos<=mid)
        updatepoint(i<<1,pos,val);
    else
        updatepoint(i<<1|1,pos,val);
    tree[i].mmax=max(tree[i<<1].mmax,tree[i<<1|1].mmax);
}
void queryregion(int i,int val)
{
    if(tree[i].left==tree[i].right&&tree[i].num>=val&&tree[i].pre!=0)
    {
        updatepoint(1,tree[i].pre,1);
        return;
    }
    if(tree[i].left==tree[i].right)
        return;
    if(val<=tree[i<<1].mmax)
        queryregion(i<<1,val);
    else
        queryregion(i<<1|1,val);
    tree[i].mmax=max(tree[i<<1].mmax,tree[i<<1|1].mmax);
}

void ans(int i)
{
    if(tree[i].left==tree[i].right)
    {
        printf("%d\n",tree[i].num);
        return ;
    }
    int mid=(tree[i].left+tree[i].right)>>1;
    ans(i<<1);
    ans(i<<1|1);
}

int main()
{
    int n=0,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<maxn;i++)
    {
        tree[i].left=0;
        tree[i].num=0;
        tree[i].right=0;
        a[i]=0;
    }
    for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            maxh=max(maxh,a[i]);
        }
    build(1,1,n);
    for(int i=0;i<m;i++)
    {
        int mid;
        scanf("%d",&mid);
        if(mid>maxh)
            continue;
        queryregion(1,mid);
    }
    ans(1);

}