백준/실버

접두사 찾기 - JAVA, Trie 구조

아찌방 2025. 6. 28. 01:03

 

출처 : https://www.acmicpc.net/problem/14426

 

 

 

 

 

코드 & 풀이

 

import java.util.*;
import java.io.*;

public class 백준_접두사찾기_14426 {
    static class TrieNode {
        Map<Character, TrieNode> children = new HashMap<>();
    }

    static class Trie {
        TrieNode root = new TrieNode();

        void inset(String word) {
            TrieNode node = root;
            for (char ch : word.toCharArray()) {
                node = node.children.computeIfAbsent(ch, c -> new TrieNode());
            }
        }

        boolean starts_with(String prefix) {
            TrieNode node = root;
            for (char ch : prefix.toCharArray()) {
                if(!node.children.containsKey(ch)) return false;
                node = node.children.get(ch);
            }

            return true;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        Trie root = new Trie();
        for (int i = 0; i < N; i++) {
            String word = br.readLine();
            root.inset(word);
        }

        int count = 0;
        for (int i = 0; i < M; i++) {
            String prefix = br.readLine();
            if (root.starts_with(prefix)) count++;
        }
        System.out.println(count);
    }
}

 

 

Trie(트라이) 자료구조라는게 있는데

 

예를 들어서 apartment가 있고 apple가 있으면

 

a - p 까지는 공유

 

이후부터 갈라짐

 

a - p

                          p - p - l - e

                                        a - r - t - m - e - n - t

 

이런식

 

insert는 위 처럼 문자들을 나눠서 넣어주는 거임

 

해시 맵에 a 있어?

 

→ yes → a 의 자식들 가져와

→ no → 새로 넣어

 

starts_with 는 이제 주어진 단어가 다른 문장의 접두사가 되는지 체크

 

해시 맵에 a 있어?

 

→ yes → 다음 단어 탐색 → 끝까지 ㄱㄱ

→ no → 실패

 

 

 

 

 

다음에 또 봐요

 

728x90