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