BOJ 1339 단어 수학

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

문제

민식이는 수학학원에서 단어 수학 문제를 푸는 숙제를 받았다.

단어 수학 문제는 N개의 단어로 이루어져 있으며, 각 단어는 알파벳 대문자로만 이루어져 있다. 이때, 각 알파벳 대문자를 0부터 9까지의 숫자 중 하나로 바꿔서 N개의 수를 합하는 문제이다. 같은 알파벳은 같은 숫자로 바꿔야 하며, 두 개 이상의 알파벳이 같은 숫자로 바뀌어지면 안 된다.

예를 들어, GCF + ACDEB를 계산한다고 할 때, A = 9, B = 4, C = 8, D = 6, E = 5, F = 3, G = 7로 결정한다면, 두 수의 합은 99437이 되어서 최대가 될 것이다.

N개의 단어가 주어졌을 때, 그 수의 합을 최대로 만드는 프로그램을 작성하시오.

---

입력

첫째 줄에 단어의 개수 N(1 ≤ N ≤ 10)이 주어진다. 둘째 줄부터 N개의 줄에 단어가 한 줄에 하나씩 주어진다. 단어는 알파벳 대문자로만 이루어져있다. 모든 단어에 포함되어 있는 알파벳은 최대 10개이고, 수의 최대 길이는 8이다. 서로 다른 문자는 서로 다른 숫자를 나타낸다.

---

출력

첫째 줄에 주어진 단어의 합의 최댓값을 출력한다.

---

예제 입력 1

2
AAA
AAA

예제 출력 1

1998

예제 입력 2

2
GCF
ACDEB

예제 출력 2

99437

… 이하 예제 생략

---

코드

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

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        int[] alphabets = new int[26];
        for (int i = 0; i < N; i++) {
            // 단어에 있는 알파벳들을 모두 같은 가중치인 1로 생각했을 때, 위치한 자릿수에 해당하는 값을
            // 배열에서 각 알파벳을 의미하는 인덱스의 위치에 더함
            String input = br.readLine();
            for (int j = 0; j < input.length(); j++)
                alphabets[input.charAt(j) - 'A'] += Math.pow(10, input.length() - j - 1);
        }
        // 알파벳들을 모두 같은 가중치로 계산했을 때 값이 가장 큰 알파벳 순으로 꺼내며
        // 9부터 1까지 곱한 수를 answer에 더함 (0을 곱한 수는 0이므로 더해도 의미 없기 때문에 생략)
        PriorityQueue<Integer> queue = new PriorityQueue<>(Collections.reverseOrder());
        for (int alphabet: alphabets) {
            if(alphabet != 0) queue.offer(alphabet);
        }

        int answer = 0;
        for (int i = 9; !queue.isEmpty() && i > 0; i--) {
            answer += queue.poll() * i;
        }

        System.out.println(answer);
    }
}