• 1. sorted() 란 무엇인가??
• 2. sorted() 의 사용법
  • 2-1. 객체 비교
  • 2-2. 커스텀(자신이 정한) 정렬 키로 비교하기
  • 2-3. 내림차순(반대)으로 정렬
    • String 문자열의 문자들을 역순으로 정렬
    • 기본타입 배열을 역순으로 정렬

알고리즘과 자료구조를 공부할 때, 정렬의 기능을 많이 써야하는 알고리즘 문제들 특성상

Stream 의 sorted() 를 굉장히 자주 사용하게 된다.

알고리즘 문제를 풀면서 sorted() 에 대해 알아보면서, 그리고 직접 써보면서 알아낸 것들을 정리해보고자 한다.

1. sorted() 란 무엇인가??

• Stream 을 정렬할 때 사용하는 Stream 의 중간연산자이며, 오름차순이 아닌 다른 방식으로 정렬할 땐 Comparator 또는 Comparable 을 사용해서 정렬 형식을 지정해준다.
  • 속도면에서는 본인이 직접 반복문을 통해서 정렬작업을 하는 것보다 느릴 수 있다.

    • 하지만 속도의 성능이 제일 중요한 상황이 아닌 이상, 하드웨어가 많이 발전하고 유지보수의 중요성이 높아지고 있는 현재엔 유용한 기능이라고 생각한다.

    • 왜냐면 Stream 의 가장 큰 장점은 가독성 즉, 유지보수성이라고 생각하기 때문이다.

2. sorted() 의 사용법

• 이제 지금까지의 공부와 경험에 기반해서 몇 가지 사용법을 정리해볼 것이다.

2-1. 객체 비교

• 비교 대상이 되는 객체에 Comparable 인터페이스를 구현해서 직접 compareTo() 를 오버라이딩 해놓으면, sorted() 메서드가 알아서 compareTo() 메서드를 사용해서 비교한다.

  • 이것은 비단 stream 의 sorted() 뿐만이 아닌 Arrays.sort(), Collections.sort() 에도 해당되는 사항이다.

• 예제

  • 이 예제는 객체의 name 문자열에서 2번째 문자를 기준으로 비교해서 객체를 오름차순 정렬을 하는 것이다.

  • class Scratch {
        public static void main(String[] args) {
            Scratch scratch = new Scratch();
            // String[] 배열, 비교 기준이 될 2번째 문자를 지정하기 위한 인덱스 1을 인자로 전달
            String[] strArr = scratch.getSortedStrArr(new String[]{"aba", "bca", "cac"}, 1); 
            System.out.println(Arrays.toString(strArr));
        }
            
        public String[] getSortedStrArr(String[] strings, int n) {
            return Arrays.stream(strings).sorted() // 먼저 문자열을 오름차순으로 정렬
                // 각 문자열들을 문자열과 지정된 n번째 문자를 인자로 전달해서 Str 클래스로 치환한다. 
                    .map(str -> new Str(str, str.charAt(n))) 
                    .sorted() // Str 클래스에 정의된 compareTo 를 통해 Str 클래스들을 정렬한다.
                    .map(Str::getName).toArray(String[]::new); // Str 클래스들을 getName() 메서드를 사용해서 다시 문자열들로 치환한다.
        }
    }
        
    class Str implements Comparable<Str> {
        String name;
        char ch;
            
        public Str(String name, char ch) {
            this.name = name;
            this.ch = ch;
        }
            
        String getName() {
            return name;
        }
            
        @Override
        public int compareTo(Str o) {
            return this.ch - o.ch; // 문자열 name 의 n번째 문자를 기준으로 오름차순 정렬
        }
    }
        
    // 출력 결과
    // [cac, aba, bca]
    

  • 출력 결과를 보면 문자열의 2번째 문자들을 기준으로 오름차순 정렬이 되어있는 모습을 볼 수 있다.

2-2. 커스텀(자신이 정한) 정렬 키로 비교하기

