Algoogle

Algorithm for Programming Contest

PKU 3044 City Skyline

Category: PKU Tag: data-structure, stack

City Skyline

問題概要


左側から順に, 影の高さが変化する位置と変化後の高さが与えられる.
推測される建物の数の最小数を求めよ

解法


基本的に高さが増える場合に答えが1増える.
スタックを利用する.
スタックに高さが増える場合積む. その際に答えを1増やす.
高さが減る場合はスタックトップがその高さより高い限りポップする.
ポップした結果, 減った後の高さがスタックトップより高ければスタックに積む. その際に答えを1増やす.

####


(3044.cpp) download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#include <cstdio>
#include <stack>

using namespace std;

int N, W;

#include <iostream>
int main(){
    scanf("%d%d", &N, &W);
    stack<int> st;
    int ans = 0;
    for(int i = 0; i < N; i++){
        int x, y; scanf("%d%d", &x, &y);
        if(st.empty()){
            if(y > 0){
                ans++;
                st.push(y);
            }
        }
        else if(y > st.top()){
            st.push(y);
            ans++;
        }
        else if(y < st.top()){
            while(!st.empty() and st.top() > y)
                st.pop();
            if(y > 0 and (st.empty() or st.top() < y)){
                st.push(y);
                ans++;
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}

Comments