[자바-실버2]빈도 정렬 - Java

iamjinseo·2023년 2월 2일
0

문제풀이-Java

목록 보기
7/53
post-custom-banner


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


문제 설명

위대한 해커 창영이는 모든 암호를 깨는 방법을 발견했다. 그 방법은 빈도를 조사하는 것이다.

창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 N개로 이루어진 수열이고, 숫자는 모두 C보다 작거나 같다. 창영이는 이 숫자를 자주 등장하는 빈도순대로 정렬하려고 한다.

만약, 수열의 두 수 X와 Y가 있을 때, X가 Y보다 수열에서 많이 등장하는 경우에는 X가 Y보다 앞에 있어야 한다. 만약, 등장하는 횟수가 같다면, 먼저 나온 것이 앞에 있어야 한다.

이렇게 정렬하는 방법을 빈도 정렬이라고 한다.

수열이 주어졌을 때, 빈도 정렬을 하는 프로그램을 작성하시오.

입력
첫째 줄에 메시지의 길이 N과 C가 주어진다. (1 ≤ N ≤ 1,000, 1 ≤ C ≤ 1,000,000,000)

둘째 줄에 메시지 수열이 주어진다.

출력
첫째 줄에 입력으로 주어진 수열을 빈도 정렬한 다음 출력한다.

예제 입력 1

5 2
2 1 2 1 2

예제 출력 1

2 2 2 1 1

예제 입력 2

9 3
1 3 3 3 2 2 2 1 1

예제 출력 2

1 1 1 3 3 3 2 2 2

예제 입력 3

9 77
11 33 11 77 54 11 25 25 33

예제 출력 3

11 11 11 33 33 25 25 77 54

풀이

package silver2;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class B2910_빈도_정렬 {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		int N = Integer.parseInt(st.nextToken()); //N입력
		int C = Integer.parseInt(st.nextToken()); //C입력
		
		st = new StringTokenizer(br.readLine()); //배열 입력
		
		Map<Integer, int[]> map = new HashMap<>(); //숫자 key, 빈도와 등장순서value가 있는 map
		int order = 1;
		while(st.hasMoreTokens()) {
			int n = Integer.parseInt(st.nextToken());
			
			if (!map.containsKey(n)) { //n이 나온적 없는 숫자면
				int[] arr = new int[2];
				arr[0] = 1; arr[1] = order++; //빈도는 1, 순서 지정
				map.put(n, arr); //첫등장하는 숫자일 시 
			}
			else { //나온 적 있는 숫자면
				int[] arr = map.get(n); //n의 빈도, 등장순서 가져오기
				arr[0]+=1;  //빈도 늘리기
				map.put(n, arr); //넣기
			}
		}
		///////////////////////입력완료//////////////////////////
		List<Map.Entry<Integer, int[]>> list = new ArrayList<>(map.entrySet());
		Collections.sort(list, (o1, o2)-> {
			int[] o1_value = o1.getValue(); //어떤 숫자의 빈도와, 등장순서
			int[] o2_value = o2.getValue(); //다른 숫자의 빈도와, 등장순서
			
			if(o1_value[0] == o2_value[0]) { //빈도 같으면?
				return o1_value[1]-o2_value[1]; //등장순서가 더 낮은 숫자가 앞으로
				// o1[1]이 더 낮은 숫자여야 음수가 나와 교환하지 않음
			}else { //빈도 다르면?
				return o2_value[0]-o1_value[0]; //빈도가 더 큰 숫자가 앞으로
				// o1[0]이 더 큰 숫자여야 양수가 나와 교환됨
			}
		});
		//////////////////////정렬완료/////////////////////////
		StringBuilder res = new StringBuilder(); //결과출력용
		for (Map.Entry<Integer, int[]> entry : list) {
			for (int i = 0; i < entry.getValue()[0]; i++) { //빈도만큼 숫자출력
				res.append(entry.getKey()+" "); 
			}
		}
		System.out.println(res);
	}
}

<숫자: [빈도, 등장순서]>
Map을 위와 같은 형태로 정의했다.

while(st.hasMoreTokens()) {
			int n = Integer.parseInt(st.nextToken());
			//
			if (!map.containsKey(n)) { //n이 나온적 없는 숫자면
				int[] arr = new int[2];
				arr[0] = 1; arr[1] = order++; //빈도는 1, 순서 지정
				map.put(n, arr); //첫등장하는 숫자일 시 
			}
			else { //나온 적 있는 숫자면
				int[] arr = map.get(n); //n의 빈도, 등장순서 가져오기
				arr[0]+=1;  //빈도 늘리기
				map.put(n, arr); //넣기
			}
		}

map에 수열을 입력하는 과정이다. 입력된 숫자의 등장 여부에 따라 코드가 다르게 실행된다.

List<Map.Entry<Integer, int[]>> list = new ArrayList<>(map.entrySet());
//
Collections.sort(list, (o1, o2)-> {
			int[] o1_value = o1.getValue(); //어떤 숫자의 빈도와, 등장순서
			int[] o2_value = o2.getValue(); //다른 숫자의 빈도와, 등장순서
            //
			if(o1_value[0] == o2_value[0]) { //빈도 같으면?
				return o1_value[1]-o2_value[1]; //등장순서가 더 낮은 숫자가 앞으로
				// o1[1]이 더 낮은 숫자여야 음수가 나와 교환하지 않음
			}else { //빈도 다르면?
				return o2_value[0]-o1_value[0]; //빈도가 더 큰 숫자가 앞으로
				// o2[0]이 더 큰 숫자여야 양수가 나와 교환됨
			}
		});

그다음 Map의 모든 Entitiy를 불러와 리스트 형태로 만든 후
value([빈도, 등장순서]) 값을 기반으로 정렬하였다.


결과


후기

첨에 LinkedHashMap써봤는데 원하는대로 안나와서 좀 더 공부를 한 다음에 쓰기로 했다^^;;;;

profile
일단 뭐라도 해보는 중
post-custom-banner

0개의 댓글