[BOJ 7088 // C++] Word counting

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 7088번 문제인 Word counting이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/7088

 

루트에서 서로 다른 두 리프로 가는 경로에 있는 (탐색하고자 하는) 문자열의 등장 횟수는 (두 리프의 LCA까지의 등장횟수) + (각 경로의 문자를 포함하는 등장횟수)로 생각할 수 있다. 따라서 KMP를 각 노드의 자식별로 이어서 진행하는 것으로 문제를 해결할 수 있다.

 

경로별로 KMP를 루트부터 다시 시작할 필요가 없음을 확인하자. LCA까지 등장하는 문자는 동일하므로, 실패함수(pi배열)도 동일하기 때문이다.

 

아래는 제출한 소스코드이다.

#include <iostream>
#include <vector>
using namespace std;

int N, ans;
string S; int Slen;
vector<pair<int, string>> G[15000];
int pi[15000015];

void dfs(int cur, int ii) {
    for (auto &p:G[cur]) {
        int nxt = p.first; S += p.second;
        int SSlen = S.length();
        for (int i = ii; i < SSlen; i++) {
            int idx = pi[i - 1];
            while (idx && S[idx] != S[i]) idx = pi[idx - 1];
            if (S[idx] == S[i]) pi[i] = idx + 1;
            else pi[i] = idx;
            if (pi[i] == Slen) ans++;
        }
        dfs(nxt, SSlen);
        int K = p.second.length();
        while (K--) S.pop_back();
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> N;
    for (int k = 1; k < N; k++) {
        int x, y; string s; cin >> x >> y >> s;
        G[x].emplace_back(make_pair(y, s));
    }

    cin >> S; Slen = S.length(); S += '#';
    for (int i = 1; i <= Slen; i++) {
        int idx = pi[i - 1];
        while (idx && S[idx] != S[i]) idx = pi[idx - 1];
        if (S[idx] == S[i]) pi[i] = idx + 1;
        else pi[i] = idx;
    }
    
    dfs(0, Slen + 1);

    cout << ans;
}

 

랜덤 마라톤 · 코스 14 F번