• 자신이 비교하고 싶은 값(정렬 키)을 마음대로 정하고, 그 해당 키로 비교해서 정렬하는 기능이다.

  • Comparator.comparing() 메서드는 함수 내부에서 반환받는 값이, 비교할 때의 기준값이 된다.
  • 이 예시는 각 문자열의 두번째 문자로 비교해서 정렬하는 코드다.

• 예시

  • class Scratch {
        public static void main(String[] args) {
            Scratch scratch = new Scratch();
            String[] st = {"aba", "bca", "cac"};
            String[] strings = Arrays.stream(st)
                    .sorted() // 오름차순 정렬
                    .sorted(Comparator.comparing(s -> s.charAt(1))) // 2번째 문자를 기준으로 비교해서 오름차순 정렬
                    .toArray(String[]::new); // String[] 배열로 모아서 반환
            System.out.println(Arrays.toString(strings));
        }
    }
        
    // 출력 결과
    // [cac, aba, bca]
    

2-3. 내림차순(반대)으로 정렬

1. String 문자열의 문자들을 역순으로 정렬

   • Comparator.reverseOrder() 를 인자로 주면 역정렬 - Comparator.reverseOrder() 는 Comparable 인터페이스를 반환한다.

   • 예제

     • class Scratch {
           public static void main(String[] args) {
               String s = "dafaafq";
               String collect = Stream.of(s.split(""))
                       .sorted(Comparator.reverseOrder())
                       .collect(Collectors.joining());
               System.out.println(collect);
           }
       }
              
       // 출력 결과
       // qffdaaa
       

     • 예제 설명

       1. 스트림에 문자열의 문자들을 하나씩 풀어서 넣어준다.

       2. 역순 정렬

       3. 다시 전부 합쳐서 String으로 반환 (joining 메서드에 인자가 하나도 없으면 그대로 붙여서 반환)

          • Collectors.joining() 은 인자를 사이사이에 이어 붙여서 String 으로 반환해준다.

2. 기본타입 배열을 역순으로 정렬

   • Stream 의 sorted() 메서드에서 Comparator.reverseOrder() 를 통해 역순정렬을 하는 것은, 객체 형태의 배열만 가능하다. - 그래서 기본타입 배열을 Comparator.reverseOrder() 를 통해 역순정렬 하려면 래퍼클래스로 포장해서 정렬해야한다.

   • 예시

     • class Scratch {
           public static void main(String[] args) {
               int[] A = {5, 3, 2, 4, 1};
               Integer[] tmp = Arrays.stream(A)
                       .boxed()
                       .sorted(Comparator.reverseOrder())
                       .toArray(Integer[]::new);
               System.out.println(Arrays.toString(tmp));
           }
       }
              
       // 출력 결과
       // [5, 4, 3, 2, 1]
       

     • 설명

       1. 스트림에 기본타입 숫자들을 풀어서 넣어준다.

       2. 래퍼클래스인 Integer 로 하나씩 포장한다.

       3. 역순으로 정렬한다.

       4. Integer[] 타입으로 역순 정렬된 배열을 반환한다.

     • 만약 여기서 역순 정렬된 배열을 Integer[] 타입이 아닌 int[] 타입으로 반환받고 싶다면,

       • class Scratch {
             public static void main(String[] args) {
                 int[] A = {5, 3, 2, 4, 1};
                 int[] ints = Arrays.stream(A)
                         .boxed()
                         .sorted(Comparator.reverseOrder())
                         .mapToInt(Integer::intValue)
                         .toArray();
                 System.out.println(Arrays.toString(ints));
             }
         }
                  
         // 출력 결과
         // [5, 4, 3, 2, 1]
         

       • 설명

         1. Integer 값들을 sorted 로 역순 정렬한다.
         2. mapToInt() 에 Integer::intValue 메서드를 줘서 int 타입으로 하나씩 변환시켜서 IntStream 을 반환받는다.
         3. IntStream 의 toArray() 을 통해 int[] 로 반환받는다.