[LeetCode] 472번 Concatenated Words 문제를 풀어보았다. (ft. java)

2023년 01월 27일 금요일 - 아니 왤케 추워...

오늘 올려볼 문제는 472번 Concatenated Words 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

하드치고는 너무 쉽게 풀어버림...

입력

사진에서도 볼 수 있듯이 String 배열 1개가 입력으로 들어온다.

---

풀이 및 코드

주어진 String들을 이어붙였을 때 만들어낼 수 있는 값중 배열에 있는 모든 값들을 찾아서 리턴하는 문제이다.

오늘은 처음부터 정답을 생각해냈다.

먼저 String의 길이순으로 정렬한다.

이 후 substring으로 잘라가면서 dfs를 하는 식으로 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    HashSet<String> set = new HashSet<>();
    public List<String> findAllConcatenatedWordsInADict(String[] words) {
        var result = new ArrayList<String>();
        Arrays.sort(words, (a, b) -> a.length() - b.length());
        
        for(var w : words) {
            for(int i = 1; i < w.length(); i++) {
                if(set.contains(w.substring(0, i)) && solve(i, w)) {
                    result.add(w);
                    break;
                }
            }
            set.add(w);
        }
        
        return result;
    }
    
    public boolean solve(int start, String word) {
        if(start == word.length()) return true;
        
        for(int i = start + 1; i <= word.length(); i++) if(set.contains(word.substring(start, i)) && solve(i, word)) return true;
        
        return false;
    }
}

---

제출 화면

이게 되나 싶었는데 이게 됐다... ㄷㄷ

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.