[JAVA] 중복 제거 방법(+정렬)

코딩테스트 공부를 하다 보면 원소들의 중복을 허용하지 않는 문제를 자주 접하게 됩니다.

이를 해결하기 위해 흔히 사용하는 두 가지 중복 제거 방법을 정리해 보았습니다.

추가로, 중복 제거 후 정렬하는 방법에 대해서도 살짝 적어보았습니다 🤗

 

1️⃣ distinct() 사용

int[]arr = new int[n];
arr[0] =3; arr[1]=2;

// 중복 제거
int[] arr2 = Arrays.stream(arr).distinct().toArray(); 
Arrays.sort(arr2);

 

sort 함수는 기본적으로 오름차순입니다.

내림차순 정렬하고 싶으면 `Collections.reverseOrder()` 를 사용하면 됩니다.

 

근데 `Collections.reverseOrder()`으로 정렬할 배열은 프리미티브 타입이 아닌 래퍼클래스 이어야 한다는 점!

정수의 경우, `int` 가 아닌 `Integer` 여야 한다는 것이죠.

 

따라서 `int []` 타입을 내림차순으로 정렬하고 싶으면 아래처럼 `Integer []`로 변경하여 정렬한 후, 다시 `int []`타입으로 변경하면 됩니다.

 

int[] arr1 = {3, 1, 2, 4};

// 1. int[] -> Integer[]
Integer[] arr2 = Arrays.stream(arr).boxed().distinct().toArray(Integer[]::new);

// 2. 내림차순 정렬
Arrays.sort(arr2, Collections.reverseOrder());

// 3. Integer[] -> int[]
int[] arr3 = Arrays.stream(arr2).mapToInt(Integer::intValue).toArray();
System.out.println(Arrays.toString(arr3));

• `boxed()` 메서드는 원시 타입에 대한 스트림 지원을 클래스 타입으로 전환해 줍니다.
  • ex. IntStream -> Stream <Integer>
• `mapToInt()` 메서드는 Stream을 IntStream으로 전환해 줍니다.
  • `int []` 를 스트림화하면 `IntStream` /  `Integer []`를 스트림화하면 `Stream <Integer>` 
• 다른 Stream을 int[]로 변경하고 싶으면 mapToInt() 사용하기

 

위 코드에서는 3가지 과정으로 분리하여 작성했지만, 한 줄로 아래와 같이 작성할 수도 있습니다.

int[] arr1 = {3, 1, 2, 4};

// 한 줄로 작성
int[] arr2 = Arrays.stream(arr)
                     .distinct()
                     .boxed()
                     .sorted(Comparator.reverseOrder())
                     .mapToInt(Integer::intValue)
                     .toArray();

 

2️⃣ HashSet 사용

집합은 중복을 허용하지 않습니다.

따라서 애초에 정답을 담는 자료형으로 HashSet을 사용하면, 이후에 따로 중복 제거를 하지 않아도 됩니다.

HashSet<Integer> set = new HashSet<>();
int[] answer = set.stream().sorted().mapToInt(Integer::intValue).toArray()

 

 

---

 

✅ 기본형 배열  <-> 래퍼 클래스 배열

🏷 Integer [] -> int [] 변환

Integer[] a = {1, 2, 3, 4};
int[] b = Arrays.stream(a).mapToInt(Integer::intValue).toArray();

 

🏷 int [] -> Integer [] 변환

int[] a = {1, 2, 3, 4};
Integer[] b = Arrays.stream(a).boxed().toArray(Integer[]::new);