首页 > 代码库 > URAL 1542. Autocompletion 字典树

URAL 1542. Autocompletion 字典树

给你最多10w个单词和对应的频率 接下来最多1w5千次询问 每次输入一个字符串让你从前面的单词中按照频率从大到小输出最多10个以该字符串为前缀的单词

开始把单词建成了字典树 然后每次询问找到所有满足条件的单词 在排序输出 不是超时就是超内存 还来了一发数组越界

最后换方法 因为最多只要输出前10个 那么可以把要询问的字符串建字典树 每个结尾节点在做一个映射 存10个单词(当然只是存下标)

然后再把单词按照频率从大到小排序一个一个插入字典序 满足有前缀是结尾节点就放进去

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int maxnode = 400110;
const int sigma_size = 26;
const int maxn = 100010;
const int maxm = 15010;
int ch[maxnode][sigma_size];
int val[maxnode];
int sz;

struct word
{
	int v;
	char s[22];
}w[maxn];

char p[maxm][22];
int ans[maxm][12];
int sum[maxm], id[maxn];
bool cmp(word a, word b)
{
	if(a.v != b.v)
		return a.v > b.v;
	return strcmp(a.s, b.s) < 0 ? 1 : 0;
}
void init()
{
	sz = 1;
	memset(ch[0], 0, sizeof(ch[0]));
	memset(sum, 0, sizeof(sum));
}

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


void find(char *s, int v)
{
	int n = strlen(s), u = 0;
	for(int i = 0; i < n; i++)
	{
		int c = s[i]-'a';
		if(!ch[u][c])
			return;
		u = ch[u][c];
		if(val[u] && sum[val[u]] < 10)
		{
			ans[val[u]][sum[val[u]]] = v;
			sum[val[u]]++;
		}
	}
}

int main()
{
	int n, m;
	while(scanf("%d", &n) != EOF)
	{
		init();
		for(int i = 0; i < n; i++)
		{
			int x;
			scanf("%s %d", w[i].s, &w[i].v);
		}
		sort(w, w+n, cmp);
		scanf("%d", &m);
		for(int i = 0; i < m; i++)
		{
			scanf("%s", p[i]);
			insert(p[i], i+1);
		}
		for(int i = 0; i < n; i++)
		{
			find(w[i].s, i);
		}
		for(int i = 0; i < m; i++)
		{
			int u = 0;
			for(int j = 0; p[i][j]; j++)
				u = ch[u][p[i][j]-'a'];
			u = val[u];
			if(i)
				puts("");
			for(int j = 0; j < sum[u]; j++)
				printf("%s\n", w[ans[u][j]].s);
		}
	}
	return 0;
}


URAL 1542. Autocompletion 字典树