hihoCoder 1261 String Problem II

时间限制:50000ms

单点时限:5000ms

内存限制:512MB

描写叙述

我们有一个字符串集合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。

例子输入

5 5
dsxmlkxp
asxglqkdxp
asxgavxp
asxglp
sxglkx
kebvpyky
hjpntqft
asxglkxp
polbmzgq
jdczlmtd

例子输出

-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;
}