[백준] 30403번 : 무지개 만들기

https://www.acmicpc.net/problem/30403

30403번: 무지개 만들기

주어진 문자열을 활용해서 무지개(ROYGBIV)을 만들 수 있는지 체크하는 문제입니다.

단, 소문자인지 대문자인지 가능한지 체크한 후에 따라 결과를 다르게 하면 됩니다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		br.readLine();
		String s = br.readLine();
		
		boolean[] upperAlpha = new boolean[26], lowerAlpha = new boolean[26];
		
		for(char c : s.toCharArray()) {
			if(c >= 'A' && c <= 'Z') {
				upperAlpha[c - 'A'] = true;
			}
			else {
				lowerAlpha[c - 'a'] = true;
			}
		}
		
		int[] num = {1, 6, 8, 14, 17, 21, 24};
		
		int upper = 0, lower = 0;
		for(int i = 0; i < num.length; i++) {
			if(upperAlpha[num[i]]) {
				upper++;
			}
			if(lowerAlpha[num[i]]) {
				lower++;
			}
		}
		
		System.out.print(upper == 7 ? lower == 7 ? "YeS" : "YES" : lower == 7 ? "yes" : "NO!");
	}
}