본문 바로가기
Problem Solving/Baekjoon

[백준] 1062 가르침 - Backtracking / Java

by graycode 2022. 9. 25.

 문제 링크

 

1062번: 가르침

첫째 줄에 단어의 개수 N과 K가 주어진다. N은 50보다 작거나 같은 자연수이고, K는 26보다 작거나 같은 자연수 또는 0이다. 둘째 줄부터 N개의 줄에 남극 언어의 단어가 주어진다. 단어는 영어 소문

www.acmicpc.net

 

 풀이 과정

 

 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class Main {

    static int n, k, max;
    static boolean[][] word;
    static boolean[] visit;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st = new StringTokenizer(br.readLine());

        n = Integer.parseInt(st.nextToken());
        k = Integer.parseInt(st.nextToken());

        if (k >= 5) {
            word = new boolean[n][26];
            visit = new boolean[26];

            for (int i = 0; i < n; i++) {
                for (char c : br.readLine().toCharArray())
                    word[i][c - 'a'] = true;
            }

            recur(0, 0);
        }

        bw.write(String.valueOf(max));
        bw.flush();
    }

    private static void recur(int idx, int depth) {
        if (depth == k) {
            max = Math.max(max, getCount());
            return;
        }

        for (int i = idx; i < 26; i++) {
            if (!visit[i]) {
                visit[i] = true;
                recur(i, depth + 1);
                visit[i] = false;
            }
        }
    }

    private static int getCount() {
        int cnt = 0;

        boolean flag;
        for (int i = 0; i < n; i++) {
            flag = true;
            for (int j = 0; j < 26; j++) {
                if (word[i][j] && !visit[j]) {
                    flag = false;
                    break;
                }
            }
            if (flag)
                cnt++;
        }

        return cnt;
    }

}

댓글