URAL 1542. Autocompletion 字典树
Posted llguanli
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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 字典树的主要内容,如果未能解决你的问题,请参考以下文章