Submission #921637

---

Source Code Expand

#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <climits>
#include <iostream>
#include <algorithm>
#include <set>
#include <map>
#include <queue>
#include <vector>
#include <sstream>
#include <typeinfo>
#include <fstream>

#define DIV 1000000007

using namespace std;

long long N;
string S[100005];

int k[100005];
string rules[100005];

//               ch1 < ch2 なら　小さくなる
int memo[100005][26][26];

struct node {
	node *next[26] = {};
	int num[26];
	int fin;
};
node *root = new node();
 
void add(string s) {
	node *curr = root;
	for (char c : s) {
		if (curr->next[c - 'a'] == NULL) {
			curr->next[c - 'a'] = new node();
			for(int i = 0; i < 26; i++){
				curr->next[c-'a']->num[i] = 0;
			}
			curr->next[c-'a']->fin = 0;
		}
		curr->num[c-'a']++;
		curr = curr->next[c - 'a'];
	}
	curr->fin++;
}

long long count(int kdx, string rule){
	long long ret = 0;
	for(int i = 0; i < 26; i++){
		for(int j = i + 1; j < 26; j++){
			int ch1 = rule[i] - 'a';
			int ch2 = rule[j] - 'a';
			ret += memo[kdx][ch1][ch2];
		}
	}
	return ret + memo[kdx][0][0] + 1;

}


void create_table(int idx){
	node *curr = root;

	for (char c : S[idx] ) {
		long long tmp = 0;
		for(int i = 0; i < 26; i++){
			if(i != c - 'a'){
				memo[idx][i][c-'a'] += curr->num[i];
			}
		}
		memo[idx][0][0] += curr->fin;
		curr = curr->next[c - 'a'];
	}


}



int main(){
	cin >> N;

	//400000
	for(int i = 0; i < N; i++){
		cin >> S[i];
		add(S[i]);
	}

	for(int i = 0; i < N; i++){
		create_table(i);
	}


	long long Q;
	cin >> Q;
	for(int i = 0; i < Q; i++){
		cin >> k[i] >> rules[i];
		k[i]--;
	}

	/*
	for(int i = 0; i < N; i++){
		cout << "i=" << i  << endl;
		for(int j = 0; j < 26; j++){
			for(int k = 0; k < 26; k++){
				cout << "memo[" << i << "][" << (char)('a' + j) << "][" << (char)('a' + k) << "]=" << memo[i][j][k] << endl;
			}
		}
	}
	*/

	//1e5
	for(int i = 0; i < Q; i++){
		cout << count(k[i], rules[i]) << endl;
	}
}

Submission Info

Judge Result