結果
問題 | No.1994 Confusing Name |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
#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;}