문제

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

제한조건

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

입출력 예

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

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

코드

static String[] solution(String[] strings, int n) {
    String[] answer = {};
    answer = new String[strings.length];
    String temp = "";

    for (int i = 0; i < strings.length - 1; i++) {
	for (int j = i + 1; j < strings.length; j++) {
	    char ch1 = strings[i].charAt(n);
	    char ch2 = strings[j].charAt(n);

	    if (ch1 > ch2) {
		temp = strings[i];
		strings[i] = strings[j];
		strings[j] = temp;
	    } else if (ch1 == ch2) {
		int num = strings[i].compareTo(strings[j]);

		if (num > 0) {
	            temp = strings[i];
		    strings[i] = strings[j];
		    strings[j] = temp;
		}
	     }
	  }
     }
    answer = strings;
    return answer;
}

chartAt() 함수

• char charAt(int index)
• 입력받은 인자값(index) 의 위치에 있는 문자를 뽑아내는 함수

compareTo() 함수

• 문자열의 사전순 값을 비교하여 그에 해당하는 int값을 리턴한다.

String str1 = "AA";
String str2 = "BB";
String str3 = "AA";

System.out.println(str1.compareTo(str2));	// 결과 -1
System.out.println(str2.compareTo(str1));	// 결과 1
System.out.println(str3.compareTo(str1));	// 결과 0

더 좋은 코드

static String[] solution2(String[] strings, int n) {
    String[] answer = {};

    ArrayList<String> arr = new ArrayList<String>();
    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).toString().substring(1);
    }
    return answer;
}

Collections.sort() 함수

정렬을 하기 위해서는 리스트 속 데이터 간 크기 비교가 가능해야한다. 즉, 리스트 안에 들어있는 Object들이 수치화되어야 한다.
Java에서는 비교가 가능한(수치화가 가능한) Object들은 Comparable 인터페이스를 implements 하고있으며, 우리가 사용하는 대부분의 데이터 클레스가 속한다.
결과적으로 sort메소드는 Comparable 인터페이스를 구현한 오브젝트 타입의 ArrayList라면 특별한 구현이 없이도 가능하다.

참고문서