51nod 1287 加农炮

雨点打透心脏的1/2处 2022-06-09 11:17 190阅读 0赞

[1287 加农炮][1287] ![ok.png][]

题目来源： [Codility][]

基准时间限制：1 秒 空间限制：131072 KB 分值: 40 [难度：4级算法题][4]

![star.png][] 收藏

![plus.png][] 关注

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

[1287]: https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1287
[ok.png]: https://file.51nod.com/images/icon/ok.png
[Codility]: https://codility.com/
[4]: https://www.51nod.com/onlineJudge/problemList.html#!groupId=5
[star.png]: https://file.51nod.com/images/icon/star.png
[plus.png]: https://file.51nod.com/images/icon/plus.png