結果

問題 No.1994 Confusing Name
ユーザー MasKoaTS
提出日時 2022-03-30 20:43:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 533 ms / 2,000 ms
コード長 1,210 bytes
コンパイル時間 877 ms
コンパイル使用メモリ 78,708 KB
最終ジャッジ日時 2025-01-28 13:21:17
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#define rep(i, l, n) for (int i = (l); i < (n); i++)
using namespace std;
using str = string;
template <class T>
using V = vector<T>;
template <class T>
using VV = V<V<T> >;


struct Node {
	int num = 0;
	Node* next[26] = { NULL };
};

struct Trie {
	Node* root = new Node;

	void add(V<int>& lis) {
		Node* last = root;
		for (auto i : lis) {
			if (last->next[i] == NULL) {
				last->next[i] = new Node;
			}
			last = last->next[i];
		}
		(last->num)++;
	}

	bool search(V<int>& lis) {
		Node* last = root;
		for (auto i : lis) {
			if (last->next[i] == NULL) {
				return 0;
			}
			last = last->next[i];
			//cout << (char)(i + 'a') << endl;
		}
		return last->num;
	}
};


int main(void) {
	int n;	cin >> n;
	V<str> s(n);
	VV<int> lis(n, V<int>({}));
	Trie trie;
	rep(i, 0, n) {
		cin >> s[i];
		for (auto c : s[i]) {
			lis[i].push_back(c - 'a');
		}
		trie.add(lis[i]);
	}

	for (auto l : lis) {
		V<int> cl = l;
		int ans = -(int)l.size();
		rep(j, 0, 26) {
			cl[0] = j;
			ans += trie.search(cl);
		}
		rep(i, 1, l.size()) {
			cl[i - 1] = l[i - 1];
			rep(j, 0, 26) {
				cl[i] = j;
				ans += trie.search(cl);
			}
		}
		cout << ans << endl;
	}

	return 0;
}
0