복서 정렬하기

import java.util.*;
 
class Solution {
    public int[] solution(int[] weights, String[] head2head) {
        int n = weights.length;
        int[] answer = new int[n];
        ArrayList<Person> list = new ArrayList<Person>();
        
        // 선수 정보 계산
        for(int i=0;i<n;i++){
            int fight_count = 0;
            int heavy_count = 0;
            int winning_count = 0;
            float winning_rate = 0;
            for(int j=0;j<n;j++){
                if(head2head[i].charAt(j) == 'W'){
                    fight_count++;
                    winning_count++;
                    if(weights[j] > weights[i])
                        heavy_count++;
                } else if(head2head[i].charAt(j) == 'L')
                    fight_count++;
            }
            if(fight_count != 0)
                winning_rate = winning_count * 100 / (float) fight_count;
            list.add(new Person(i + 1, weights[i], fight_count, heavy_count, winning_count, winning_rate));
        }
 
        // 정렬
        Collections.sort(list, new Comparator<Person>(){
            @Override
            public int compare(Person o1, Person o2){
                float f_a = o2.winning_rate - o1.winning_rate;
                int a;
                if(f_a == 0){
                    a = o2.heavy_count - o1.heavy_count;
                    if(a == 0){
                        a = o2.weight - o1.weight;
                        if(a == 0)
                            a = o1.index - o2.index;
                    }
                } else 
                    return f_a > 0 ? 1 : -1;
                return a;
            }
        });
 
        for(int i=0;i<n;i++){
            answer[i] = list.get(i).index;
        }
        return answer;
    }
}
 
class Person {
    int index;
    int weight;
    int fight_count;
    int heavy_count;
    int winning_count;
    float winning_rate;
    public Person(int index, int weight, int fight_count, int heavy_count, int winning_count, float winning_rate){
        this.index = index;
        this.weight = weight;
        this.fight_count = fight_count;
        this.heavy_count = heavy_count;
        this.winning_count = winning_count;
        this.winning_rate = winning_rate;
    }
}

Comparator를 이용하여 4가지 조건을 검사하였다.