[ABC301] C - AtCoder Cards (AtCoder Beginner Contest 301 in Java)

목차

문제 : ABC301 - C

 

필요 알고리즘

• 그리디
  
  • 그리디라고 하기 좀 애매하긴한데, 아무튼 바꿀 수 있으면 대충 바꾸는 로직으로 풀 수 있으므로 일단 그리디라 적었음.

※ 제 코드에서 왜 main 함수에 로직을 직접 작성하지 않았는지, 왜 Scanner를 쓰지 않고 BufferedReader를 사용했는지 등에 대해서는 '자바로 백준 풀 때의 팁 및 주의점' 글을 참고해주세요. 백준을 자바로 풀어보려고 시작하시는 분이나, 백준에서 자바로 풀 때의 팁을 원하시는 분들도 보시는걸 추천드립니다.

 

 

풀이

  cnt 배열에 'a'부터 'z' 까지 S에 나온 문자만큼 +1, T에 나온 문자만큼 -1을 해주며 세줬다. 예를들어 S가 "ABC", T가 "BCD" 라면 cnt[0] = 1, cnt[1] = 0, cnt[2] = 0, cnt[3] = -1 이 될 것이다. 그리고 '@'의 갯수는 별도로 cntS와 cntT에 세줬다.

 

  우선 처음으로 생각해볼만한 점은 a,t,c,o,d,e,r 에 포함되지 않는 문자의 갯수가 다를 경우 어떤식으로든 만드는게 불가능하다는 점이다.

for (int i = 0; i < cnt.length; i++) {
    if (i == atoi('a') || i == atoi('t') || i == atoi('c')
            || i == atoi('o') || i == atoi('d') || i == atoi('e') || i == atoi('r'))
        continue;

    if (cnt[i] != 0) {
        System.out.println("No");
        return;
    }
}

 

  다음으로는 a,t,c,o,d,e,r에 포함된 글자가 서로 다른 갯수가 각 문자의 남는 '@'을 초과해서 필요하면 이 경우도 불가하다.

String tmp = "atcoder";
for (int i = 0; i < tmp.length(); i++) {
    int idx = atoi(tmp.charAt(i));

    if (cnt[idx] == 0) continue;

    if (cnt[idx] > 0) {
        cntT -= cnt[idx];
    } else {
        cntS += cnt[idx];
    }
}

System.out.println(cntT >= 0 && cntS >= 0 ? "Yes" : "No");

 

 

코드 : github

import java.io.BufferedReader;
import java.io.InputStreamReader;
 
public class Main {
 
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
    public static void main(String[] args) throws Exception {
        new Main().solution();
    }
 
    private void solution() throws Exception {
        String s = br.readLine();
        String t = br.readLine();
 
        int[] cnt = new int['z'-'a'+1];
        int cntS = 0;
        int cntT = 0;
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == '@') {
                cntS++;
                continue;
            }
            cnt[atoi(s.charAt(i))]++;
        }
        for (int i = 0; i < t.length(); i++) {
            if (t.charAt(i) == '@') {
                cntT++;
                continue;
            }
            cnt[atoi(t.charAt(i))]--;
        }
 
        for (int i = 0; i < cnt.length; i++) {
            if (i == atoi('a') || i == atoi('t') || i == atoi('c')
                    || i == atoi('o') || i == atoi('d') || i == atoi('e') || i == atoi('r'))
                continue;
 
            if (cnt[i] != 0) {
                System.out.println("No");
                return;
            }
        }
 
        String tmp = "atcoder";
        for (int i = 0; i < tmp.length(); i++) {
            int idx = atoi(tmp.charAt(i));
 
            if (cnt[idx] == 0) continue;
 
            if (cnt[idx] > 0) {
                cntT -= cnt[idx];
            } else {
                cntS += cnt[idx];
            }
        }
 
        System.out.println(cntT >= 0 && cntS >= 0 ? "Yes" : "No");
    }
 
    private int atoi(char c) {
        return c-'a';
    }
}