【Noip2013】積木遊戲
描述:
春春幼兒園舉辦了一年一度的“積木大賽”。今年比賽的內容是搭建一座寬度爲 n 的大廈,大廈可以看成由 n 塊寬度爲1的積木組成,第��塊積木的最終高度需要是hi。
在搭建開始之前,沒有任何積木(可以看成 n 塊高度爲 0 的積木)。接下來每次操作,小朋友們可以選擇一段連續區間[L,R],然後將第 L 塊到第 R 塊之間(含第 L 塊和第 R 塊)所有積木的高度分別增加1。
小 M 是個聰明的小朋友,她很快想出了建造大廈的最佳策略,使得建造所需的操作次數最少。但她不是一個勤於動手的孩子,所以想請你幫忙實現這個策略,並求出最少的操作次數。
輸入:
輸入包含兩行,第一行包含一個整數 n,表示大廈的寬度。
第二行包含 n 個整數,第i個整數爲hi。
輸出:
僅一行,即建造所需的最少操作數。
題解:
單調隊列可以做。這裏窩用了棧,每次輸入hi就把它和棧頂比較(爲了防止出錯可以先push一個-1進棧)。如果hi大,push進棧;如果hi小於或等於棧頂,記錄爲a,循環彈出棧頂元素,直到新的棧頂元素小於hi,答案加上a-hi。輸入結束後注意,如果棧頂不爲-1,答案加上棧頂。
#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;
int n;
stack<int> q;
int main(){
int ain;
q.push(-1);
scanf("%d",&n);
int a;
int cnt=0;
for(int i=1;i<=n;i++){
scanf("%d",&ain);
a=q.top();
if(ain>a){
q.push(ain);
continue;
}
cnt+=a-ain;
while(ain<=a){
q.pop();
a=q.top();
}
q.push(ain);
}
a=q.top();
if(a!=-1) cnt+=a;
printf("%d",cnt);
return 0;
}