[Java] 2456 나는 학급회장이다 - 정렬

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

2456번: 나는 학급회장이다

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;
 
public class bj2456 {
    static int[][] candidate;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
        Integer N = Integer.parseInt(br.readLine());
        candidate = new int[4][N];
        int[][] sum = new int[4][5];
        for(int i=0;i<N;i++){
            String tmp = br.readLine();
            StringTokenizer st = new StringTokenizer(tmp);
            candidate[1][i] = Integer.parseInt(st.nextToken()); //1번점수
            candidate[2][i] = Integer.parseInt(st.nextToken()); //2번점수
            candidate[3][i] = Integer.parseInt(st.nextToken()); //3번점수
 
            for(int j=0;j<3;j++){
                sum[j+1][4] =j+1; // 후보 번호가 몇번인지
                sum[j+1][0] += candidate[j+1][i]; //해당 후보가 몇점인지
                sum[j+1][((candidate[j+1][i])%3)+1]++; //후보별 점수 개수 세기
            }
        }
 
        Arrays.sort(sum, new Comparator<int[]>(){
            @Override
            public int compare(int[] o1, int[] o2){
                if(o1[0] ==o2[0]){
                    return o2[1]-o1[1];
                }else{
                    return o2[0]-o1[0];
                }
            }
        });
 
        int[] arr1 = Arrays.copyOfRange(sum[0], 1,4);
        int[] arr2 = Arrays.copyOfRange(sum[1], 1,4);
 
        for(int[] i : sum){
            for(int in : i){
                System.out.print(in + " ");
            }
            System.out.println();
        }
 
        if(sum[0][0] == sum[1][0]){
            if (is(arr1, arr2)) {
                System.out.println(0 + " " + sum[0][0]);
            }else{
                System.out.println(sum[0][4] + " " + sum[0][0]);
            }
        }else{
            System.out.println(sum[0][4] + " " + sum[0][0]);
        }
    }
    private static boolean is(int[] a, int[] b){
        int cnt=0;
        for(int s=0;s<3;s++){
            if(a[s] ==b[s]) {
                cnt++;
            }
        }
        if(cnt==3) return true;
        else return false;
    }
}

sum 이라는 배열에 각 값을 저장했다.

[후보n이 받은 점수의 총 합] [후보 n이 받은 3점의 개수] [후보 n이 받은 2점의 개수] [후보 n이 받은 1점의 개수] [후보 n의 후보 번호 n]

 

이렇게 저장한 후, sum배열을 내림차순으로 정렬하여

만약 sum[0]의 총합과 sum[1]의 총합이 같으면,

sum[0]의 [1:3] 즉 3,2,1점의 개수를 비교해 모두 같으면 0을 출력, 하나라도 다르면 sum[0]을 출력했다. 끄읕~