LeetCode #3093

Longest Common Suffix Queries

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
struct Trie {
    struct Node {
        vector<pair<char, Node*>> nxt;
        int idx, len;
        Node():Node(INT_MAX, INT_MAX) { nxt.reserve(26); }
        Node(int idx, int len):idx(idx),len(len) {}
    };

    Node* head;
    Trie() { head = new Node(); }

    void Insert(const string &S, int idx) {
        Node *cur = head;
        int N = (int)S.size();
        if(N < cur->len) cur->len = N, cur->idx = idx;
        for(int i = N - 1; i >= 0; --i) {
            int nxtIdx = (int)cur->nxt.size();
            for(int j = 0; j < (int)cur->nxt.size(); ++j) {
                if(cur->nxt[j].first == S[i]) nxtIdx = j;
            }
            if(nxtIdx == cur->nxt.size()) {
                cur->nxt.resize(cur->nxt.size() + 1);
                cur->nxt[nxtIdx] = make_pair(S[i], new Node(idx, N));
            }
            cur = cur->nxt[nxtIdx].second;
            if(N < cur->len) cur->len = N, cur->idx = idx;
        }
    }
    int Query(const string &S) {
        Node *cur = head;
        int N = (int)S.size();
        for(int i = N - 1; i >= 0; --i) {
            int nxtIdx = (int)cur->nxt.size();
            for(int j = 0; j < (int)cur->nxt.size(); ++j) {
                if(cur->nxt[j].first == S[i]) nxtIdx = j;
            }
            if(nxtIdx == cur->nxt.size()) break;
            cur = cur->nxt[nxtIdx].second;
        }
        return cur->idx;
    }
};

class Solution {
public:
    vector<int> stringIndices(vector<string>& wordsContainer, vector<string>& wordsQuery) {
        Trie trie;
        int idx = 0;
        for(const auto &a: wordsContainer) trie.Insert(a, idx ++);
        vector<int> answer;
        for(const auto &a: wordsQuery)  answer.push_back(trie.Query(a));
        return answer;
    }
};

int init = [] {
    ios_base::sync_with_stdio(false);  cin.tie(nullptr);
    return 0;
}();

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.

GITHUB COMMUNITY

커뮤니티 평가

GitHub 계정으로 남긴 최신 평가 1건만 반영하며, 데이터는 매일 저장소에 동기화됩니다.

최근 동기화

체감 난이도

아직 평가 없음

첫 난이도 평가를 남겨주세요.

문제 추천

아직 평가 없음

이 문제가 도움이 되었는지 알려주세요.

이 문제 평가하기

GitHub 로그인 후 난이도를 제출하고, 원하는 경우 추천 여부도 함께 남길 수 있습니다.

추천 여부 (선택)

평가하려면 GitHub로 로그인해 주세요.

평가는 자동으로 저장소 Discussion에 기록되므로 Discussion 화면을 직접 열 필요가 없습니다. 같은 문제를 다시 평가하면 기존 평가가 갱신됩니다.

DISCUSSION

댓글

GitHub 로그인 후 작성할 수 있으며 모든 댓글은 이 저장소의 Discussions에 보관됩니다.