ACdream群赛15 I 字典树-优快云博客

题目中提示了都是小写字母，很明显的字典树，但是内存微微有些大。

需要记录每个点到叶节点的最短距离。

#include<iostream>
#include<algorithm>
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<string>
#include<stdlib.h>
#include<vector>
#include<cmath>
#include<map>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define maxnode 1000010
#define sigma_size 26

char ts[maxnode];

//sigma_size=26 maxnode为节点最大数：字符串数*每个字符串最大长度+x
struct trie
{
	int ch[maxnode][sigma_size];//ch[a][3]=c表示第a个节点有一个子节点为'd'并且节点序列号为c
	int val[maxnode];//存各个字符串权值，题目不涉及权值时，可以存1
	int sz;
	int idx(char c) { return c - 'a';}//找到char c对应的id值
	void init()
	{
		sz = 1;
		memset(ch[0], 0, sizeof(ch[0]));
		memset(val,inf,sizeof(val));
	}
	void insert(char *s)//s为字符串，v为权值
	{
		int u = 0, n = strlen(s);
		for(int i = 0; i < n; i++)
		{
			int c = idx(s[i]);
			if(!ch[u][c])
			{
				memset(ch[sz], 0, sizeof(ch[sz]));
				val[sz] = n;
				ch[u][c] = sz++;
			}
			else
			{
				val[ch[u][c]]=min(n,val[ch[u][c]]);
			}
			u = ch[u][c];
		}
	}
	int find(char *s)//查找s，返回权值
	{
		int n=strlen(s);
		int u = 0;
		int ans=n;
		for(int i = 0; i<n; i++)
		{
			int c = idx(s[i]);
			if(!ch[u][c])
				break;
			ans=min(ans,val[ch[u][c]]-2*(i+1)+n+1);
			u = ch[u][c];
		}
		return ans;
	}
}Trie;

int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int n,i,j;
		scanf("%d",&n);
		scanf("%s",ts);
		Trie.init();
		Trie.insert(ts);
		for(i=1;i<=n;i++)
		{
			scanf("%s",ts);
			printf("%d\n",Trie.find(ts));
			Trie.insert(ts);
		}
	}
	return 0;
}