99클럽 코테 스터디 15일차 prefix-and-suffix-search

📝 문제

https://leetcode.com/problems/prefix-and-suffix-search/description/

💦 풀이 실패

문제를 이해하는데 어려움이 있었다.

예시 Input으로 3중으로 배열이 구성되어있어서 도저히 무슨 의미인지 이해하지 못해서 결국 Solution을 보게되었다.

 

Solution을 보고 이해한 문제를 설명하자면 생성자의 인자로 String 배열을 받는다.

f()라는 함수의 인자로는 2개의 String이 주어지는데 위 생성자에서 받은 배열의 문자열 중에

f() 함수의 인자 각각을 prefix와 suffix로 같는 문자열중 배열내 가장 큰 index를 같는 문자열의 index를 반환하도록 한다.

 

LeetCode로는 처음 문제풀이를 해보는데 클래스를 제출하는게 살짝 생소했다.

import java.util.HashMap;
import java.util.Map;

class WordFilter {
    Map<String, Integer> map = new HashMap<>();

    public WordFilter(String[] words) {
        for (int i = 0; i < words.length; i++) {
            for (int j = 1; j <= words[i].length(); j++) {
                String pref = words[i].substring(0, j);

                for (int k = words[i].length() - 1; k >= 0; k--) {
                    String suff = words[i].substring(k);

                    map.put(pref + "&" + suff, i);
                }
            }
        }
    }

    public int f(String pref, String suff) {
        return map.getOrDefault(pref + "&" + suff, -1);
    }
}