首页 > 代码库 > 1287 加农炮
1287 加农炮
1287 加农炮
一个长度为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
思路:我们用线段树,如果该区间有比炮弹高度大的,那么我们先去看左区间最大值,没有再去右区间
1 #include<bits/stdc++.h> 2 using namespace std; 3 4 int n,m; 5 int a[50004],x; 6 int b[50004]; 7 8 struct node{ 9 int l,r,mx; 10 }c[50004*4]; 11 12 void build(int x,int L,int R){ 13 c[x].l=L; 14 c[x].r=R; 15 int mid=(L+R)>>1; 16 if(L==R){ 17 c[x].mx=a[L];return ; 18 } 19 build(2*x,L,mid);build(2*x+1,mid+1,R); 20 c[x].mx=max(c[2*x].mx,c[2*x+1].mx); 21 } 22 23 int query(int x,int xx){ 24 if(c[x].l==c[x].r){ 25 if(c[x].mx>=xx) return c[x].l-1; 26 } 27 // cout<<c[2*x].mx<<" "<<c[2*x+1].mx<<endl; 28 if(c[2*x].mx>=xx) query(2*x,xx); 29 else if(c[2*x+1].mx>=xx) query(2*x+1,xx); 30 else return 0; 31 } 32 33 void update(int x,int xx){ 34 if(c[x].l==xx&&c[x].r==xx){ 35 36 c[x].mx++;return ; 37 } 38 int mid=(c[x].l+c[x].r)>>1; 39 40 if(xx<=mid) update(2*x,xx); 41 else update(2*x+1,xx); 42 c[x].mx=max(c[2*x].mx,c[2*x+1].mx); 43 } 44 45 void hh(int x){ 46 if(c[x].l==c[x].r){ 47 b[c[x].l]=c[x].mx;return ; 48 } 49 hh(x*2);hh(2*x+1); 50 } 51 int main(){ 52 scanf("%d%d",&n,&m); 53 for(int i=1;i<=n;i++) scanf("%d",&a[i]); 54 build(1,1,n); 55 for(int i=1;i<=m;i++){ 56 scanf("%d",&x); 57 int y=query(1,x); 58 59 if(y!=0) 60 update(1,y); 61 } 62 hh(1); 63 for(int i=1;i<=n;i++) printf("%d\n",b[i]); 64 }
1287 加农炮
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。