#include <bits/stdc++.h>
using namespace std ;
//TRIE
struct node{
node *nxt[26];
int is = 0;
node(){
is = 0;
for(int i = 0;i<26;i++){
nxt[i] = NULL;
}
}
};
void inser(string w,node *root){
for(auto ch:w){
if(root->nxt[ch-'a']==NULL){
root->nxt[ch-'a'] = new node();
}
root=root->nxt[ch-'a'];
}
root->is++;
}
int Nnode = 0;
vector<int> adj[3000001];
int mark[3000001];
void build(node *root){
int cur = Nnode;
for(int j = 0;j<26;j++){
if(root->nxt[j]!=NULL){
Nnode++;
adj[cur].push_back(Nnode);
mark[Nnode] = root->nxt[j]->is;
build(root->nxt[j]);
}
}
}
long long dp[3000001];
long long calc(int node){
if(dp[node]!=-1)return dp[node];
long long ma = 0 , sum = 0;
for(auto j:adj[node]){
sum+=mark[j];
}
for(auto j:adj[node]){
if(mark[j]>0){
ma = max(ma,calc(j));
}
}
return dp[node]=ma+sum;
}
int main(){
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int n;
cin>>n;
node *root = new node();
while(n--){
string s;cin>>s;
reverse(s.begin(),s.end());
inser(s,root);
}
build(root);
memset(dp,-1,sizeof dp);
long long all =0;
for(int i = 0;i<=Nnode;i++){
if(mark[i]){
all = max(all,calc(i)+mark[i]);
}
}
cout<<all<<endl;
return 0 ;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
42 ms |
94292 KB |
Output is correct |
2 |
Correct |
42 ms |
94316 KB |
Output is correct |
3 |
Correct |
43 ms |
94308 KB |
Output is correct |
4 |
Incorrect |
315 ms |
239876 KB |
Output isn't correct |
5 |
Correct |
56 ms |
94920 KB |
Output is correct |
6 |
Incorrect |
113 ms |
180684 KB |
Output isn't correct |
7 |
Incorrect |
125 ms |
180408 KB |
Output isn't correct |
8 |
Incorrect |
111 ms |
180252 KB |
Output isn't correct |
9 |
Incorrect |
136 ms |
183668 KB |
Output isn't correct |
10 |
Incorrect |
111 ms |
180260 KB |
Output isn't correct |