題干：

時間限制:10000ms

單點時限:1000ms

內存限制:512MB

描述

給定一個包含N個單詞的字典:{W1, W2, W3, ... WN}，其中第i個單詞Wi有具有一個權值Vi。 ?

現在小Hi要進行M次查詢，每次查詢包含一個前綴字符串Pi和一個后綴字符串Si。他希望知道同時以Pi為前綴并且以Si為后綴的單詞中，權值最大的單詞的權值是多少？

假設字典包含"hihocoder"、"hijacker"和"hiker"，權值依次是30、20和10。 ?

那么對于查詢前綴="hi"，后綴="er"，答案應為30.

輸入

第一行包兩個整數N和M。(1 <= N <= M) ?

以下N行每行包含一個只包含小寫字母的字符串Wi和對應的權值Vi。 ?

再之后M行每行包含兩個字符串Pi和Si。

對于30%的數據，1 <= N, M <= 100 ?

對于100%的數據，1 <= N, M <= 50000, 1 <= |Wi|, |Pi|, |Si| <= 10， 1 <= Vi?<= 100000

輸出

輸出最大的權值。如果沒有符合條件的單詞，輸出-1。

樣例輸入

3 2 hihocoder 30 hijacker 20 hiker 10 hi er hihoco hocoder

樣例輸出

30 30

解題報告：

? ??查詢有前綴和后綴的條件，考慮固定其中一個條件；單詞長度很短，在每個單詞的每個前綴下再建一顆后綴字典樹，在后綴字典樹下維護權值。查詢時，先匹配前綴，再查詢后綴。

? ?這題還有第二種方法，字典樹哈希，先建同一棵字典樹然后按照每個點的root當做seed去哈希，前綴與后綴用一個模數分開（這里取的1000007）map去哈希同時維護最大權值

AC代碼1：

#include<cstdio> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<vector> #include<unordered_map> #include<set> #include<string> #include<cmath> #include<cstring> #define ll long long #define ull unsigned long long #define pb push_back #define pm make_pair using namespace std; const int MAX = 2e6 + 5; char s[55]; int trie[MAX][33]; int val[MAX]; int tot; void insert(char str[],int w) {int rt = 0,tmprt,len = strlen(str);for(int i = 0; i<len; i++) {int cur = str[i] - 'a';if(!trie[rt][cur]) trie[rt][cur] = ++tot;rt = trie[rt][cur];//接下來建立后綴字典樹if(!trie[rt][27]) trie[rt][27] = ++tot;tmprt = trie[rt][27];for(int j = len-1; j>=0; j--) {int cur = str[j] - 'a';if(!trie[tmprt][cur]) trie[tmprt][cur] = ++tot; tmprt = trie[tmprt][cur];val[tmprt] =max(val[tmprt],w);}} } int ask(char str[],char qq[]) {int res = 0,len = strlen(str),rt = 0;for(int i = 0; i<len; i++) {int cur = str[i] - 'a';if(trie[rt][cur] == 0) return -1;else rt = trie[rt][cur];}if(trie[rt][27] == 0) return -1;rt = trie[rt][27];len = strlen(qq);for(int i = 0; i<len; i++) {int cur = qq[i] - 'a';if(trie[rt][cur] == 0 ) return -1;else rt = trie[rt][cur];}return val[rt]; } int main() {int n,m;cin>>n>>m;for(int w,i = 1; i<=n; i++) {scanf("%s%d",s,&w);insert(s,w);}char a[55],b[55];while(m--) {scanf("%s%s",a,b);int lenb = strlen(b);reverse(b,b+lenb);printf("%d\n",ask(a,b));} return 0 ;}/* 1 1 aaa 2 a a*/

上面這份代碼有幾個細節：注意建樹的時候tmprt和rt不能混用，需分開，第二畫圖一下就可以發現對每一個節點建后綴樹的時候都會新建一個新的空節點（這也是寫完自己花了一下圖才發現的），這樣有個好處，那就是不需要做繁瑣的坐標運算，因為否則你的trie數組的第二維就得開26*2這么大，第一維就是1e6就行，而現在新開一個空節點當根節點，保持了代碼的統一性，就相當于完全就是新建一棵字典樹，代碼不需要動太多，減少出錯率。

還有一點，ask的時候別忘了查完前綴的時候rt=trie[rt][27]這一步、、、不然肯定GG（因為你繼續搜索相當于順著正向的字典樹查下去了）

AC代碼2：

#include<cstdio> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<vector> #include<unordered_map> #include<set> #include<string> #include<cmath> #include<cstring> #define ll long long #define ull unsigned long long #define pb push_back #define pm make_pair using namespace std; const int MAX = 2e6 + 5; char s[55]; int trie[MAX][26]; int Hash[MAX*20]; int tot; void insert(char str[],int a[]) {int rt = 0,len = strlen(str);for(int i = 0; i<len; i++) {int cur = str[i] - 'a';if(!trie[rt][cur]) trie[rt][cur] = ++tot;rt = trie[rt][cur];a[i] = rt;} } int ask(char str[]) {int res = 0,len = strlen(str),rt = 0;for(int i = 0; i<len; i++) {int cur = str[i] - 'a';if(trie[rt][cur] == 0) return -1;else rt = trie[rt][cur];}return rt; } unordered_map<ll,int>g; int A[55],B[55]; int main() {int n,m;cin>>n>>m;for(int w,i = 1; i<=n; i++) {scanf("%s%d",s,&w);insert(s,A);int len = strlen(s);reverse(s,s+len);insert(s,B);//至此，a和b數組的0~len-1都有對應的值了 for(int j = 0; j<len; j++) {for(int k = 0; k<len; k++) {ll HS = (ll)A[j] * 1000007 + B[k];//Hash[int(HS%mod)] = MX(Hash[int(HS%mod)],w);//因為權值w都是正數，所以不需要像那個代碼一樣unorderedmap的時候分存在和不存在這一說、、、（主要是這不是迭代器而是數組。。所以都已經分配空間了） if(!g.count(HS))g[HS]=w;else if(w>g[HS])g[HS]=w;}}}char a[55],b[55];while(m--) {scanf("%s%s",a,b);int lena = strlen(a);int lenb = strlen(b);reverse(b,b+lenb);int x = ask(a);int y = ask(b);int ans = -1;if(x!=-1&&y!=-1) {ll w=(ll)x*1000007+y;if(g.count(w))ans=g[w];}printf("%d\n",ans);} return 0 ;}

?

總結

以上是生活随笔為你收集整理的【Hihocoder - offer编程练习赛39 - D】前缀后缀查询（后缀字典树，哈希，思维）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。