I. 문자열 내 마음대로 정렬하기

https://school.programmers.co.kr/learn/courses/30/lessons/12915

1. 문제 설명

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

2. 제한 사항

• strings는 길이 1 이상, 50이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

3. 풀이

주어진 문자열들의 일정한 인덱스에 위치한 글자를 기준으로 정렬하는 문제. 여러 방법이 있겠지만, Comparable Interface와 Class를 사용해 구현하고자 했다.

결론적으로, 다음과 같이 진행했다.

1. class StringSet를 Comparable을 implement해 선언한다.
2. class의 field는 n번째 문자의 아스키 코드를 저장하는 int asci와 전체 문자열을 저장하는 String str로 구성된다.
3. @Override를 통해 compareTo의 로직을 수정한다.

        @Override
        public int compareTo(Object o) {
            StringSet s = (StringSet) o;
            if(this.asci>s.asci){
            	// 현재 변수의 asci가 비교 대상 변수의 asci보다 클 때 1 리턴
                return 1;
            }
            else if(this.asci<s.asci){
            	// 현재 변수의 asci가 비교 대상 변수의 asci보다 작을 때 1 리턴
                return -1;
            }
            else{
            	// 값이 같은 경우 str끼리 compareTo를 사용해 다시 비교
                return str.compareTo(s.str);
            }
        }

4. List<StringSet> listarr을 선언하고, 주어진 문자열 배열의 문자열마다 StringSet의 구조에 맞춰 변환해 삽입한다.
5. Collections.sort(listarr)을 통해 재정의된 compareTo를 기준으로 값들을 정렬한다.
6. 정렬 결과를 반환한다.

4. 소스 코드

import java.util.*;
import java.lang.*;

class Solution {
    static class StringSet implements Comparable {
        // field
        int asci;
        String str;

        // constructor
        public StringSet(int asci, String str){
            this.asci = asci;
            this.str = str;
        }

        @Override
        public int compareTo(Object o) {
            StringSet s = (StringSet) o;
            if(this.asci>s.asci){
                return 1;
            }
            else if(this.asci<s.asci){
                return -1;
            }
            else{
                return str.compareTo(s.str);
            }
        }
    }

    static public String[] solution(String[] strings, int n) {
        String[] answer = {};
        List<StringSet> listarr = new ArrayList<>();
        for (int i = 0; i < strings.length; i++) {
            listarr.add(new StringSet(strings[i].charAt(n), strings[i]));
        }
        Collections.sort(listarr);
        answer = new String[strings.length];
        for (int i = 0; i < answer.length; i++) {
            answer[i] = listarr.get(i).str;
        }
        return answer;
    }
}