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

프로그래머스 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의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

 

입출력 예

입출력 예

한 줄 요약

주어진 배열에서 n번째 문자를 기준으로 정렬해 출력한다. 동일 문자가 나올 수 있으므로 n번째 문자를 기존 문자열 앞에 붙여 새로운 문자열을 만들어 활용한다.

 

내 코드

1. 반환할 배열 answer의 길이는 strings배열과 같으므로 strings.length로 지정한다.

2. n번째문자를 기존 문자열에 붙여 만든 새로운 문자열을 넣을 ArrayList를 생성한다.

3. strings의 길이만큼 반복문을 사용해 새로운 문자열(strings[i].charAt(n) + strings[i])을 넣어준다.

   (ex) sun -> usun, car -> acar)

4. Collections.sort()함수로 정렬한다.

5. 정렬된 배열을 substring을 사용해 1~마지막으로 자른다.

import java.util.*;

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

 

n번째 문자를 기존 문자열에 붙여 중복값에 영향을 안 받는 것과 Collections.sort()함수 사용이 핵심!