PS/BOJ

BOJ 10825 국영수 [Java]

모달조아 2021. 7. 15. 06:27

BOJ 10825 국영수

- 문제 링크

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

- 문제 해설

문제에서 주어진 조건대로 정렬해주는 compare 메소드를 재정의해주는 방식으로 풀었다. return 값에 따라 어떻게 정렬되는지를 신경써주면 쉽게 해결할 수 있는 문제이다.

- String 클래스의 compareTo와 Integer.compare()는 왼쪽에 오는 것이 작으면 -1을, 같으면 0을, 왼쪽의 것이 더 크면 1을 리턴한다.
- Comparator의 compare 메소드의 경우에는 만약 인자로 a, b를 받았다고 하자. 음수를 리턴할 경우 a가 b보다 왼쪽에, 양수를 리턴 받을 경우 b가 a 왼쪽에 위치하게 된다. 0인 경우는 a와 b가 같은 경우이다.

위 내용을 바탕으로 코드를 구현했다.

- 코드 보기

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

public class Main
{
    static int n;

    public static void main(String[] args) throws IOException
    {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(br.readLine());
        String arr[][] = new String[n][4];

        for (int i = 0; i < n; i++)
        {
            StringTokenizer st = new StringTokenizer(br.readLine());
            arr[i][0] = st.nextToken();
            arr[i][1] = st.nextToken();
            arr[i][2] = st.nextToken();
            arr[i][3] = st.nextToken();
        }

        Arrays.sort(arr, new Comparator<String[]>()
        {
            public int compare(String[] s1, String[] s2)
            {
                if (Integer.parseInt(s1[1]) == Integer.parseInt(s2[1]))
                {
                    if (Integer.parseInt(s1[2]) == Integer.parseInt(s2[2]))
                    {
                        if (Integer.parseInt(s1[3]) == Integer.parseInt(s2[3]))
                        {
                            return s1[0].compareTo(s2[0]);
                        }

                        return Integer.compare(Integer.parseInt(s2[3]), Integer.parseInt(s1[3]));
                    }

                    return Integer.compare(Integer.parseInt(s1[2]), Integer.parseInt(s2[2]));
                }

                return Integer.compare(Integer.parseInt(s2[1]), Integer.parseInt(s1[1]));
            }
        });

        for (int i = 0; i < n; i++)
        {
            System.out.println(arr[i][0]);
        }
    }
}