[Java] 자바 2차원 배열 정렬 하는 법 간단히 알아보자!

2020-11-13

---

알고리즘 및 배열에 관련된 문제를 처리하다 보면, 각 배열을 정렬을 해야 하는 경우가 생기게 된다. 때문에 오늘은 이 정렬 중 2차원 배열을 정렬하는 방법을 알아보도록 하자.

 

흔히들 1차원 배열은 Array.sort(배열 이름)라는 메서드를 이용해서 손쉽게 정렬을 할 수 있다. 다만 2차원 배열의 경우에는 여기에 추가적인 조건을 제시하여야 한다. 우선 예제는 아래와 같다. (3행 2열의 배열을 우선적으로 만들어 주었다.)

---

ex)배열: arr

(3, 5)

(4, 2)

(3, 4)

---

그러면 바로 코드를 살펴보도록 하자

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

우선 2종류의 import를 실행해야 한다. 정렬을 위한 Arrays와 비교를 위한 Comparartor를 import 해주면 된다.

 

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

public class TemplateA {
		public static void main(String[] args)  {
			int [][] arr = {{3,5}, {4,2}, {3,4}};

			//2차월 배열 정렬 0번쨰 열 다음 1번째 열 기준(다중 배열 정렬)
			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]; 
			    	 }
		           }
		        });

다음은 구체적인 정렬 문이다. ' Arrays.sort(arr ' 까지는 동일하나 괄호를 닫기 전에 Comparator라는 클래스가 사용된다. 이는 정렬을 할 때, 자신이 원하는 조건으로 배열을 정렬할 수 있도록 해준다. (우리는 Comparator를 설명하는 것이 아니기 때문에 여기까지만 이야기하겠다.) 현재 정수를 정렬하는 것이기 때문에, 속성은 int를 선언해주고 자신이 원하는 조건을 compare안에 작성해준다. 

 

if(o1 [0] == o2[0]) 의 부분은 배열의 0번째 위치를 비교해 주고 해당 배열의 위치가 같으면,  return o1[1] - o2[1]; 를통해 오름차순 정렬을 해준다. (내림차순으로 바꾸고 싶으면  return o2[1] - o1[1]; 로 좌우만 바꾸어 주면된다.)

 

else { return o1[0] - o2 [0]; } 는 0번째 열부터 다르기 때문에 0번째 열 기준으로 오름차순으로 정렬해준다.

(마찬가지로 내림차순으로 바꾸고 싶으면  return o2[0] - o1 [0]; 로 좌우만 바꾸어 주면 된다.)

---

아래의 코드로 이제 최종적으로 코드를 출력해보도록 하자

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

public class TemplateA {
		public static void main(String[] args)  {
			int [][] arr = {{3,5}, {4,2}, {3,4}};

			//2차월 배열 정렬 0번쨰 열 다음 1번째 열 기준(다중 배열 정렬)
			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]; 
			    	 }
		           }
		        });
			
			for(int i = 0; i < 3; i++) {
				for(int j = 0; j < 2; j++) {
					System.out.print(arr[i][j] + " ");
				}
				System.out.println();
			}
		
		
		}
	}

콘솔 값에 정확히 0번 열 기준 오름차순 후 1번 열 기준 오름차순으로 정렬되어 출력된 것을 알 수 있다.