[Baekjoon] 20680_Birds Rituals

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

---

< Birds Rituals >

 

문제 풀이 

 

다음 조건에 따라 움직인다.

insert : 주어진 인덱스에 추가
depart : 삭제
relocate : 현재 있던 새 이동

 

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 _20680_ { // Birds Rituals

	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 = new StringTokenizer(bf.readLine());

		int n = Integer.parseInt(st.nextToken());
		int s = Integer.parseInt(st.nextToken());

		ArrayList<String> list = new ArrayList<>();
		st = new StringTokenizer(bf.readLine());

		for (int i = 0; i < n; i++) {
			list.add(st.nextToken());
		}

		for (int i = 0; i < s; i++) {
			st = new StringTokenizer(bf.readLine());

			String cmd = st.nextToken();
			String name = st.nextToken();

			if (cmd.equals("insert")) {
				int idx = Integer.parseInt(st.nextToken());
				list.add(idx, name);

			} else if (cmd.equals("depart")) {
				list.remove(name);

			} else {
				int idx = list.indexOf(name);
				int move = Integer.parseInt(st.nextToken());

				if (move == 0) {
					continue;
				}
				
				if(move < 0) {
					list.add(idx + move, name);
					list.remove(idx+1);
				}else {
					list.add(idx + move+1, name);
					list.remove(idx);
				}
				
			}
		}

		for (String name : list) {
			bw.write(name + " ");
		}
		bw.flush();
	}
}

변수)
n, s : 처음 줄에 앉아 있는 새들의 수, 행동의 수
list : 앉아있는 새 순서 
cmd, name : 행동 유형, 새 이름
idx : 새 위치
move : 이동 범위

 

처음 줄에 앉아 있는 새들의 수와 행동의 수를 입력받는다. 새들의 수만큼 새 이름을 입력받아 ArrayList에 저장한다. 행동의 수만큼 다음 과정을 반복한다.

 

1) 행동 유형, 새 이름 입력

2) insert라면 idx 위치에 새 추가

3) depart라면 새 제거

4) relocate라면 먼저 현재 새의 위치를 찾음. 이동 범위가 0이라면 움직이지 않는 것이므로 다음 행동으로 넘어가고, 0이 아니라면 새 이동 후 원래 위치에 있던 새를 제거

 

최종 ArrayList의 값을 출력한다.  

---

---