洛谷 P2922 / bzoj1590【Trie】

#include 

using namespace std;

const int N = 1e6 + 10;
struct Node
{
  int next[2];
  Node ()
  {
    memset (next, 0, sizeof next);
  }
} z[N];
int ans[N], cnt[N], nd, cnt2[N];
vector <int> a[N], b[N];

void cls()
{
  nd = 0;
  for (int i = 0; i < N; i ++)
    z[i] = Node();
  memset (cnt, 0, sizeof cnt);
}

void insert(vector <int> &a)
{
  int len = a.size(), now = 0;
  for (int i = 0; i < len; i ++)
  {
    if (!z[now].next[a[i]])
      z[now].next[a[i]] = ++ nd, now = nd;
    else
      now = z[now].next[a[i]];
    cnt2[now] ++;
  }
  cnt[now] ++;
}

int query(vector <int> &a)
{
  int len = a.size(), now = 0, ans = 0;
  for (int i = 0; i < len; i ++)
  {
    if (!z[now].next[a[i]])
      return ans;
    else
      now = z[now].next[a[i]];
    ans += cnt[now];
  }
  return ans - cnt[now] + cnt2[now];
}

int main()
{
  int n, m;
  cin >> n >> m;
  cls();
  for (int i = 1; i <= n; i ++)
  {
    int tot;
    cin >> tot;
    for (int j = 1; j <= tot; j ++)
    {
      int x;
      cin >> x;
      a[i].push_back(x);
    }
    insert(a[i]);
  }
  for (int i = 1; i <= m; i ++)
  {
    int tot;
    cin >> tot;
    for (int j = 1; j <= tot; j ++)
    {
      int x;
      cin >> x;
      b[i].push_back(x);
    }
    ans[i] = query(b[i]);
  }
  for (int i = 1; i <= m; i ++)
    cout << ans[i] << '\n';
  return 0;
}

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86