프로그래머스 - 복서 정렬 하기 (6주차)

https://programmers.co.kr/learn/courses/30/lessons/85002

코딩테스트 연습 - 6주차

 

• 주어진 조건에 따라 복서에게 우선 순위를 부여하여 정렬하면 된다.
• 우선 순위 정렬는 PriorityQueue를 사용하여 정렬하였다.

 

import java.util.*;

class Solution {
        public int[] solution(int[] weights, String[] head2head) {
            List<Boxer> boxers = new ArrayList<>();

            for (int i = 0; i < weights.length; i++) {
                boxers.add(new Boxer(weights[i], i + 1));
            }

            for (int i = 0; i < head2head.length; i++) {
                char[] histories = head2head[i].toCharArray();
                Boxer boxer = boxers.get(i);
                for (int j = 0; j < histories.length; j++) {
                    if (histories[j] == 'N') continue;
                    boxer.addHistory(histories[j], boxers.get(j));
                }
            }

            PriorityQueue<Boxer> queue = new PriorityQueue<>();
            for (int i = 0; i < boxers.size(); i++) {
                queue.offer(boxers.get(i));
            }


            int[] answer = new int[boxers.size()];
            int i = 0;
            while (!queue.isEmpty()) {
                answer[i] = queue.poll().id;
                i++;
            }

            return answer;
        }

        public class Boxer implements Comparable<Boxer> {
            public int winCount; // 이긴 경기 수 
            public int loseCount; // 진 경기 수
            public int playCount; // 경기 횟수
            public double winRate; // 승률
            public int winFromHigherWeight; // 몸무게가 많이 나간 사람에게 이긴 수
            public int weight; // 자기 몸무게 
            public int id; // 자기 번호

            public Boxer(int weight, int number) {
                this.weight = weight;
                this.id = number;
            }

            public void addHistory(char history, Boxer boxer) {
                playCount += 1;
                if (history == 'W') {
                    if (boxer.weight > this.weight) {
                        winFromHigherWeight += 1;
                    }
                    winCount += 1;

                } else if (history == 'L') {
                    loseCount += 1;
                }

                winRate = ((double) winCount / (winCount + loseCount)) * 100.0;
            }

            @Override
            public int compareTo(Boxer other) {
                if (this.winRate != other.winRate) return other.winRate - this.winRate > 0 ? 1 : -1;
                if (this.winFromHigherWeight != other.winFromHigherWeight) return other.winFromHigherWeight - this.winFromHigherWeight;
                if (this.weight != other.weight) return other.weight - this.weight;

                return this.id - other.id;
            }
        }
    }