정렬 - Comparable<T> ~ compareTo(T o1)

▶ 유형

• 내림차순, 오름차순 같은 정렬을 어떤 기준에 맞춰서 해야될 때가 있는데 Comparable을 사용하면 좋다

자바로 문제를 풀다보면 자료형 담아둘 클래스를 매우 자주 선언한다. 클래스 선언할 때 Comparable 인터페이스를 implements해두고, compareTo 메서드를 @Override해주면 된다.

implements Comparable<T>

@Override
public int compareTo(T o) {
	return 0 or 1 or -1
}

 

이번에 연습용으로 문제 풀면서 사용한 Node 클래스를 기준으로 보자.

class Node implements Comparable<Node>{
    int v, w;
    Node(int v, int w){
        this.v = v;
        this.w = w;
    }
    
    @Override
    public int compareTo(Node o) {
        if(this.w == o.w) {
            return Integer.compare(o.v, this.v); // 가중치가 같으면 값이 큰게 먼저 와야됨
        }else {
            return Integer.compare(this.w, o.w); // 가중치가 더 큰게 멀리가야됨(오름차순)
        }
    }
}

compareTo의 기준점을 암기해두는 게 중요하다. 항상 this가 기준이고, compareTo의 매개변수로 들어오는게 비교할 대상이다.

각 primitive 자료형에는 compare 함수가 있는데 비교하는 두 값 중 오른쪽이 크면 1, 같으면 0, 작으면 -1을 반환한다.

즉 오름차순으로 만들고 싶을때는 오른쪽이 더 크게 만들고, 내림차순으로 하고싶을 때는 작게 만들면 된다. 같은 경우에 한번더 비교시켜서 정렬 조건을 세분화 하는 것도 가능하다. 

 

이렇게 만든 compareTo는, 이 자료형으로 만든 list를 sort(null) 해주면 알아서 정렬된다. null을 넣어주면 내가 구현한 Comparable이 들어가기 때문이다.

▶ 전략

외워야 되는거라서 그냥 무조건 많이 쳐봐야된다. 프로그래머스같은 자동완성이 안되는 환경에도 수월하게 쓸 줄 알아야한다.

https://school.programmers.co.kr/learn/courses/30/lessons/120880

프로그래머스

import java.util.*;

class Node implements Comparable<Node>{
    int v, w;
    Node(int v, int w){
        this.v = v;
        this.w = w;
    }
    
	@Override
	public int compareTo(Node o) {
		if(this.w == o.w) {
			return Integer.compare(o.v, this.v); // 가중치가 같으면 값이 큰게 먼저 와야됨
		}else {
			return Integer.compare(this.w, o.w); // 가중치가 더 큰게 멀리가야됨(오름차순)
		}
	}
}

class Solution {
    public int[] solution(int[] numlist, int n) {
        int[] answer = new int[numlist.length];
        List<Node> arr = new ArrayList<Node>();
        // 거리가 같으면 더 큰수를 반환함
        int idx = 0;
        for(int i: numlist){
            arr.add(new Node(numlist[idx], Math.abs(n-numlist[idx++])));
        }
        
        arr.sort(null);
        idx = 0;
        for(Node i: arr){
            answer[idx++] = i.v;
        }
        
        return answer;
    }
}

 

 

도움이 됐다면 댓글이나 공감 버튼 한 번씩 누르고 가주세요!