[Java] 백준 2480. 주사위 세개

문제

1에서부터 6까지의 눈을 가진 3개의 주사위를 던져서 다음과 같은 규칙에 따라 상금을 받는 게임이 있다.

1. 같은 눈이 3개가 나오면 10,000원+(같은 눈)×1,000원의 상금을 받게 된다.
2. 같은 눈이 2개만 나오는 경우에는 1,000원+(같은 눈)×100원의 상금을 받게 된다.
3. 모두 다른 눈이 나오는 경우에는 (그 중 가장 큰 눈)×100원의 상금을 받게 된다.

예를 들어, 3개의 눈 3, 3, 6이 주어지면 상금은 1,000+3×100으로 계산되어 1,300원을 받게 된다. 또 3개의 눈이 2, 2, 2로 주어지면 10,000+2×1,000 으로 계산되어 12,000원을 받게 된다. 3개의 눈이 6, 2, 5로 주어지면 그중 가장 큰 값이 6이므로 6×100으로 계산되어 600원을 상금으로 받게 된다.

3개 주사위의 나온 눈이 주어질 때, 상금을 계산하는 프로그램을 작성 하시오.

입력

첫째 줄에 3개의 눈이 빈칸을 사이에 두고 각각 주어진다.

출력

첫째 줄에 게임의 상금을 출력 한다.

예제 입력 1

3 3 6

예제 출력 1

1300

풀이(TreeSet 사용)

120ms, 14268KB

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;
import java.util.TreeSet;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());
        ArrayList<Integer> arr = new ArrayList<>();

        int price = 0;

        for(int i=0; i<3; i++) {
            arr.add(Integer.parseInt(st.nextToken()));
        }
				
        TreeSet<Integer> set = new TreeSet<>(arr);

        if(set.size()==1)
            price = 10000 + (arr.get(0)*1000);
        else if(set.size()==2) {
            if(arr.get(0)==arr.get(1))
                price = 1000 + (arr.get(0)*100);
            else
                price = 1000 + (arr.get(2)*100);
        }
        else
            price = 100 * set.last();

        sb.append(price);
        System.out.println(sb.toString());
    }
}

Set 알기 전 풀이

128ms, 14292KB

import java.io.*;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        
        int A = Integer.parseInt(st.nextToken());
        int B = Integer.parseInt(st.nextToken());
        int C = Integer.parseInt(st.nextToken());
        int Max = 0;
        int price = 0;
        if(A==B) {
            if(A==C)
                price = 10000+(A*1000);
            else
                price = 1000+(A*100);
        }
        else if(B==C)
            price = 1000+(B*100);
        else if(A==C)
            price = 1000+(A*100);
        else {
            Max = Math.max(Math.max(A,B),C);
            price = 100*Max;
        }
        System.out.println(price);
    }
}