44-tgyuuAn

[tgyuuAn]

🔗 문제 링크

개미굴

✔️ 소요된 시간

30분

그냥 solved.ac 6랭크에서 아무거나 골랐는데,

문빈님이 최근에 풀었던 트라이가 바로 떠올랐음.

트라이그림은 알고있었지만 막상 구현해보려니 생각대로 되지 않아서 트라이 자료구조 레퍼런스를 보고 만들었음.

아직 생각대로 잘 되지 않았다. 더 많은 연습이 필요할 듯.

✨ 수도 코드

그림에 나와있는 그대로 트라이 자료구조를 만들면 되는 문제이다.

트라이 자료구조를 아느냐 모르느냐를 판가름 하는 문제인 것 같다.

트라이 자료구조는 #155 (comment) 에 올라온 문빈님 PR에서 긴략하게 소개되어있는데,

실제로 구현해보면 또 다른 느낌이다.

이 때는 레퍼런스를 보면 좋다.

자료구조를 만들고 데이터를 다 삽입했더라면,

가장 루트 노드부터 사전 순으로 완탐을 진행해주면 된다.

이 부분도 트라이 자료구조에 대한 개념이 많이 부족해서 구현하는 데 시간이 좀 걸렸다.

새로운 유형을 접해보니 재밌기도 했지만 조금 힘들기도 했다.

체화할 때 까지 트라이 문제를 좀 풀어봐야겠다.

이번에는 레퍼런스를 보고 구현을 했지만,

다음 번에는 레퍼런스를 보지 않고 스스로 구현해봐야겠따.

📚 새롭게 알게된 내용

• 트라이 자료구조를 알게되었다.
• 직접 구현해보니까 아는 거랑 또 다른 느낌이었다.
• 계속해서 풀어봐야지..

[H0ngJu]

문빈님 pr 이후에 바로 트라이 문제를 풀어보셨군요 실행력 ㄷㄷ!!
트라이 공부하다가 딕셔너리부터 다시 보느라 리뷰가 쬐끔 늦어졌습니다 ... 😅

저도 다음주~다다음주 DP 공부가 끝나면 트라이 문제 풀어볼려고 문제도 찾아놨슴다 ㅎㅅㅎ

이번주도 문제푸느라 고생하셨슴다-! 💪

[H0ngJu]

파이썬에서의 class... 거의 잊고 지냈네요
다시 class 개념 찾아보면서 복습하고 갑니다 🔥🔥

[pknujsp]

이전 문제는 리프노드인지 판단하는게 필요해서
좀 풀기 까다로웠는데
이 문제는 큰 조건없이 편하게 풀 수 있어서 좋았습니다.

저도 트라이 자료구조를 따라서 풀었습니다.

• 먼저 모든 원소들을 트리형태로 만들고
• DFS를 진행하는데
• 이 때 사전순으로 부모 노드를 탐색하면서
• 출력할 문자열을 깊이에 따라서 적절하게 만들어 따로 리스트에 추가한뒤에
• 최종적으로 완성된 리스트를 출력

사전 순으로 나타내야 함을 문제에서 요구해서
처음에 너무 복잡하게 생각했는데 막상 풀면서 보니 그리 까다롭지 않았고
생각보다 단순하게 풀어볼 수 있었습니다.

N = int(input())
tree = {}

for _ in range(N):
    lst = input().split()

    if lst[1] not in tree:
        tree.setdefault(lst[1], {})
    node = tree[lst[1]]
    
    for i in range(2, len(lst)):
        if lst[i] not in node:
            node.setdefault(lst[i], {})
        node = node[lst[i]]

result = []

def dfs(node, depth):
    for char in sorted(node.keys()):
        depths = '--' * depth
        result.append(depths + char)
        dfs(node[char], depth + 1)

for start in sorted(tree.keys()):
    result.append(start)
    dfs(tree[start], 1)

for v in result:
    print(v)

[Munbin-Lee]

#include <iostream>
#include <map>

using namespace std;

struct Tree {
    struct Node {
        map<string, Node *> children;

        Node *getOrCreate(string &name) {
            if (children.contains(name)) return children[name];
            return children[name] = new Node();
        }
    };

    void print(Node *cur, int depth = 0) {
        for (auto [k, v]: cur->children) {
            for (int i = 0; i < depth; i++) {
                cout << "--";
            }
            cout << k << '\n';
            print(v, depth + 1);
        }
    }

    Node *root = new Node;
};

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N;
    cin >> N;

    Tree *tree = new Tree();

    while (N--) {
        int K;
        cin >> K;

        Tree::Node *cur = tree->root;

        while (K--) {
            string t;
            cin >> t;
            cur = cur->getOrCreate(t);
        }
    }

    tree->print(tree->root);

    return 0;
}

트라이 PR 보고 바로 풀어보는 능력자ㄷㄷㄷㄷ

[Munbin-Lee]

로직이 저와 거의 같아서 드릴 말씀이 없네요.

굿입니다.