[SWEA] 색상환

 

 

 

[문제]

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AZROsPgqE88DFAWB

SW Expert Academy

 

 

 

[소스 코드]

import java.io.*;
import java.util.*;
public class Solution {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		String[] color = {"red", "orange", "yellow", "green", "blue", "purple"};
		HashMap<String,Integer> hm = new HashMap<>();
		for(int i=0;i<6;i++) {
			hm.put(color[i], i);
		}
		int T = Integer.parseInt(br.readLine());
		for(int tc=1;tc<=T;tc++) {
			st = new StringTokenizer(br.readLine());
			String a = st.nextToken();
			String b = st.nextToken();
			String result = "X";
			if(a.equals(b)) result = "E";
			else {
				int idxA = hm.get(a);
				int idxB = hm.get(b);
				if((idxA+1)%6==idxB || (idxA-1+6)%6==idxB) result = "A";
				else if((idxA+3)%6==idxB) result = "C";
			}
			System.out.println(result);
		}
	}
}