加农炮

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

刚看到题的时候真的被题目上面的话题吓到了，什么线段树啊，二分啊，于是嫌麻烦缓缓地打了一波O（N*M）的暴力，我擦竟然水过了，代码如下，只能抱怨数据弱了呵呵呵

#include
      
      
       
       
#define N 50005
using namespace std;
int a[N],b[N];
int main(){
	int m,n,max=0;
	scanf("%d%d",&m,&n);
	for (int i=1;i<=m;i++){
		scanf("%d",&a[i]);
		max=a[i]>max?a[i]:max;
	}
	a[0]=0;
	for (int i=1;i<=n;i++){
		int b,j;
		scanf("%d",&b);
		if (b<=a[0] || b>max) continue;
		for (j=1;j<=m;j++)
		  if (a[j]>=b) break;
		a[j-1]++;
		max=a[j-1]>max?a[j-1]:max;
	}
	for (int i=1;i<=m;i++) printf("%d\n",a[i]);
	return 0;
}
      
      

不过后来转念一想觉得这样做有点不道德，于是开始寻找线段树的解法。首先对于每一颗炮弹的读入肯定是优化不了的，O（M）一定有，所以我们只能在查找地形的过程中进行优化，O（log N）貌似是个不错的选择。因为地形无序，且给他排序会影响答案，所以直接二分是不可取的，然而题目要求的是第一个大于等于该炮弹的地形，于是自然而然想到线段树，代码如下（懒得打lazy标记了，表示很懒，不过对于这种题加不加影响不大，毕竟暴力和正解时间只差了100多ms，可见数据有多水）

   
   

    
    
#include
      
      
       
       
#define N 50010 
using namespace std;
int tree[4*N],a[N];
void build(int p,int l,int r){
	if (l==r){
		tree[p]=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	tree[p]=tree[p*2]>tree[p*2+1]?tree[p*2]:tree[p*2+1];
}
int search(int p,int l,int r,int k){
	if (l==r) return l;
	int mid=(l+r)>>1;
	if (tree[p*2]>=k) return search(p*2,l,mid,k);
	return search(p*2+1,mid+1,r,k);
}
void update(int p,int l,int r,int k){
	if (l==r){
		tree[p]=a[k];
		return;
	}
	int mid=(l+r)>>1;
	if (k<=mid) update(p*2,l,mid,k); else update(p*2+1,mid+1,r,k);
	tree[p]=tree[p*2]>tree[p*2+1]?tree[p*2]:tree[p*2+1];
}
int main(){
	int m,n;
	scanf("%d%d",&m,&n);
	for (int i=1;i<=m;i++) scanf("%d",&a[i]);
	build(1,1,m);
	for (int i=1;i<=n;i++){
		int b;
		scanf("%d",&b);
		if (b<=a[1] || b>tree[1]) continue;
		int k=search(1,1,m,b);
		a[k-1]++;
		update(1,1,m,k-1);
	}
	for (int i=1;i<=m;i++) printf("%d\n",a[i]);
}