51nod 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

思路：线段树+二分，利用线段树存储每一个区间内的山峰最高值，通过二分查找找出高度大于等于当前炮弹的山峰下标k，将k-1位置的山峰高度+1即可

 #include<iostream>
 #include<algorithm>
 #include<cstring>
 #include<cstdio>

 using namespace std;
 //typedef long long LL;
 const int maxn = ;
 int a[maxn],rs[maxn<<];
 int n,m;
 void pushup(int rt){
     rs[rt] = max(rs[rt<<],rs[rt<<|]);
 }
 void build(int l,int r,int rt)
 {
     if(l==r){
         rs[rt] = a[l];
         return;
     }
     int mid=(l+r)>>;
     build(l,mid,rt<<);
     build(mid+,r,rt<<|);
     pushup(rt);
 }
 void update(int x,int l,int r,int rt)
 {
     if(l==r){
         rs[rt]++;
         return;
     }
     int mid=(l+r)>>;
     if(x<=mid) update(x,l,mid,rt<<);
     else update(x,mid+,r,rt<<|);
     pushup(rt);
 }
 int query(int x,int l,int r,int rt)//根据x的值找出从左往右找出第一个能碰到的山峰的下标
 {
     if(l==r) return l;
     int mid = (l+r)>>;
     if(x<=rs[rt<<]) query(x,l,mid,rt<<);
     else query(x,mid+,r,rt<<|);
 }
 int Query(int x,int l,int r,int rt)//根据点查询数据
 {
     if(l==r)return rs[rt];
     int mid = (l+r)>>;
     if(x<=mid)Query(x,l,mid,rt<<);
     else Query(x,mid+,r,rt<<|);
 }
 int main()
 {
     ios::sync_with_stdio(false);
 //    freopen("in.txt","r",stdin);
     while(cin>>n>>m){
         int x;
         memset(rs,,sizeof(rs));
         for(int i=;i<=n;i++)
             cin>>a[i];
         build(,n,);
         for(int i=;i<=m;i++){
             cin>>x;
 //            if(x>rs[1]||x<=a[1])continue;一开始在这里卡了好久 后头想起来第一个位置的高度可能也会更新所以每一次都需要查询
             if(x>rs[]||x<=Query(,,n,))continue;
             int k=query(x,,n,);
             update(k-,,n,);//更新前一个点的高度
         }
         for(int i=;i<=n;i++)
             cout<<Query(i,,n,)<<endl;
     }
     return ;
 }