백준 10828: 스택 - javascript (구현)

문제 출처 

https://www.acmicpc.net/problem/10828

10828번: 스택

문제

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

명령은 총 다섯 가지이다.

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

입력

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

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력 1 복사

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

예제 출력 1 복사

2
2
0
2
1
-1
0
1
-1
0
3

예제 입력 2 복사

7
pop
top
push 123
top
pop
top
pop

예제 출력 2 복사

-1
-1
123
123
-1
-1

문제 풀이 힌트 

문제 풀이 코드 1) 

const input = require('fs').readFileSync('/dev/stdin').toString().trim().split('\n'); 
const N = Number(input.shift()); 
let stack = [];

for(let i=0; i<N; i++){
   let orders = input[i].split(' ')
    if(orders.length === 2){
        let order = orders[0]; 
        let number = Number(orders[1]); 
        stackOrder(order, number); 
    }else{
        let order = orders[0];
        stackOrder(order, 0)
     }
   }
let answer = [];
function stackOrder(order, number){
    if(order === 'push'){
        stack.push(number); 
    }else if(order === 'pop'){
        if(stack.length === 0){
            answer.push(-1)
        }else{
            answer.push(stack.pop())
        }
    }else if(order === 'size'){
           answer.push(stack.length);
    }else if(order === 'empty'){
        if(stack.length === 0){
            answer.push(1);
        }else{
            answer.push(0);
        }
    }else{
        if(stack.length === 0){
           answer.push(-1)
        }else{
           answer.push(stack[stack.length-1])
        }
    }
}
console.log(answer.join('\n'));

 

 

문제 풀이 코드 2)

const array = require('fs').readFileSync('/dev/stdin').toString().split('\n');

const stack = [];
const result = [];

const len = array.shift();

for (let i = 0; i < len; i++) {        
    switch(array[i]) {
        case 'pop': 
          result.push(stack.pop() || -1);
          break;

        case 'size': 
          result.push(stack.length);
          break;

        case 'empty': 
          result.push(stack[0] ? 0 : 1);
          break;

        case 'top': 
          result.push(stack[stack.length - 1] || -1);
          break;

        default: 
          stack.push(array[i].split(" ")[1]);
          break;
    }
}

console.log(result.join('\n'));

문제 풀이 코드 3)

const array = require('fs').readFileSync('/dev/stdin').toString().split('\n');
array.shift();

const stack = [];

const fun = {
  pop: () => stack.pop() || -1,
  size: () => stack.length,
  empty: () => stack[0] ? 0 : 1,
  top: () => stack[stack.length - 1] || -1,
  push: (item) => {
    stack.push(item.split(" ")[1]); 
    return '';
  }
}

const result = array.reduce((acc, v) => 
	acc + (fun[v] ? `${fun[v]()}\n` : fun.push(v)), '');

console.log(result);

 

기억하고갈 Point

- for문을 하나만 작성할 땐 시간복잡도가 BigO(n)

- let answer = [];을 중간에 위치했을 때는 런타임 에러가 떴음.

- switch 문 참고 주소 https://bigtop.tistory.com/29

[JavaScript] 자바스크립트 switch문 조건문 이해하기