HDU 3901 Wildcard

题目：Wildcard

链接：http://acm.hdu.edu.cn/showproblem.php?pid=3901

题意：给一个原串（只含小写字母）和一个模式串（含小写字母、？、* ，*号可替换为0到无穷个任意字母，？可替换为一个任意字母），问两个字符串是否匹配。

思路：

　　这是经典题吧。。。

　　AC自动机。（网上大多代码是错的，用普通kmp是不行的，不知道修改过后可不可以）。

　　HDU已经加强数据了，可我前一个代码abcd ？显示是YES居然还过了，看来数据还是有点弱啊。

　　*号的处理是很容易的，只需要把模式串以*号为分界分解开来，然后除了最后一段，其他尽量匹配靠前的原串子串就差不多了，比如ab*cd*ef*gh，那么原串必须以ab开头、gh结尾、然后找到最前的cd、再找到ef，如果互相不重叠就OK。

　　比较难的是？的处理了，*分解开模式串分别处理。现在，根据？分解，将分解后的字符串（不含？号了）构建AC自动机。

　　比如串A：abcd?abc?cd?abcd，构建AC自动机，然后在结束点记录他在串A中的位置。

　　红色为根，黄色为每一个结束点，这里的fail指针与普通AC自动机相同

　　现在我们创建一个数组cnt，对应原串的每一位，让原串在自动机上跑，如果遇到黄点cnt[当前位置-记录的值+1]++，比如遇到abcd，那说明原串当前位置前四位一定是abcd，满足这一段，而如果有某一个cnt值等于4，就说明他满足了所有段，那么他可以作为串A匹配的起点，a b c d _ a b c _ c d _ a b c d，只有上面这种情况才会使第一个a的cnt值为4，而cnt等于4，当前位就是该次匹配满足的最前位置了。

　　上一段理解后，题目就可以得解了，注意细节就是了。

AC代码：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<set>
#include<map>
#include<list>
#include<stack>
#include<queue>
#include<vector>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
#define lson rt<<1
#define rson rt<<1|1
#define N 100010
#define M 26
#define Mod 1000000007
#define LL long long
#define INF 0x7fffffff
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;i++)
#define For(i,f_start,f_end) for(int i=f_start;i<f_end;i++)
#define REP(i,f_end,f_start) for(int i=f_end;i>=f_start;i--)
#define Rep(i,f_end,f_start) for(int i=f_end;i>f_start;i--)
#define MT(x,i) memset(x,i,sizeof(x))
#define gcd(x,y) __gcd(x,y)
const double PI = acos(-1);

struct Node
{
  vector<int> d;
  int fail;
  int next[M];
}v[N];
int vNum;

void Init()
{
  v[0].fail = -1;
  FOR(i,0,25){
    v[0].next[i]=-1;
  }
  vNum=1;
}

int create()
{
  v[vNum].d.clear();
  v[vNum].fail=-1;
  for(int i=0;i<M;i++) v[vNum].next[i]=-1;
  return vNum++;
}

int getPos(char c)
{
  return c-'a';
}

int add(char *s)
{
  int i,ret=0,p=0;
  for(i=0;s[i];i++){
    if(s[i]=='?'){
      if(p!=0){
        ret++;
        v[p].d.push_back(i);
        p=0;
      }
    }
    int pos = getPos(s[i]);
    if(v[p].next[pos]==-1) v[p].next[pos]=create();
    p = v[p].next[pos];
  }
  if(p!=0){
    ret++;
    v[p].d.push_back(i);
  }
  return ret;
}

queue<int> q;
void makeFail()
{
  q.push(0);
  while(q.size())
  {
    int p=q.front();
    q.pop();
    for(int i=0;i<M;i++)
    {
      if(v[p].next[i]!=-1)
      {
        int t=v[p].fail;
        while(1)
        {
          if(t == -1)  // p 已经是根了
          {
            v[v[p].next[i] ].fail = 0;
            break;
          }
          else if(v[t].next[i] != -1)
          {
            v[v[p].next[i] ].fail = v[t].next[i];
            break;
          }
          else t=v[t].fail;
        }
        if(t!=-1){
          for(int j=0;j<v[t].d.size();j++){
            v[p].d.push_back(v[t].d[j]);
          }
        }
        q.push(v[p].next[i]);
      }
    }
  }
}

int cnt[N];
int match(char *s,char *s1)
{
  Init();
  int duan = add(s1);
  if(duan==0) return 0;

  makeFail();

  //printf("%s %s %d
",s,s1,duan);

  memset(cnt,0,sizeof(cnt));
  int i,p=0;
  for(i=0;s[i];i++){
    int pos = getPos(s[i]);
    if(p==-1){
      p=0;
      continue;
    }
    if(v[p].next[pos]==-1){
      p = v[p].fail;
      i--;
      continue;
    }
    p = v[p].next[pos];
    for(int j=0;j<v[p].d.size();j++){
      cnt[i-v[p].d[j]+1]++;
      if(cnt[i-v[p].d[j]+1]==duan) return i-v[p].d[j]+1;
    }
  }
  return -1;
}

char s[N],s1[N],s2[N];
bool solve(char *s,char *s1)
{
  int ls=strlen(s),ps=0;
  int i,len=0,flag=0;
  for(i=0;s1[i];i++){
    if(s1[i]=='*'){
      if(len!=0){
        s2[len]=0;
        int qi = match(s+ps,s2);
        if(qi==-1) return false;
        if(flag==0 && qi!=0) return false;
        ps += qi + len;
        if(ps>ls) return false;
        len = 0;
      }
      flag=1;
    }
    else s2[len++]=s1[i];
  }
  if(len!=0){
    len--;
    for(i=ls-1;i>=ps && len>=0;i--){
      if(s2[len]!='?' && s2[len]!=s[i]) return false;
      len--;
    }
    if(i>=ps && flag==0 || len>=0) return false;
  }
  return true;
}

int main()
{
  while(scanf("%s%s",s,s1)!=EOF)
  {
    printf("%s
",solve(s,s1)?"YES":"NO");
  }
  return 0;
}