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

문제

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

프로그래머스

 

 

정답

import java.util.*;
class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = new String[strings.length];
        String[] alpha = new String[strings.length];
        ArrayList<String> wordList = new ArrayList<>();
        ArrayList<String> answers = new ArrayList<>();
        int count = 0;
        
        for(int i = 0; i < alpha.length; i++) {
            alpha[i] = String.valueOf(strings[i].charAt(n));
        }
        
        Arrays.sort(alpha);
        
        for(int i = 0; i < alpha.length; i++) {
            for(int j = 0; j < strings.length; j++) {
                if(alpha[i].equals(String.valueOf(strings[j].charAt(n)))) {
                    wordList.add(strings[j]);
                    count++;
                }
            }
            if(count == 1) {
                answers.add(wordList.remove(0));
                count = 0;
            } else {
                if(answers.contains(wordList.get(0))) {
                    wordList.clear();
                    count = 0;
                    continue;
                } else {
                    Collections.sort(wordList);
                    for(String word:wordList) {
                        answers.add(word);
                    }
                    wordList.clear();
                    count = 0;
                }
            }
        }
        for(String al:alpha) {
            System.out.println(al);
        }
        
        int i = 0;
        for(String a:answers) {
            answer[i++] = a;
        }
        
        return answer;
    }
}

 

strings에 있는 n번째 문자를 alpha에 저장하고 정렬을 해주었습니다.

그 다음에 이 alpha를 반복문으로 돌리면서 alpha와 같은 글자가 있는 단어가 1개 있는 경우 wordList에 추가해주고 만약에 2개 이상인 경우에는 첫 한번만 wordList에 추가해주고 그 다음으로 같은 단어가 있을때(wordList에 존재하면) 추가를 안하고 넘어가는 방법으로 문제를 풀었습니다.