[Java] 2차원 배열 정렬

1. 정렬할 2차원 배열

int[][] arr = {{0,1}, {2,5}, {1,8}, {1,7}};

 

2. 정렬 이후의 결과

{{0,1}, {1,7}, {1,8}, {2,5}}

 

Comparator로 구현

import java.util.Arrays;
import java.util.Comparator;

int[][] arr = {{0,1}, {2,5}, {1,8}, {1,7}};

Arrays.sort(arr, new Comparator<int[]>() {
	
    @Override
    public int compare(int[] o1, int[] o2) {
    	if(o1[0] == o2[0])
        	return o1[1] - o2[1];
        else
        	return o1[0] - o2[0];
    }
});

 

 

Lambda로 구현

import java.util.Arrays;

int[][] arr = {{0,1}, {2,5}, {1,8}, {1,7}};

Arrays.sort(arr, (o1, o2) -> o1[0] == o2[0] ? o1[1] - o2[1] : o1[0] - o2[0]);