[백준]18870번 좌표 압축(java)

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

18870번: 좌표 압축

백준 18870번 좌표 압축 문제 링크입니다.

 

 

핵심

여러 가지 수를 받고, 작은 값이 0 값을 갖고 그 뒤로 1씩 증가되는 형태이다. 이때 핵심은 HashMap을 사용하는 것이다.

 

정답 코드

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
public class Main {
    private static Map<Integer, Integer> map = new HashMap<>();
    private static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[] nums = new int[N];
        int[] sortNums = new int[N];
        for (int i = 0; i < N; i++) {
            int num = sc.nextInt();
            nums[i] = num;
            sortNums[i] = num;
        }
        Arrays.sort(sortNums);
        
        int cnt = 0;
        for (int i = 0; i < N; i++) {
            if (map.get(sortNums[i]) == null) { 
                map.put(sortNums[i], cnt++);
            }
        }        

        for (int i = 0; i < N; i++) {
            sb.append(map.get(nums[i])).append(" ");
        }
        System.out.println(sb.toString());
    }
}

 

 

해설

HashMap을 선언한다. 

int형 배열 num과 그와 같은 배열 sortNums를 선언한다.

이 두 배열에 같은 값을 집어넣는다.

이후 sortNums는 정렬을 시킨다.

예제 입력을 보면 2 4 -10 4 -9가 들어왔는데 이를 정렬시키면, -10 -9 2 4 4가 된다.

map에 key값이 지정되지 않았을 경우에 새로운 key를 지정하고, value값은 0부터 넣어준다.

즉 -10이란 key는 0이란 value를 -9란 key는 1이란 value를 갖는다.

이후 4가 연속으로 나오는데, 4라는 key는 이미 존재하므로 value의 값은 변경되지 않는다.

출력할 때는 원래 배열의 순서대로 key를 2 4 -10 4 9 순서로 주었기 때문에 2 3 0 3 1이 된다.

 

 

 

개선점이나 오류가 있다면 댓글 부탁드립니다.