您的位置:首页 > 其它

最短的名字 CSU - 1115

2017-07-19 09:38 211 查看
字典树,因为没有一个人的名字是其他人的前缀,所以只要保存下每个节点经过的次数即可,次数为1时可以保证唯一。

两种方法,动态和静态

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
#include <cmath>
#include <set>
#include <queue>
using namespace std;

const int INF=1e9+10;
const double EPS = 1e-10;
typedef long long ll;

char a[100005];
int ch[100005][26];
int val[100005];
int sz,ans;

void insert(char *s){
int u=0,len=strlen(s);
for(int i=0;i<len;i++){
int c=s[i]-'a';
if(!ch[u][c]){
memset(ch[sz],0,sizeof(ch[sz]));
ch[u][c]=sz++;
}
u=ch[u][c];
val[u]++;
}
}

void find(int p,int cnt){
if(p&&(val[p]==1||val[p]==-1)){
ans+=cnt;
return;
}
for(int i=0;i<26;i++){
if(ch[p][i]){
find(ch[p][i],cnt+1);
}
}
}
int main(){
//freopen("out.txt","w",stdout);
int T;
scanf("%d",&T);
while(T--){
int n;
memset(val,0,sizeof(val));
memset(ch[0],0,sizeof(ch[0]));
sz=1;ans=0;
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%s",a);
insert(a);
}
find(0,0);
printf("%d\n",ans );
}
return 0;
}
#include<iostream>
#include
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
typedef struct node{
node *next[27];    //存放当前字符后面那一个字符的存放的地址
int count;          //记录此字符在这一层的个数
}node;

node *T;
char s[1000005];

void build(node* T)             //建树
{
node *q,*p=T;
int l=strlen(s);
for(int i=0;i<l;i++) int="" k="s[i]-'a';" p-="">next[k]==NULL)
{
node *t;
t=(node*)malloc(sizeof(node));
t->count=1;
for(int i=0;i<26;i++)
t->next[i]=NULL;
p->next[k]=t;
p=t;
}
else
{
p=p->next[k];
p->count++;
}
}
}

void release(node *T)           //释放结点
{
for(int i=0;i<26;i++)
if(T->next[i]!=NULL)
release(T->next[i]);
free(T);
}

int search(node *T)         //查询
{
node *q=T;
int sum=0;
for(int i=0;i<26;i++)
if(T->next[i]!=NULL)
{
q=T->next[i];
sum+=q->count;
if(q->count>1)
sum+=search(q);   //计数
}
return sum;
}

int main()
{
int t,n;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
T=(node*)malloc(sizeof(node));
T->count=0;
for(int i=0;i<26;i++)
T->next[i]=NULL;
for(int i=1;i<=n;i++)
{
scanf("%s",s);
build(T);
}
printf("%d\n",search(T));
release(T);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: