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

문제 설명

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

제한 조건

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

입출력 예

strings	n	return
["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"] 입니다.

 

 

풀이 방법

 

문자열 배열 strings와 정수 n을 입력으로 받아서, 각 문자열을 n번째 문자를 기준으로 사전순으로 정렬하고 반환하는 함수를 만듭니다.

1. answer 배열을 초기화하고, 문자열 배열 strings와 같은 크기로 설정합니다.
2. Arrays.sort(strings)를 통해 문자열 배열 strings를 사전순으로 정렬합니다. (오름차순)
3. ASCII 코드를 이용하여 a부터 z까지의 모든 알파벳을 순회하는 반복문을 실행합니다. (a=97, z= 122)
4. 각 알파벳에 대해 strings 배열을 순회하면서 n번째 문자가 현재 알파벳과 일치하는지 확인합니다.
5. 일치하는 문자열이 있을 경우, 해당 문자열을 answer 배열의 인덱스 cnt에 할당하고 cnt 값을 증가시킵니다.
6. 모든 알파벳에 대한 반복이 끝나면, 정렬된 문자열이 담긴 answer 배열을 반환합니다

import java.util.*;
class Solution {
  public String[] solution(String[] strings, int n) {
      String[] answer = {};
        answer = new String[strings.length];
        int cnt = 0;
        
        Arrays.sort(strings);
        
        for(int i = 97; i < 123; i++) {
            for(int j = 0; j < strings.length; j++) {
                if(strings[j].charAt(n) == (char)i) {
                    answer[cnt++] = strings[j];
                }
            }
        }
        return answer;
  }
}