[Baekjoon] 11254_Hydraulic Arm

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

---

< Hydraulic Arm >

 

문제 풀이 

 

예제를 봤을 때 문제에서 말한 "left to right"가 아닌 "right to left"인 것 같아 "right to left"로 생각하고 문제를 해결했다.

Queue 1개와 Stack 1개를 사용하여 문제를 해결한다.

 

* Queue : 입력값

* Stack :  shelf

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Stack;
import java.util.StringTokenizer;

public class _11254_ { // Hydraulic Arm

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

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

		for (int i = 0; i < t; i++) {
			int n = Integer.parseInt(bf.readLine());

			int cnt = 1;
			Stack<Integer> shelf = new Stack<>();
			Queue<Integer> belt = new LinkedList<>();

			st = new StringTokenizer(bf.readLine());
			for (int j = 0; j < n; j++) {
				belt.add(Integer.parseInt(st.nextToken()));
			}

			boolean flag = false;
			while (cnt <= n) {
				if (!belt.isEmpty() && belt.peek() == cnt) {
					belt.poll();
					cnt += 1;
				} else if (!shelf.isEmpty() && shelf.peek() == cnt) {
					shelf.pop();
					cnt += 1;
				} else {
					if (!belt.isEmpty()) {
						shelf.add(belt.poll());
					} else {
						flag = true;
						break;
					}
				}
			}
			if (flag) {
				bw.write("no\n");
			} else {
				bw.write("yes\n");
			}
		}
		bw.flush();
	}
}

변수)
t : 테스트 케이스 수
n : egg packages 개수
cnt : 현재 번호
shelf : 선반에 있는 값
belt : 왼쪽 벨트에 있는 값
flag : 조건 충족 여부 

 

테스트 케이스 수를 입력받아 그 수만큼 다음 과정을 반복한다.

 

1) egg packages 개수를 입력받는다.

2) egg packeges 개수만큼 값을 입력받아 Queue에 저장한다.

3) cnt를 1로 초기화한 후 cnt가 n이하일 때 다음 과정을 반복한다.

3-1) belt의 peek 값이 cnt와 같다면 poll후 cnt+1

3-2) shelf의 peek 값이 cnt와 같다면 pop후 cnt+1

3-3) belt에 아직 값이 있다면 shelf에 belt 값 추가

3-4) 모든 조건에 맞지 않다면 조건에 충족할 수 없으므로 종료

4) 최종 flag 값에 따라 답을 출력한다.

---

---