程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> HDU 2896 病毒侵襲 (AC自動機)

HDU 2896 病毒侵襲 (AC自動機)

編輯：C++入門知識

HDU 2896 病毒侵襲 (AC自動機)

病毒侵襲
Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 11796    Accepted Submission(s): 3067

Problem Description 當太陽的光輝逐漸被月亮遮蔽，世界失去了光明，大地迎來最黑暗的時刻。。。。在這樣的時刻，人們卻異常興奮——我們能在有生之年看到500年一遇的世界奇觀，那是多麼幸福的事兒啊~~
但網路上總有那麼些網站，開始借著民眾的好奇心，打著介紹日食的旗號，大肆傳播病毒。小t不幸成為受害者之一。小t如此生氣，他決定要把世界上所有帶病毒的網站都找出來。當然，誰都知道這是不可能的。小t卻執意要完成這不能的任務，他說：“子子孫孫無窮匮也！”（愚公後繼有人了）。
萬事開頭難，小t收集了好多病毒的特征碼，又收集了一批詭異網站的源碼，他想知道這些網站中哪些是有病毒的，又是帶了怎樣的病毒呢？順便還想知道他到底收集了多少帶病毒的網站。這時候他卻不知道何從下手了。所以想請大家幫幫忙。小t又是個急性子哦，所以解決問題越快越好哦~~

Input 第一行，一個整數N（1<=N<=500），表示病毒特征碼的個數。
接下來N行，每行表示一個病毒特征碼，特征碼字符串長度在20—200之間。
每個病毒都有一個編號，依此為1—N。
不同編號的病毒特征碼不會相同。
在這之後一行，有一個整數M（1<=M<=1000），表示網站數。
接下來M行，每行表示一個網站源碼，源碼字符串長度在7000—10000之間。
每個網站都有一個編號，依此為1—M。
以上字符串中字符都是ASCII碼可見字符（不包括回車）。

Output 依次按如下格式輸出按網站編號從小到大輸出，帶病毒的網站編號和包含病毒編號，每行一個含毒網站信息。
web 網站編號: 病毒編號病毒編號 …
冒號後有一個空格，病毒編號按從小到大排列，兩個病毒編號之間用一個空格隔開，如果一個網站包含病毒，病毒數不會超過3個。
最後一行輸出統計信息，如下格式
total: 帶病毒網站數
冒號後有一個空格。

Sample Input

3
aaa
bbb
ccc
2
aaabbbccc
bbaacc

Sample Output

web 1: 1 2 3
total: 1

Source 2009 Multi-University Training Contest 10 - Host by NIT

題意：

自己看題。

分析：

簡單粗暴的AC自動機模板題。

/*
 *
 * Author : fcbruce
 *
 * Time : Sat 04 Oct 2014 07:39:54 PM CST
 *
 */
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include
#include 
#define sqr(x) ((x)*(x))
#define LL long long
#define itn int
#define INF 0x3f3f3f3f
#define PI 3.1415926535897932384626
#define eps 1e-10

#ifdef _WIN32
  #define lld %I64d
#else
  #define lld %lld
#endif

#define maxm 128
#define maxn 100007

using namespace std;

bool found[507];
char T[10007];
char P[233];
int q[maxn<<2];

struct ACauto
{
  int ch[maxn][maxm];
  int val[maxn];
  int nex[maxn];
  int last[maxn];
  int cnt;
  int sz;

  ACauto()
  {
    sz=1;
    val[0]=0;
    memset(ch[0],0,sizeof ch[0]);
  }

  void clear()
  {
    sz=1;
    val[0]=0;
    memset(ch[0],0,sizeof ch[0]);
  }

  int idx(const char ch)
  {
    return ch-'';
  }

  void insert(const char *s,int v=1)
  {
    int u=0;
    for (int i=0;s[i]!='';i++)
    {
      int c=idx(s[i]);
      if (ch[u][c]==0)
      {
        val[sz]=0;
        memset(ch[sz],0,sizeof ch[sz]);
        ch[u][c]=sz++;
      }
      u=ch[u][c];
    }
    val[u]=v;
  }

  void get_fail()
  {
    int f=0,r=-1;
    nex[0]=0;
    last[0]=0;
    for (int c=0;c0 && ch[j][c]==0) j=nex[j];
        nex[u]=ch[j][c];
        last[u]=val[nex[u]]>0?nex[u]:last[nex[u]];
      }
    }
  }

  void calc(int j)
  {
    if (j!=0)
    {
      cnt++;
      found[val[j]]=true;
      calc(last[j]);
    }
  }

  void find(const char *T)
  {
    cnt=0;
    memset(found,0,sizeof found);

    for (int i=0,j=0;T[i]!='';i++)
    {
      int c=idx(T[i]);
      while (j>0 && ch[j][c]==0) j=nex[j];
      j=ch[j][c];
      if (val[j]!=0)
        calc(j);
      else if (last[j]!=0)
        calc(last[j]);
    }
  }
}acauto;

int main()
{
#ifdef FCBRUCE
  freopen(/home/fcbruce/code/t,r,stdin);
#endif // FCBRUCE

  int n;

  while (scanf(%d,&n)==1)
  {
    acauto.clear();

    for (int i=1;i<=n;i++)
    {
      scanf(%s,P);
      acauto.insert(P,i);
    }

    acauto.get_fail();

    int m;
    scanf(%d,&m);

    int cnt=0;

    for (int i=1;i<=m;i++)
    {
      scanf(%s,T);
      acauto.find(T);

      if (acauto.cnt>0)
      {
        cnt++;
        printf(web %d:,i);
        for (int j=1;j<=n;j++)
          if (found[j])
            printf( %d,j);
        putchar('
');
      }
    }

    printf(total: %d
,cnt);
  }

  return 0;
}