본문 바로가기

Java/Java 알고리즘 SWEA

[SW 아카데미] 13229. 일요일

반응형

오늘의 요일을 나타내는 문자열 S가 주어진다. S는 “MON”(월), “TUE”(화), “WED”(수), “THU”(목), “FRI”(금), “SAT”(토), “SUN”(일) 중 하나이다.

다음 (즉, 내일 이후의 가장 빠른) 일요일까지는 며칠 남았을까?
 

[입력]

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.

각 테스트 케이스는 하나의 줄로 이루어진다. 각 줄에는 문자열 S가 주어진다.
 

[출력]

각 테스트 케이스마다, 다음 일요일까지 며칠 남았는지를 한 줄에 하나씩 출력한다.

입력
3
SUN
SAT
MON
 
출력
#1 7
#2 1
#3 6

import java.util.HashMap;
import java.util.Scanner;

public class  Solution {
	public void solution(String [] arr) {
		String [] week = {"MON","TUE","WED","THU","FRI","SAT","SUN"};
		int [] count = {6, 5, 4, 3, 2,1,7};
		HashMap<String, Integer> map = new HashMap<>();
		for(int i=0;i<7;i++) {	
			map.put(week[i], count[i]);
		}
		int i=1;
		for(String x : arr) {
			System.out.print("#"+i+++" "+map.get(x));
			System.out.println();
		}
		
	}
	public static void main(String[] args) {
		Solution T= new Solution();
		Scanner kb = new Scanner(System.in);
		int n=kb.nextInt();
		String [] arr=new String[n];
		for(int i=0;i<n;i++) {
			arr[i]=kb.next();
		}
		T.solution(arr);
		
	}

}
반응형