인프런 커뮤니티 질문&답변

한번에 정렬을 하면 될걸 시작부터 잘못 접근해서 개판이 난 코드입니다.

import java.util.*;

public class Prob2910 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        sc.nextInt();
        Map<Integer,Integer> freqs = new HashMap<>(); // 숫자 - 빈도
        Map<Integer,Integer> index = new HashMap<>(); // 숫자 - 첫 등장 위치
        for (int i = 0; i < n; i++) {
            int num = sc.nextInt();
            if(freqs.containsKey(num)){
                freqs.put(num, freqs.get(num)+1);
            }else {
                index.put(num,i);
                freqs.put(num,1);
            }
        }
        Map<Integer,List<Integer>> freqNums = new HashMap<>(); // 빈도 - 숫자들
        for (int num: freqs.keySet()) {
            int freq = freqs.get(num);
            if(freqNums.containsKey(freq)){
                freqNums.get(freq).add(num);
            }else {
                List<Integer> nums = new ArrayList<>();
                nums.add(num);
                freqNums.put(freq,nums);
            }
        }
        for (List<Integer> nums: freqNums.values()) {
            Collections.sort(nums, Comparator.comparingInt(index::get));
        }
        freqNums.keySet().stream().sorted(Comparator.reverseOrder()).forEach(key->{
            List<Integer> list = freqNums.get(key);
            list.forEach(e->{
                int count = freqs.get(e);
                for (int i = 0; i < count; i++) {
                    System.out.print(e+" ");
                }
            });
        });
    }
}

해설 강의처럼 한번에 정렬을 하는 코드입니다.

import java.util.*;

public class Prob2910_sortOnce {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        sc.nextInt();
        Map<Integer,Integer> numFreq = new HashMap<>();
        Map<Integer,Integer> numFirstIndex = new HashMap<>();
        List<Integer> nums = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            int num = sc.nextInt();
            nums.add(num);
            if(numFreq.containsKey(num)){
                numFreq.put(num,numFreq.get(num)+1);
            }else {
                numFreq.put(num,1);
                numFirstIndex.put(num,i);
            }
        }
        nums.sort((o1, o2) -> {
            if (numFreq.get(o1).equals(numFreq.get(o2))) {
                return numFirstIndex.get(o1) - numFirstIndex.get(o2);
            } else {
                return numFreq.get(o2) - numFreq.get(o1);
            }
        });
        nums.forEach(e->System.out.print(e+" "));
    }
}