[ C ] 백준 10828번 스택

 

 

문제

 

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

 

입력

 

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

 

출력

 

 

 

 

 

 

 

 소스코드

 

#include <stdio.h>
#include <string.h>

int stack[10000];
int stack_size=0;

void push(int push_data){
    stack[stack_size] = push_data;
    stack_size += 1;
}

int empty(){
    if(stack_size == 0){
        return 1;
    }
    return 0;
}

int pop(){
    if(empty()){
        return -1;
    }
    stack_size -= 1;
    return stack[stack_size];
}

int top(){
    if(empty()){
        return -1;
    }
    return stack[stack_size-1];
}

int main(){

    int N = 0, push_data = 0;
    char command[5] = {0,};

    scanf("%d",&N);

    for(int i=0;i<N;i++){

        scanf("%s",command);

        if(!strcmp(command,"push")){
            scanf("%d",&push_data);
            push(push_data);
        }
        else if(!strcmp(command,"pop")){
            printf("%d\n",pop());
        }
        else if(!strcmp(command,"empty")){
            printf("%d\n",empty());
        }
        else if(!strcmp(command,"size")){
            printf("%d\n",stack_size);
        }
        else if(!strcmp(command,"top")){
            printf("%d\n",top());
        }

    }

    return 0;
}

 

 

설명

 

#include <stdio.h>
#include <string.h>

int stack[10000];
int stack_size=0;

 

해더는 기본적인 stdio.h 와 string.h을 사용하였다.

string.h를 사용한 이유는 strcmp를 사용하여 입력받은 문자와 push, pop, empty, size, top을 비교하여 사용자 정의 함수를 불러서 사용하기 위해서 이다.

stack 배열과 stack_size 변수를 전역 변수로 선언하였다.

stack_size 변수는 처음 시작할 때 stack 배열에 아무 값이 없을 것이기 때문에 0으로 초기화를 했다.

 

int main(){

    int N = 0, push_data = 0;
    char command[5] = {0,};

    scanf("%d",&N);

    for(int i=0;i<N;i++){

        scanf("%s",command);

        if(!strcmp(command,"push")){
            scanf("%d",&push_data);
            push(push_data);
        }
        else if(!strcmp(command,"pop")){
            printf("%d\n",pop());
        }
        else if(!strcmp(command,"empty")){
            printf("%d\n",empty());
        }
        else if(!strcmp(command,"size")){
            printf("%d\n",stack_size);
        }
        else if(!strcmp(command,"top")){
            printf("%d\n",top());
        }

    }

    return 0;
}

 

main에서는 반복의 관한 N 변수와 push로 입력받는 데이터의 값을 받아주는 push_data 변수

그리고 push, pop, empty, size, top 중 어떤 함수를 불러올 것인지 입력을 받아오는 command 배열이 있다.

 

void push(int push_data){
    stack[stack_size] = push_data;
    stack_size += 1;
}

int empty(){
    if(stack_size == 0){
        return 1;
    }
    return 0;
}

int pop(){
    if(empty()){
        return -1;
    }
    stack_size -= 1;
    return stack[stack_size];
}

int top(){
    if(empty()){
        return -1;
    }
    return stack[stack_size-1];
}

 

사용자 정의 함수로는 push, empty, pop, top 이 있다. size 함수를 만들지 않은 이유는 stack_size를 출력만 하면 끝이기

때문에 딱히 만들 필요가 없다고 생각하여 만들지 않았다.

 

void push(int push_data){
    stack[stack_size] = push_data;
    stack_size += 1;
}

 

push 함수에서는 push_data변수의 값을 받아와서 stack배열 stack_size(카운터의 역할도 함께 함) 번째 요소에 push_data 값을 넣어준다. 그 후 배열에 값이 추가되었기 때문에 stack_size에 값을 1 더한다.

 

int empty(){
    if(stack_size == 0){
        return 1;
    }
    return 0;
}

 

empty 함수에서는 stack 배열에 값이 있는지 없는지만 판단해 주면 되는 함수이기 때문에 카운터의 역할을 하고 있는

stack_size 함수값이 0인지 아닌지만 확인해서 값을 반환해 주도록 했다.

 

int pop(){
    if(empty()){
        return -1;
    }
    stack_size -= 1;
    return stack[stack_size];
}

 

pop 함수는 stack 배열에서 가장 위에 (마지막에 입력한) 값을 출력하는 역할의 함수이다.

문제에서 stack 배열에 값이 없을 경우 -1을 출력하라는 부분이 있는데 이것은 empty 함수에서 stack_size 값이 0인지

확인하는 것으로 대체 가능하기 때문에 여기서도 사용했다. 만약 stack 배열에 값이 없다면 (만약 stack_size가 0이라면) -1 값을

반환하고 함수가 끝난다. 아닐 경우 가장 마지막에 입력한 값을 출력하고 stack_size에 값을 1 마이너스 한다.

 

int top(){
    if(empty()){
        return -1;
    }
    return stack[stack_size-1];
}

 

top 함수에서도 pop 함수와 마찬가지로 stack 배열에 값이 없을 경우 -1을 출력하라는 문제가 있기 때문에 pop 함수와

같은 방식으로 처리하였다.

아닐 경우 stack 배열에 stack_size 에서 1을 뺀 값의 요소값을 반환한다.

값을 입력하면 stack 배열 stack_size 번째 요소에 값을 넣은 후 stack_size에 1을 더하기 때문에 마지막에 입력한 값을

확인하려면 stack_size 에서 1을 뺀 요소값을 반환해야 한다.

 

스택 구조는 중요하다. 또 정리하여 공부하도록 하자!!!

 

 

www.acmicpc.net/problem/10828

10828번: 스택