51Nod 1287 加农炮

Problem

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

Solution

1e6数组存每个高度的炮弹能打到的位置，每次更新。

另一种是维护前缀最大值，因为每次前一个+1之后仍然<=后一个，O（1）维护即可。

Code

#include<stdio.h>
#include<set>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#define mem(ss) memset(ss,0,sizeof(ss))
#define rep(d, s, t) for(int d=s;d<=t;d++)
#define rev(d, s, t) for(int d=s;d>=t;d--)
typedef long long ll;
typedef long double ld;
typedef double db;
const ll mod = 1e9+7;
const int N = 1e4 + 10;
#define io_opt ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
const int INF = 1e8;
using namespace std;
int m,n;
int a[50020],b[50020];
int t[1000020];
int main() {
    io_opt;
    cin>>m>>n;
    for(int i=1;i<=m;i++){
        cin>>a[i];
        for(int j=a[i];j>=0&&t[j]==0;j--){
            t[j]=i;
        }
    }
    for(int i=1;i<=n;i++){
        cin>>b[i];
        if(b[i]>a[1]){
            a[t[b[i]]-1]++;
            if(t[b[i]]-1<t[a[t[b[i]]-1]]){
                t[a[t[b[i]]-1]]=t[b[i]]-1;
            }
        }
    }
    for(int i=1;i<=m;i++){
        cout<<a[i]<<endl;
    }
    return 0;
}