[Baekjoon] 15287_Easy Quest

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

---

< Easy Quest >

 

문제 풀이 

 

0 이상일 때의 개수를 저장해 두고 음수일 때 |음수| 또는 0을 활용해서 적을 물리친다. 이때 0을 어떤 타입의 아이템을 요청했는지 저장해 둔다. 

 

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.ArrayList;
import java.util.StringTokenizer;

public class _15287_ { // Easy Quest

	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 n = Integer.parseInt(bf.readLine());

		int arr[] = new int[1001];
		ArrayList<Integer> result = new ArrayList<>();
		boolean flag = false;

		st = new StringTokenizer(bf.readLine());
		for (int i = 0; i < n; i++) {
			int num = Integer.parseInt(st.nextToken());

			if (num >= 0) {
				arr[num] += 1;
			} else {
				if (arr[-num] > 0) {
					arr[-num] -= 1;
				} else if (arr[0] > 0) {
					arr[0] -= 1;
					result.add(-num);
				} else {
					flag = true;
					break;
				}
			}
		}

		if (flag) {
			bw.write("No");
		} else {
			bw.write("Yes\n");

			for (int x : result) {
				bw.write(x + " ");
			}

			while (arr[0] > 0) {
				arr[0] -= 1;
				bw.write("1 ");
			}
		}
		bw.flush();
	}
}

변수)
n : 정수 개수
arr : 0 이상의 값 개수
result : 유니콘 값
flag : 적 물리침 여부
num : 정수 

 

정수 개수를 입력받은 뒤 정수 개수만큼 정수를 입력받는다. 0 이상이라면 배열에 +1 한다. 음수라면 먼저 배열에 (-정수) 값이 있는지 확인한다. 값이 있다면 -1 하고 값이 없다면 arr[0] 값이 있는지 확인한다. arr[0] 값이 있다면 -1 하고 result에 (-정수)를 저장한다. 둘 다 값이 없다면 모든 적을 물리치는 것이 불가능하므로 No를 출력한다. 모든 적을 물리치는 것이 가능하다면 Yes를 출력하고 result 값들도 출력한다. 이때, 아직 arr[0] 값이 0보다 크다면 그 값만큼 1을 출력한다. 

---

---