[백준] 1339번 단어 수학 Java 풀이

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
import java.io.*;
import java.util.*;

public class Main {

  static int N;
  static int[] alphabetCount;
  static String[] words;
  static long max;

  public static void main(String args[]) throws Exception {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

    N = Integer.parseInt(br.readLine());    // 단어의 개수

    int count = 0;                  // 순열의 크기를 결정하기 위해 등장한 알바펫의 종류를 센다.
    alphabetCount = new int[26];    // 등장했던 알파벳들에 대해서 번호를 부여하기 위해 개수를 센다.
    words = new String[N];          // 주어지는 단어들

    for (int i=0; i<N; i++) {
      words[i] = br.readLine();

      for (int j=0; j<words[i].length(); j++) {
        int alpha = words[i].charAt(j) - 'A';
        if (alphabetCount[alpha] == 0)    count++;      // 0이면 첫 등장 이므로, 알파벳 종류 개수 + 1
        alphabetCount[alpha]++;
      }
    }

    perm(0, new boolean[26], new int[count]);

    System.out.println(max);
    br.close();
  }

  static void perm(int cnt, boolean[] visited, int[] res) {
    if (cnt == res.length) {            // 순열 완성
      int[] charToNum = new int[26];    // 알파벳에 숫자를 부여한 것을 저장할 배열

      int idx = 0;
      for (int i=0; i<26; i++)
        if (alphabetCount[i] != 0)     // 등장했던 알파벳에 대해서
          charToNum[i] = res[idx++];   // 순열 순서대로 숫자를 부여

      // 등장한 알파벳에 숫자를 부여한 후에
      // 주어진 문자열 words를 정수로 변환하여 result에 누적한다.
      long result = 0L;
      for (String word : words) {
        long wtl = wordToLong(word, charToNum);
        result += wtl;
      }

      if (result > max)   // max값 업데이트
        max = result;

      return;
    }

    // Permutation
    for (int i=0; i<10; i++) {
      if (visited[i])
        continue;
      visited[i] = true;
      res[cnt] = i;
      perm(cnt+1, visited, res);
      visited[i] = false;
    }
  }

  // 문자열 word를 charToNum 배열을 통해서 각 글자를 정수로 변환
  static long wordToLong(String word, int[] charToNum) {
    StringBuilder resultSb = new StringBuilder();

    for (char c : word.toCharArray())
      resultSb.append(charToNum[c-'A']);

    return Long.parseLong(resultSb.toString());
  }
}