C++ 스택 자료구조 문제풀이

 

top = -1로 시작

 

top = -1;
void push(int x) {
	stack[++top] = x;
}

int pop() {
    return stack[top--];
}

#include<stack> 하면 해당 함수를 사용할 수 있다.

 

 

문제 : K진수 출력

10진수 N이 입력되면 K진수로 변환하여 출력하는 프로그램 작성

(10<=N<=1,000) 과 K(2, 5, 8, 16)

 

ex) 11 2 -> 1011

 

my그림 == 예술의 경지

스택의 push, pop 이용하면 1011 나옴

 

코드를 작성해보자

 

#include<stdio.h>
#include<vector>
#include<stack>
#include<algorithm>

using namespace std;

int main() {
    int n, k;
    stack<int> s;
    char str[20] = "0123456789ABCDEF";
    scanf("%d %d", &n, &k);
    
    while(n>0) {
        s.push(n % k);
        n = n / k;
    }
    while(!s.empty()) {
        printf("%c", str[s.top()]);
        s.pop();
    }
    return 0;
}

 

 

문제 : 올바른 괄호

 

괄호가 입력되면 올바른 괄호이면 "YES", 올바르지 않으면 "NO"를 출력한다.

(())()는 YES, (()()))는 NO

스택을 이용해서 문제풀이

 

#include<iostream>
#include<string>
#include<stack>
#include<algorithm>

using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    stack<char> s;
    string b;
    cin >> b;
    bool ending = false;
    
    for(int i=0; i<b.size(); i++) {
        if (b[i] == '(') s.push('(');
        else {
            if(s.empty()) {
                printf("NO\n");
                ending = true;
                break;
            }
            else s.pop();
        }
    }
    
    if (ending == false) {
        if(s.empty()) printf("YES\n");
        else printf("NO\n");
    }

    return 0;
}

 

 

 

문제 : 기차운행 (stack 응용)

 

A도시에서 출발한 기차는 B도시로 도착한다.

그런데 도로 중간에 T자형 교차로가 있어 출발한 기차의 도착 순서를 조정할 수 있다.

 

교차로에서는 다음과 같은 두 개의 작업을 한다.

P(push)작업 : A도시에서 오는 기차를 교차로에 넣는다.

O(out)작업 : 교차로에 들어온 가장 최근 기차를 B도시로 보낸다.

만약 2 1 3 기차 번호 순으로 A도시에서 출발하더라도

B도시에는 T교차로를 이용하여 1 2 3 순으로 도착하게 할 수 있다.

그 작업 P, P, O, O, P, O순으로 작업을 하면 B도시에 1, 2, 3 순으로 도착한다.

1부터 N까지 번호를 가진 기차가 A도시에서 어떤 순으로 출발하든,

B도시에 번호순으로 도착 하도록 하는 교차로 작업을 출력한다.

모든 기차는 교차로에 들어가야만 B도시로 갈 수 있다.

번호순서대로 도착이 불가능하면 impossible 이라고 출력한다.

 

▣ 입력설명

첫 번째 줄에 자연수 N(3<=N<=30)가 주어진다.

두 번째 줄에 A도시에서 출발하는 기차번호순이 차례대로 입력된다.

 

▣ 출력설명

교차로 작업을 순서대로 P와 O로 출력한다.

 

▣ 입력예제

3

2 1 3

 

▣ 출력예제

PPOOPO

 

 

풀이 :

 

impossible이 아닐 경우

 

impossible일 경우

소스코드 :

#include<iostream>
#include<vector>
#include<stack>
#include<algorithm>

using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    int n, a, t;
    int j = 1;
    stack<int> s;
    vector<int> v(n+1);
    cin >> n;
    for(int i=1; i<=n; i++) {
        v[i] = i;
    }
    vector<char> result;
    for(int i=1; i<=n; i++) {
        cin >> t;
        s.push(t);
        result.push_back('P');
        while(true) {
            if(s.empty()) break;
            if(v[j] == s.top()) {
                s.pop();
                j++;
                result.push_back('O');
            }
            else break;
        }
    }
    
    if(!s.empty()) {
        cout << "impossible" << "\n";
    }
    else {
        for(int i=0; i<result.size(); i++) {
            cout << result[i];
        }
    }
    

    return 0;
}

 

 

 

출처 (학습 자료) :

 

it 취업을 위한 알고리즘 문제풀이

https://www.inflearn.com/course/%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98#curriculum