Submission #921539

---

Source Code Expand

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
//#include<cctype>
#include<climits>
#include<iostream>
#include<string>
#include<vector>
#include<map>
//#include<list>
#include<queue>
#include<deque>
#include<algorithm>
//#include<numeric>
#include<utility>
//#include<memory>
#include<functional>
#include<cassert>
#include<set>
#include<stack>
#include<random>

const int dx[] = {1, 0, -1, 0};
const int dy[] = {0, -1, 0, 1};
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef pair<int, int> pii;

const int MAXN = 100100;
string S[MAXN];

struct Node {
    int sum;
    int id;
    int child[26];
    Node() : sum(0), id(-1) {
        memset(child, -1, sizeof(child));
    }
};

int trieNum = 1;
Node trie[4*MAXN];

void addTrie(int from, int num) {
    if (trie[from].child[num] == -1) trie[from].child[num] = trieNum++;
}

int cnt[26][26];
int ans[MAXN][26][26];
int pf[MAXN];

void calc(int now, int p) {
    if (trie[now].id >= 0) {
        int id = trie[now].id;
        pf[id] = p++;
        for (int i = 0; i < 26; i++) {
            for (int j = 0; j < 26; j++)
                ans[id][i][j] = cnt[i][j];
        }
    }
    for (int i = 0; i < 26; i++) {
        if (trie[now].child[i] >= 0) {
            for (int j = 0; j < 26; j++) {
                if (i != j && trie[now].child[j] >= 0) {
                    int id = trie[now].child[j];
                    cnt[i][j] += trie[id].sum;
                }
            }
            calc(trie[now].child[i], p);
            for (int j = 0; j < 26; j++) {
                if (i != j && trie[now].child[j] >= 0) {
                    int id = trie[now].child[j];
                    cnt[i][j] -= trie[id].sum;
                }
            }
        }
    }
}
int pos[26];

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N, Q;
    cin >> N;
    for (int i = 0; i < N; i++) {
        cin >> S[i];
        int now = 0;
        for (char c : S[i]) {
            trie[now].sum++;
            addTrie(now, c-'a');
            now = trie[now].child[c-'a'];
        }
        trie[now].sum++;
        trie[now].id = i;
    }
    calc(0, 0);
    cin >> Q;
    while (Q--) {
        int k;
        string perm;
        cin >> k >> perm;
        --k;
        for (int i = 0; i < 26; i++)
            pos[perm[i]-'a'] = i;
        int res = 1 + pf[k];
        for (int i = 0; i < 26; i++) {
            for (int j = 0; j < 26; j++) {
                if (pos[i] > pos[j]) res += ans[k][i][j];
            }
        }
        cout << res << endl;
    }
    return 0;
}

Submission Info

Judge Result