[Baekjoon] 4836_춤

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

---

< 춤 >

 

문제 풀이 

 

문제에서 제시한 조건에 일치하는지 확인한다.

1) dip : jiggle 다음 or 다다음 / twirl 전
2) 끝 : clap stomp clap
3) twirl이 있다면 hop도 있어야 함
4) jiggle로 시작 x
5) dip은 무조건 포함 

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;
import java.io.OutputStreamWriter;
import java.io.InputStreamReader;

public class _4836_ { // 춤

	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;

		String str = "";

		while ((str = bf.readLine()) != null) {
			st = new StringTokenizer(str);

			ArrayList<String> list = new ArrayList<>();
			while (st.hasMoreTokens()) {
				list.add(st.nextToken());
			}

			Set<Integer> set = new HashSet<>();
			ArrayList<Integer> num = new ArrayList<>();
			boolean twirl = false, hop = false, dip = false;
			for (int i = 0; i < list.size(); i++) {
				if (list.get(i).equals("dip")) {
					dip = true;
					if ((i - 1 >= 0 && list.get(i - 1).equals("jiggle"))
							|| (i - 2 >= 0 && list.get(i - 2).equals("jiggle"))
							|| (i + 1 < list.size() && list.get(i + 1).equals("twirl"))) {
						continue;
					} else {
						set.add(1);
						num.add(i);
					}
				}
				if (list.get(i).equals("twirl")) {
					twirl = true;
				}
				if (list.get(i).equals("hop")) {
					hop = true;
				}
			}
			if (list.size() < 3 || !(list.get(list.size() - 3).equals("clap")
					&& list.get(list.size() - 2).equals("stomp") && list.get(list.size() - 1).equals("clap"))) {
				set.add(2);
			}
			if (twirl) {
				if (!hop) {
					set.add(3);
				}
			}
			if (list.get(0).equals("jiggle")) {
				set.add(4);
			}
			if (!dip) {
				set.add(5);
			}

			if (set.size() == 0) {
				bw.write("form ok: ");
				for (int i = 0; i < list.size(); i++) {
					bw.write(list.get(i) + " ");
				}
			} else {
				ArrayList<Integer> result = new ArrayList<>(set);
				Collections.sort(result);

				if (result.size() == 1) {
					bw.write("form error " + result.get(0) + ": ");
				} else {
					bw.write("form errors ");

					Collections.sort(result);
					for (int i = 0; i < result.size(); i++) {
						bw.write(result.get(i) + "");
						if (i < result.size() - 2) {
							bw.write(", ");
						} else if (i == result.size() - 2) {
							bw.write(" and ");
						}
					}
					bw.write(": ");
				}
				for (int i = 0; i < list.size(); i++) {
					if (num.contains(i)) {
						bw.write("DIP ");
					} else {
						bw.write(list.get(i) + " ");
					}
				}
			}
			bw.write("\n");
		}
		bw.flush();
	}
}

변수)
str : 입력값
list : ArrayList <String> 입력값을 단어로 저장 
set : HashSet <Integer> 어긴 규칙 번호
num : 1번 규칙을 어겼을 때 dip 위치
twirl, hop, dip : 각 춤 여부
result : ArrayList <Integer>

 

입력값이 없을 때까지 다음 과정을 반복한다.

 

1) 입력값을 단어로 ArrayList에 저장

2) ArrayList를 전체 탐색하며 1번 규칙을 지켰는지 확인. 어겼다면 set에 1을, num에 dip 위치를 저장 / twirl과 hop, dip이 있는지 여부 확인

3) ArrayList의 끝에 3개가 clap stomp clap이 아니라면 2번을 어긴 것이므로 set에 2 저장 

4) twirl가 있지만 hop이 없다면 set에 3 저장

5) ArrayList의 0번째 값이 jiggle이라면 규칙을 어긴 것이므로 set에 4 저장

6) dip이 없다면 set에 5 저장

7) set 크기가 0이라면 규칙을 모두 지킨 것이므로 "form ok: "와 입력값 출력

8) set 크기가 0이 아니라면 어긴 규칙을 오름차순으로 출력하기 위해 ArrayList로 변환 후 정렬 

8-1) 규칙을 1개 어겼다면 "form error "와 어긴 규칙 번호 출력

8-2) 규칙을 여러 개 어겼다면 "form errors "와 어긴 규칙 번호 출력

8-3) 입력값을 출력하며 1번 규칙을 어겼다면 규칙에 어긋난 dip을 대문자로 출력 

---

---