[Java] - 과일 장수 (135808) (역순 정렬/ Comparator, Collections.reverseOrder())

📚 문제 - 135808

과일 장수가 사과 상자를 포장하고 있습니다. 사과는 상태에 따라 1점부터 k점까지의 점수로 분류하며, k점이 최상품의 사과이고 1점이 최하품의 사과입니다. 사과 한 상자의 가격은 다음과 같이 결정됩니다.

• 한 상자에 사과를 m개씩 담아 포장합니다.
• 상자에 담긴 사과 중 가장 낮은 점수가 p (1 ≤ p ≤ k)점인 경우, 사과 한 상자의 가격은 p * m 입니다.

과일 장수가 가능한 많은 사과를 팔았을 때, 얻을 수 있는 최대 이익을 계산하고자 합니다.(사과는 상자 단위로만 판매하며, 남는 사과는 버립니다)

예를 들어, k = 3, m = 4, 사과 7개의 점수가 [1, 2, 3, 1, 2, 3, 1]이라면, 다음과 같이 [2, 3, 2, 3]으로 구성된 사과 상자 1개를 만들어 판매하여 최대 이익을 얻을 수 있습니다.

• (최저 사과 점수) x (한 상자에 담긴 사과 개수) x (상자의 개수) = 2 x 4 x 1 = 8

사과의 최대 점수 k, 한 상자에 들어가는 사과의 수 m, 사과들의 점수 score가 주어졌을 때, 과일 장수가 얻을 수 있는 최대 이익을 return하는 solution 함수를 완성해주세요.

제한사항

• 3 ≤ k ≤ 9
• 3 ≤ m ≤ 10
• 7 ≤ score의 길이 ≤ 1,000,000
• 1 ≤ score[i] ≤ k
• 이익이 발생하지 않는 경우에는 0을 return 해주세요.

입출력 예

k	m	score	result
3	4	[1, 2, 3, 1, 2, 3, 1]	8
4	3	[4, 1, 2, 2, 4, 4, 4, 4, 1, 2, 4, 2]	33

• 입출력 예 #1
  문제의 예시와 같습니다.
• 입출력 예 #2

  사과 상자	가격
  [1, 1, 2]	1 x 3 = 3
  [2, 2, 2]	2 x 3 = 6
  [4, 4, 4]	4 x 3 = 12
  [4, 4, 4]	4 x 3 = 12

  다음과 같이 사과 상자를 포장하여 모두 팔면 최대 이익을 낼 수 있습니다. 사과 상자 가격 [1, 1, 2] 1 x 3 = 3 [2, 2, 2] 2 x 3 = 6 [4, 4, 4] 4 x 3 = 12 [4, 4, 4] 4 x 3 = 12 따라서 (1 x 3 x 1) + (2 x 3 x 1) + (4 x 3 x 2) = 33을 return합니다.

 

⌨️ 작성한 코드

import java.util.Arrays;

class Solution {
    public int solution(int k, int m, int[] score) {
        int answer = 0;
        if (score.length >= m) {
            Arrays.sort(score); // 점수를 오름차순으로 정렬
            reverse(score); // 정렬된 점수 배열을 내림차순으로

            int boxes = score.length / m; // 총 상자의 개수
            for (int i = 1; i <= boxes; i++) {
                answer += score[m * i - 1] * m;
            }
        }
        return answer;
    }

    public static void reverse(int[] arr) {
        int n = arr.length;
        for (int i = 0; i < n / 2; i++) {
            int temp = arr[i];
            arr[i] = arr[n - i - 1];
            arr[n - i - 1] = temp;
        }
    }
}

• 접근 방식:
  최저 사과 점수가 상자의 가격을 좌우하니까 점수를 역순으로 정렬해서 m을 곱하면 편할 거라 생각했다.  reverse()를 만들어서 역정렬했다. 그 다음 for문을 통해 가격을 계산한 뒤 return 되도록 했다.
  
  
• 문제&해결:
  처음에는 Collections.reverseOrder()를 사용하려고 했는데, 이는 Integer 형식이어야 쓸 수 있는 것이었다. 그래서 reverse()를 따로 만들어서 역정렬했다.
  
  
• 최적화: 사실 내림차순 정렬해서 문제를 푸는 것이 비효율적이라고 생각했다. 그냥 배열의 뒤쪽부터 for문을 돌면 나을 것같아서 코드를 수정했다.
  

  import java.util.Arrays;
  
  class Solution {
      public int solution(int k, int m, int[] score) {
          int answer = 0;
          Arrays.sort(score);
  
          for(int i = score.length; i >= m; i -= m){
              int index = i - m;
              answer += score[index] * m;
          }
          return answer;
      }
  }

  
  왼쪽은 원래 코드, 오른쪽이 변경한 코드 결과이다. 변경한 편이 코드도 간결하고 속도도 조금 더 빠르다.

 

✅ 배운 점

[ 역순 정렬 ]

• Comparator
  Comparator 인터페이스를 구현하여 compare() 메서드를 오버라이드한다. compare() 메서드 내에서는 두 값의 순서를 바꾸어 역순으로 비교한다. 그 후, Arrays.sort() 메서드를 사용하여 배열을 정렬한다.

  import java.util.Arrays;
  import java.util.Comparator;
  
  public class Main {
      public static void main(String[] args) {
          Integer[] array = {5, 3, 9, 1, 7};
  
          // 역순으로 정렬하는 Comparator 구현
          Comparator<Integer> reverseComparator = new Comparator<Integer>() {
              @Override
              public int compare(Integer o1, Integer o2) {
                  return o2.compareTo(o1); // 역순으로 비교
              }
          };
  
          // 배열을 역순으로 정렬
          Arrays.sort(array, reverseComparator);
  
          // 정렬된 배열 출력
          System.out.println(Arrays.toString(array));
      }
  }

• Collections.reverseOrder() 메소드
  Collections.reverseOrder()를 사용하여 Comparator 객체를 생성하고, 이를 Arrays.sort() 메서드의 두 번째 인자로 전달한다. Collections.reverseOrder()는 기본적으로 오름차순 정렬을 역순으로 바꿔주는 Comparator를 생성한다.
  Collections.reverseOrder()는 기본적으로 모든 타입의 객체에 대해서 역순으로 정렬할 수 있다.
  
  1. Integer: 정수형
  2. Double: 실수형
  3. String: 문자열
  4. 기타 Comparable 인터페이스를 구현한 모든 클래스

  import java.util.Arrays;
  import java.util.Collections;
  
  public class Main {
      public static void main(String[] args) {
          Integer[] array = {5, 3, 9, 1, 7};
          Arrays.sort(array, Collections.reverseOrder());
          System.out.println(Arrays.toString(array)); // 역순으로 정렬된 배열 출력
      }
  }