2021-05-09

题目

一个长度为M的正整数数组A,表示从左向右的地形高度。测试一种加农炮,炮弹平行于地面从左向右飞行,高度为H,如果某处地形的高度大于等于炮弹飞行的高度H(Ai >= H),炮弹会被挡住并落在i - 1处,则Ai−1 + 1。如果H <= A0,则这个炮弹无效,如果H > 所有的Ai,这个炮弹也无效。现在给定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}。

输入格式

第1行:2个数M, N中间用空格分隔,分别为数组A和B的长度(1 <= m, n <= 50000) 第2至M + 1行:每行1个数,表示对应的地形高度(0 <= Ai <= 1000000)。 第M + 2至N + M + 1行,每行1个数,表示炮弹的高度(0 <= Bi <= 1000000)。

输出格式

输出共M行,每行一个数,对应最终的地形高度。

数据范围

样例输入

9 11
1
2
0
4
3
2
1
5
7
2
8
0
7
6
5
3
4
5
6
5

样例输出

2
2
2
4
3
3
5
6
7

题意

地形高度各不同,如果炮弹小于第一个地形高度,则是无效炮弹,大于所有地形的高度也是无效炮弹,找到大于等于炮弹的第一个数的位置i,则地形i-1的位置高度加一

思路

  • 1.用一个数组 pre 记录前缀地形的最大高度,对于每个炮弹,二分找到大于在炮弹高度的最小值 pre[i],把 i−1 位置的地形 +1。由于每次只发射 1 个炮弹,因此维护 pre 数组是 O(1) 的。这样就可以在 O(mlogn) 的时间内解决这个问题。

坑点

  • 1.暴力会超时,二分是个好东西

代码

#include<bitsdc++.h>
using namespace std;
int main()
{
    
    
	int n,m;
	cin>>n>>m;
	int num[50004]={
    
    0};
	int sum[50004]={
    
    0};
	for(int i=1;i<=n;i++)
	{
    
    
		cin>>num[i];
		sum[i]=max(num[i],sum[i-1]);
	}
	int x;
	for(int k=1;k<=m;k++)
	{
    
    
		cin>>x;
		int mid=lower_bound(sum+1,sum+n+1,x)-sum;
		if(mid>n||mid<=1)
		{
    
    
			continue;
		}
		num[mid-1]++;
		//因为炮弹肯定会落在所有相同数字的最后一个,下一个就是比这个大的地形,
		//而且炮弹的高度只有1,所以这样处理不会影响序列的单调性 
		if(num[mid-1]>sum[mid-1])
		{
    
    
			sum[mid-1]=num[mid-1];
		}
	}
	for(int i=1;i<=n;i++)
	{
    
    
		cout<<num[i];
		if(i!=n)
		{
    
    
			cout<<endl;
		}
	}
	return 0;
 }

总结

熟练掌握二分技巧,不要被题的外表所迷惑,透过现象看本质

猜你喜欢

转载自blog.csdn.net/Autumn_snow/article/details/116561068