[프로그래머스]문자열 내 마음대로 정렬하기 Level-1

문제 설명

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

제한 조건

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

입출력 예

stringsnreturn

[sun, bed, car]	1	[car, bed, sun]
[abce, abcd, cdx]	2	[abcd, abce, cdx]

입출력 예 설명

입출력 예 1
sun, bed, car의 1번째 인덱스 값은 각각 u, e, a 입니다. 이를 기준으로 strings를 정렬하면 [car, bed, sun] 입니다.

입출력 예 2
abce와 abcd, cdx의 2번째 인덱스 값은 c, c, x입니다. 따라서 정렬 후에는 cdx가 가장 뒤에 위치합니다. abce와 abcd는 사전순으로 정렬하면 abcd가 우선하므로, 답은 [abcd, abce, cdx] 입니다.

<내 코드>

import java.util.*;
import java.util.ArrayList;
import java.util.Collections;

class Solution {
  public String[] solution(String[] strings, int n) {
    //String[] answer = {};  
	Arrays.sort(strings, null);

	Arrays.sort(strings, new Comparator<String>() {
		@Override
		public int compare(String o1, String o2) {
			return String.valueOf(o1.charAt(n)).compareTo(String.valueOf(o2.charAt(n)));
		}
	});
      	
      return strings;
  }
}

최대한 간결함을 중점으로 두고 짜보았습니다.

초기 구상으로는 2번째 문자만 가져와서 오름차순 정렬해주면 끝나겠다 싶어서

	Arrays.sort(strings, new Comparator<String>() {
		@Override
		public int compare(String o1, String o2) {
			return String.valueOf(o1.charAt(n)).compareTo(String.valueOf(o2.charAt(n)));
		}
	});

이렇게 해버렸는데 이게 입출력 예 2 의 경우를 제대로 판별하지 못했습니다.

입출력 예 2
abce와 abcd, cdx의 2번째 인덱스 값은 c, c, x입니다. 따라서 정렬 후에는 cdx가 가장 뒤에 위치합니다. abce와 abcd는 사전순으로 정렬하면 abcd가 우선하므로, 답은 [abcd, abce, cdx] 입니다.

 그래서 앞에다 아무런 조건 없이 오른차순으로 sort를 넣었습니다.

결국 1번째 sort에서는 n번째 문자가 같은 경우를 대비하여 오름차순 정렬

2번째에서는 문제의 원래 조건에 맞게 n번째 문자에 따른 오름차순 정렬이라고 보시면 될 듯합니다.

<다른 분 코드>

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = {};
        ArrayList<String> arr = new ArrayList<>();
        for (int i = 0; i < strings.length; i++) {
            arr.add("" + strings[i].charAt(n) + strings[i]);
        }
        Collections.sort(arr);
        answer = new String[arr.size()];
        for (int i = 0; i < arr.size(); i++) {
            answer[i] = arr.get(i).substring(1, arr.get(i).length());
        }
        return answer;
    }
}

이 분의 코드는 제가 비록 매우 빠르진 않지만 참신하다고 생각하여 가져왔습니다.

발생은 <n번째 문자를 기준으로 정렬임으로 n번째 문자를 문자열의 맨 앞에 붙여준 후에 정렬한다> 입니다. 

역시나 정말 지혜로우신 분이 많습니다.