P2922 [USACO08DEC] Secret Message G - 字典树

题面

分析

并不是统计恰好匹配，而是部分匹配，可能是目标字符串短，也可能目标字符串长，可以多开一个数组记录途中每个字符串有多少个字符串经过，同时统计以某个字符结尾的字符串个数，然后在查询时就可以边查询边加上以当前字符结尾的字符串，也就是比目标字符串短并且匹配的字符串，然后在匹配成功后，再加上以当前结尾字符串的经过数量，因为经过的和结尾的是重复的，所以还要减去当前结尾的数量。

代码

#include 

using namespace std;

const int N = 1e6 + 10;

int son[N][2];
int vis[N];
int sum[N];
int idx;

void insert(string s) {
    int p = 0;
    for(int i = 0; i < s.size(); i ++) {
        int c = s[i] - '0';
        if(!son[p][c]) son[p][c] = ++ idx;
        p = son[p][c];
        sum[p] ++;
    }
    vis[p] ++;
}

int query(string s) {
    int ans = 0;
    int p = 0;
    for(int i = 0; i < s.size(); i ++) {
        int c = s[i] - '0';
        if(!son[p][c]) return ans;
        p = son[p][c];
        ans += vis[p];
    }
    ans += sum[p] - vis[p];
    return ans;
}

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

    int m, n;
    cin >> m >> n;
    for(int i = 0; i < m; i ++) {
        string s;
        int k;
        cin >> k;
        while(k --) {
            string x;
            cin >> x;
            s += x;
        }
        insert(s);
    }
    for(int i = 0; i < n; i ++) {
        string s;
        int k;
        cin >> k;
        while(k --) {
            string x;
            cin >> x;
            s += x;
        }
        cout << query(s) << "\n";
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64