1. 程式人生 > >51nod 1287: 加農炮 好題啊好題

51nod 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,這樣的話我自己現在的方法只能是更新整個的線段樹,一直到最小節點上。所以估計時間會超時,這個思路out。

然後自己的思路是記錄整個地方的高地。就是 {1, 2, 0, 4, 3, 2, 1, 5, 7},記錄1 2 4 5 7。然後炮彈打過來不斷更新這個高地陣列。過了19個用例,其中一個死活過不了。。。後來想自己更新這個也不對啊,中間會產生高地,比方說2 2 5,一個炮彈4的打過來,高地原來是2 5,現在就得是 2 3 5了。。。還是麻煩。

最後,還是老實用二分找炮彈打過來的高地,將其之前的土地高度+1,然後更新高地高度。

程式碼:

#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#pragma warning(disable:4996)
using namespace std;

int canno[1000003];
int highet[1000003];
int A,B;

int main()
{
	int i,j,h_max,gun,pos;
	memset(canno,0,sizeof(canno));

	scanf("%d%d",&A,&B);
	h_max=-1;
	for(i=1;i<=A;i++)
	{
		scanf("%d",highet+i);
		h_max=max(h_max,highet[i]);
		canno[i]=h_max;
	}
	int count=0;
	for(i=1;i<=B;i++)
	{
		scanf("%d",&gun);
		if(gun<=canno[0]||gun>canno[A])
			continue;
		pos=lower_bound(canno+1,canno+1+A,gun)-canno;
		highet[pos-1]++;
		canno[pos-1]=max(canno[pos-1],highet[pos-1]);
	}
	for(i=1;i<=A;i++)
	{
		printf("%d\n",highet[i]);
	}
	return 0;
}