[JAVA][Level1]PROGRAMMERS 문자열 내 마음대로 정렬하기

 

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

---

문제 설명

문자열로 구성된 리스트 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"] 입니다.

주어진 문제

class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = {};
        return answer;
    }
}

---

나의 문제풀이

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        // answer 배열 문자열 배열의 크기로 초기화
        String[] answer = new String[strings.length];

        // 먼저 사전순으로 정렬하기 위해 문자열배열을 오름차순 정렬해준다.
        Arrays.sort(strings);
        // LinkedHashMap을 이용해서 key값으로 문자열,
        // value값으로 문자로 하여 나중에 value값으로 정렬을 하기위해 선언한다.
        Map<String, Character> map = new LinkedHashMap<>();
        // for-each문을 이용해 문자열배열의 값을 key값으로 넣어주고
        // value는 해당 문자열의 n인덱스 위치의 문자를 넣어준다.
        for (String word : strings) {
            map.put(word, word.charAt(n));
        }

        // list를 이용해 map을 list에 넣어 Collections.sort()를 이용해서 정렬 해준다.
        // 기존 map과 동일한 형태의 map을 선언하고 for-each를 이용해 정렬한 key-value값을 넣는다.
        List<Map.Entry<String, Character>> entries = new LinkedList<>(map.entrySet());
        Collections.sort(entries, (o1, o2) -> o1.getValue().compareTo(o2.getValue()));
        LinkedHashMap<String, Character> sorted = new LinkedHashMap<>();
        for (Map.Entry<String, Character> entry : entries) {
            sorted.put(entry.getKey(), entry.getValue());
        }

        // 정렬이 끝난 LinkedHashMap의 value값을 for-each를 이용해 answer에 넣어준다.
        int cnt = 0;
        for (Map.Entry<String, Character> entry : sorted.entrySet()) {
            answer[cnt++] = entry.getKey();
        }
        
        return answer;
    }
}

• 먼저 제한 조건에서 인덱스 1의 문자가 같은 문자열이 여럿 일 경우 사전순으로 정렬한다고 하여 문자열을 정렬하기 전에 먼저 오름차순으로 정렬을 하고 시작했다.
• key-value형태로 문자열과 인덱스의 문자를 연결하여 사용하면서 순서를 유지하기 위해 LinkedHashMap을 사용한다.
• key에는 문자열, value에는 인덱스 위치의 문자를 넣어준다.
• LinkedHashMap을 value기준으로 정렬을 하기 위해서 List에 Map.entrySet()을 사용하여 key값과 value값을 한 번에 가져온다.
• Collections.sort()를 이용해서 list를 정렬해 주는데, o1, o2의 value값을 비교하여 정렬해 준다. (list안의 오브젝트인 map을 순차적으로 비교하는 것)
• 정렬한 list의 entry를 새로운 LinkedHashMap에 넣어준다.
• LinkedHashMap에서 key값을 answer 배열에 순차적으로 넣어준다.

알고리즘을 풀면서 공부를 하기 위해서 안 써본 개념들을 쓰기 위해서 HashMap을 사용했는데, 너무 복잡하게 생각을 해서 정답까지 많이 돌아간 느낌은 있었지만 그 과정안에서 많은 점을 배울 수 있었다.
비교적 Map을 사용한 적이 없었는데 이번 경험으로 실패과정이 많아서 이해하는데 도움이 되었고, Map.Entry를 이용해서 key, value를 한 번에 컨트롤하고 이걸 list안으로 넣어서 사용할 수 있어 Map도 활용성이 많다는 걸 알 수 있었다.
다른 사람들의 풀이를 보았을 때는 느낌이 정말 다르긴 했다. 앞서 말했었지만 문자열을 변형해서 하는 방법들이 다양해서 이 부분도 활용성이 많다고 했었는데 이문제에서도 문자를 추가하고 정렬하는 방법들을 사용하면서 손쉽게 풀 수 있는 방법들이 있었고, 정렬을 하는 방법도 다양하게 존재하고 자바에서는 Comparator를 객체로 생성해하고 싶은데로 정렬을 시켜 간단하게도 접근할 수 있다는 걸 배웠다.

 

문제 출처 - https://school.programmers.co.kr/learn/courses/30/lessons/12915