Submission #1161987

---

Source Code Expand

#pragma optimization_level 3
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <cmath>
using namespace std;

typedef pair<int, int> P;
#define rep(i, n) for (int i=0; i<(n); i++)
#define all(c) (c).begin(), (c).end()
#define _1 first
#define _2 second
#define pb push_back
#define INF 1145141919
#define MOD 1000000007

int N, Q;
string S[100000];
int C[100000];
int R[100000][26][26];
int p[256];

class Trie {
public:
  Trie *a[26];
  int ctr;
  bool leaf;

  Trie() {
    for (int i=0; i<26; i++) a[i] = NULL;
    leaf = false;
    ctr = 0;
  }
  void insert(const char *s) {
    if (*s == '\0') {
      leaf = true;
      ctr++;
      return;
    }
    int k = *s - 'a';
    if (a[k] == NULL) {
      a[k] = new Trie();
    }
    a[k]->insert(s+1);
    ctr++;
  }

  int find(const char *s, int x) {
    if (*s == '\0') return leaf;
    int c = leaf;
    int k = *s - 'a';
    for (int i=0; i<26; i++) {
      if (a[i] != NULL) {
        R[x][i][k] += a[i]->ctr;
      }
    }
    if (a[k] == NULL) return c;
    c += a[k]->find(s+1, x);
    return c;
  }
};


signed main() {
  ios::sync_with_stdio(false); cin.tie(0);
  cin >> N;
  rep(i, N) cin >> S[i];
  cin >> Q;

  Trie trie;
  rep(i, N) trie.insert(S[i].c_str());
  rep(i, N) {
    C[i] = trie.find(S[i].c_str(), i);
  }
  rep(i, Q) {
    int k; cin >> k;
    k--;
    rep(i, 26) {
      char c;
      cin >> c;
      p[c - 'a'] = i;
    }
    int s = C[k];
    rep(i, 26) {
      rep(j, 26) {
        if (p[i] < p[j]) s += R[k][i][j];
      }
    }
    cout << s << "\n";
  }
  return 0;
}

Submission Info

Judge Result