【Noip2013】積木遊戲

【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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章