hihoCoder 1261 String Problem II
时间限制:50000ms
单点时限:5000ms
内存限制:512MB
- 例子输入
-
5 5 dsxmlkxp asxglqkdxp asxgavxp asxglp sxglkx kebvpyky hjpntqft asxglkxp polbmzgq jdczlmtd
例子输出
描写叙述
我们有一个字符串集合S，当中有N个两两不同的字符串。还有M个询问，每一个询问都会先给出一个字符串w，你须要回答下面三个问题：
1. 输出全部S的串中。能够由w恰好加入两个字符得到的串中。编号最小的3个。
2. 输出全部S的串中，能够由w改动不超过2个字符得到的串中，编号最小的3个。
3. 输出全部S的串中，能够由w删除恰好两个字符得到的串中，编号最小的3个。
字母能够加入在包含开头结尾在内的任何位置，比方在"abc"中加入"x"和"y"。可能能够得到"yxabc"，"aybxc"，"axbyc"等等的串。
全部字符串仅仅由小写字母构成。
输入
第一行两个数N和M，表示集合S中字符串的数量和询问的数量。
接下来N行。当中第i行给出S中第i个字符串。
第i个字符串的编号就是i。
接下来M行。当中第i行给出第i个询问串。
数据范围：
N,M<=10000。
S中字符串长度和<=100000。
全部询问中字符串长度和<=100000。
输出
对每一个询问输出三行，每行三个数，分别表示每一个问题编号最小的3个串的编号。从小到大排列。
假设不到3个串，则在最后用-1补到3个输出，比方假设结果是1和2，那么输出1 2 -1。假设S中没有满足条件的串，就输出-1 -1 -1。
-1 -1 -1
-1 -1 -1
-1 -1 -1
1 -1 -1
-1 -1 -1
2 -1 -1
1 3 -1
4 5 -1
-1 -1 -1
-1 -1 -1
-1 -1 -1
-1 -1 -1
-1 -1 -1
-1 -1 -1
跟上一题一样，仅仅是上一题的拓展。
代码：
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <string>
#include <vector>
#include <map>
#include <set>
using namespace std;
const int maxn=100000+100;
char s[maxn];
const int Max=27;
struct node
{
int sign;
int next[Max];
} a[maxn];
set<int> ve;
int cur=0;
void insert(char *s,int flg)
{
int len,ans;
int p=0;
len=strlen(s);
for(int i=0; i<len; i++)
{
ans=s[i]-'a';
if(a[p].next[ans]!=0)
{
p=a[p].next[ans];
}
else
{
a[p].next[ans]=++cur;
a[cur].sign=0;
p=a[p].next[ans];
}
}
a[p].sign=flg;
}
int len;
void find0(int sign,int p,int x)
{
if(x==len&&sign==2&&a[p].sign)
{
ve.insert(a[p].sign);
return;
}
if(sign<2&&x<=len)
{
for(int i=0; i<26; i++)
{
if(a[p].next[i]>0)
{
int tp=a[p].next[i];
find0(sign+1,tp,x);
}
}
}
if(x<len)
{
int ne=s[x]-'a';
if(a[p].next[ne]>0)
{
int tp=a[p].next[ne];
find0(sign,tp,x+1);
}
}
}
void find1(int sign,int p,int x)
{
if(x==len&&sign<=2&&a[p].sign)
{
ve.insert(a[p].sign);
return;
}
int ne=s[x]-'a';
if(sign<2&&x<=len)
{
for(int i=0; i<26; i++)
{
if(a[p].next[i]>0&&i!=ne)
{
int tp=a[p].next[i];
find1(sign+1,tp,x+1);
}
}
}
if(x<len)
{
//int ne=s[x]-'a';
if(a[p].next[ne]>0)
{
int tp=a[p].next[ne];
find1(sign,tp,x+1);
}
}
}
void find2(int sign,int p,int x)
{
if(x+(2-sign)==len&&sign<=2&&a[p].sign)
{
ve.insert(a[p].sign);
return;
}
int ne=s[x]-'a';
if(sign<=1)
find2(sign+1,p,x+1);
if(x<len)
{
if(a[p].next[ne]>0)
{
int tp=a[p].next[ne];
find2(sign,tp,x+1);
}
}
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
for(int i=0; i<maxn; i++)
{
cur=0;
a[i].sign=0;
memset(a[i].next,0,sizeof(a[i].next));
}
for(int i=0; i<n; i++)
{
scanf("%s",s);
insert(s,i+1);
}
for(int i=0; i<m; i++)
{
ve.clear();
scanf("%s",s);
len=strlen(s);
find0(0,0,0);
int lo=ve.size();
set<int>::iterator it;
if(lo>0)
it=ve.begin();
for(int j=0;j<3;j++)
{
if(j!=0)
printf(" ");
if(j>=lo)
printf("-1");
else
{
printf("%d",*it);
it++;
}
}
printf("\n");
ve.clear();
find1(0,0,0);
lo=ve.size();
if(lo>0)
it=ve.begin();
for(int j=0;j<3;j++)
{
if(j!=0)
printf(" ");
if(j>=lo)
printf("-1");
else
{
printf("%d",*it);
it++;
}
}
printf("\n");
ve.clear();
find2(0,0,0);
// sort(ve.begin(),ve.end());
lo=ve.size();
if(lo>0)
it=ve.begin();
for(int j=0;j<3;j++)
{
if(j!=0)
printf(" ");
if(j>=lo)
printf("-1");
else
{
printf("%d",*it);
it++;
}
}
printf("\n");
}
}
return 0;
}
