[Baekjoon] 9843_LVM

문제(출처: https://www.acmicpc.net/problem/9843)

---

< LVM >

 

문제 풀이 

PUSH x : stack에 x 저장 
STORE : register에 stack의 top 저장
LOAD : stack에 register 값 저장
PLUS : stack의 top 2개의 값을 가져와 합을 구한 후 저장
TIMES : stack의 top 2개의 값을 가져와 곱을 구한 후 저장
IFZERO n : stack의 top 값이 0이라면 n번째 명령어로 이동
DONE : stack의 top 값 출력

 

 

my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;

public class _9843_ { // LVM

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;

		int n = Integer.parseInt(bf.readLine());

		String arr[] = new String[n];

		for (int i = 0; i < n; i++) {
			arr[i] = bf.readLine();
		}

		Stack<Integer> stack = new Stack<>();
		int idx = 0, register = 0;
		while (true) {
			st = new StringTokenizer(arr[idx]);

			String cmd = st.nextToken();
			if (cmd.equals("PUSH")) {
				int x = Integer.parseInt(st.nextToken());
				stack.add(x);
				idx += 1;
			} else if (cmd.equals("STORE")) {
				register = stack.pop();
				idx += 1;
			} else if (cmd.equals("LOAD")) {
				stack.add(register);
				idx += 1;
			} else if (cmd.equals("PLUS")) {
				stack.add(stack.pop() + stack.pop());
				idx += 1;
			} else if (cmd.equals("TIMES")) {
				stack.add(stack.pop() * stack.pop());
				idx += 1;
			} else if (cmd.equals("IFZERO")) {
				int num = stack.pop();
				int move = Integer.parseInt(st.nextToken());
				if (num == 0) {
					idx = move;
				} else {
					idx += 1;
				}
			} else {
				System.out.println(stack.pop());
				break;
			}
		}
	}
}

변수)
n : 명령어 수
arr : 명령어 저장하는 배열
stack : Stack <Integer>
idx : 현재 실행하는 명령어 위치
cmd : 명령어

 

명령어 수를 입력받는다. 명령어 수만큼 명령어를 입력받아 arr에 저장한다. 종료 조건인 "DONE"이 나올 때까지 다음 과정을 반복한다.

 

1) PUSH라면 x를 stack에 저장

2) STORE라면 stack에서 pop 한 값을 register에 저장

3) LOAD라면 stack에 register 값을 저장

4) PLUS라면 stack에서 2개를 pop해와 더한 값을 stack에 저장

5) TIMES라면 stack에서 2개를 pop해와 곱한 값을 stack에 저장

6) IFZERO라면 stack에서 pop 한 값이 0인지 확인. 0이라면 move번째 명령어를 실행하기 위해 idx를 move로 저장

7) DONE이라면 stack의 pop 한 값을 출력하고 종료 

---

---