(构造+字典树)
题意:给定
n
个字符串
思路:这个题很巧妙,难点是有前缀和两个条件共同作用。而这题关键是观察到每个字符串长度都很小,并且想到先固化某一个条件的方法。观察到这个单词长度较小,于是可以拆单词,将一个单词拆成10个单词:它的第
i
个前缀连接上它的逆序列。这样把拆出来的第
吐槽:一直用定长数组做要么RE要么TLE,于是改用指针动态分配空间的方法做就AC了,以后字典树还是用指针动态分配空间的方法比较好。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define LL long long
using namespace std;
const double eps = 1e-8;
const int maxn = 10000;
struct node {
int cnt;
node *nxt[26];
node(){ cnt = -1; for(int i=0; i<26; i++) nxt[i] = NULL; }
};
struct Trie {
node *root;
void init() {
root = new node;
for (int i = 0; i < 26; i++)
root->nxt[i] = NULL;
}
void Insert(char *s, int val) {
node *p = root;
int len = strlen(s);
for (int i = 0; i < len; i++) {
int id = s[i] - 'a';
if (p->nxt[id] == NULL) {
node *q = new node();
p->nxt[id] = q;
p = p->nxt[id];
}
else {
p = p->nxt[id];
}
p->cnt = max(p->cnt, val);
}
}
int Query(char *s) {
node *p = root;
int len = strlen(s);
for (int i = 0; i < len; i++) {
int id = s[i] - 'a';
if (p->nxt[id] == NULL) return -1;
else {
p = p->nxt[id];
}
}
return p->cnt;
}
void Clear(node *p) {
if (p == NULL) return;
for (int i = 0; i < 26; i++) {
if (p->nxt[i] != NULL) Clear(p->nxt[i]);
}
free(p);
p = NULL;
}
} T[12];
int main()
{
//freopen("test.txt","r",stdin);
int n, q;
while(scanf("%d%d",&n,&q) == 2) {
for(int i=0; i<11; i++)
T[i].init();
for(int i=0; i<n; i++) {
int w = 0;
char s1[30], s2[30];
scanf("%s%d",s1,&w);
int len = strlen(s1);
for(int j=0; j<len; j++) {
int k = 0;
for(k=0; k<=j; k++) s2[k] = s1[k];
for(k=j+1; k<j+1+len; k++) s2[k] = s1[j+len-k];
s2[k] = '\0';
//printf("i:%d j:%d s:%s\n",i,j,s2);
T[j].Insert(s2, w);
}
}
while(q --) {
char pre[30], suf[30];
scanf("%s%s",pre,suf);
int len1 = strlen(pre), len2 = strlen(suf), k = len1;
for(k=len1; k<len1+len2; k++) pre[k] = suf[len1+len2-1-k];
pre[k] = '\0';
//printf("id:%d query: %s\n",len1-1,pre);
int ans = T[len1-1].Query(pre);
printf("%d\n",ans);
}
}
return 0;
}