POJ1204 Word Puzzles（AC自动机）
给一个L*C字符矩阵和W个字符串，问那些字符串出现在矩阵的位置，横竖斜八个向。
就是个多模式匹配的问题，直接AC自动机搞了，枚举字符矩阵八个方向的所有字符串构成主串，然后在W个模式串构造的AC自动机上跑。
另外，temp指针的那个找遗漏后缀的过程执行时标记一下，下一次再到这个结点就不需要再进行一次temp的过程，这样的时间复杂度就是O(W个模式串总长+LC)。
一开始还想8个方向分别计算坐标= =写第二个方向懒得写了，然后就忽然想到可以一开始构造主串时就存坐标。。最后代码很是挺长的。。
1 #include<cstdio> 2 #include<cstring> 3 #include<queue> 4 using namespace std; 5 #define MAXN 111111 6 int tn,ch[MAXN][26],flag[MAXN],fail[MAXN],len[MAXN]; 7 void insert(char *s,int k,int l){ 8 int x=0; 9 for(int i=0; s[i]; ++i){ 10 int y=s[i]-'A'; 11 if(ch[x][y]==0) ch[x][y]=++tn; 12 x=ch[x][y]; 13 } 14 flag[x]=k; 15 len[x]=l; 16 } 17 void getFail(){ 18 memset(fail,0,sizeof(fail)); 19 queue<int> que; 20 for(int i=0; i<26; ++i){ 21 if(ch[0][i]) que.push(ch[0][i]); 22 } 23 while(!que.empty()){ 24 int x=que.front(); que.pop(); 25 for(int i=0; i<26; ++i){ 26 if(ch[x][i]) que.push(ch[x][i]),fail[ch[x][i]]=ch[fail[x]][i]; 27 else ch[x][i]=ch[fail[x]][i]; 28 } 29 } 30 } 31 32 int n,m; 33 int ansx[1111],ansy[1111],anso[1111]; 34 char str[1111]; int sx[1111],sy[1111]; 35 void match(char o){ 36 int x=0; 37 for(int i=0; str[i]; ++i){ 38 int y=str[i]-'A'; 39 x=ch[x][y]; 40 int tmp=x; 41 while(flag[tmp] && tmp){ 42 if(flag[tmp]>0){ 43 int &idx=flag[tmp]; 44 ansx[idx]=sx[i-len[tmp]+1]; 45 ansy[idx]=sy[i-len[tmp]+1]; 46 anso[idx]=o; 47 idx=-1; 48 }else if(flag[tmp]==-1) break; 49 tmp=fail[tmp]; 50 } 51 } 52 } 53 char map[1111][11111]; 54 int main(){ 55 int w; 56 scanf("%d%d%d",&n,&m,&w); 57 for(int i=0; i<n; ++i) scanf("%s",map[i]); 58 for(int i=1; i<=w; ++i){ 59 scanf("%s",str); 60 insert(str,i,strlen(str)); 61 } 62 getFail(); 63 int sn; 64 //zero 65 for(int i=0; i<m; ++i){ 66 for(int j=0; j<n; ++j){ 67 sx[n-j-1]=j; sy[n-j-1]=i; 68 str[n-j-1]=map[j][i]; 69 } 70 str[n]=0; 71 match('A'); 72 } 73 //one 74 for(int i=0; i<n; ++i){ 75 int x=i,y=0,sn=0; 76 while(x>=0 && y<m){ 77 sx[sn]=x; sy[sn]=y; 78 str[sn++]=map[x][y]; 79 --x; ++y; 80 } 81 str[sn]=0; 82 match('B'); 83 } 84 for(int i=1; i<m; ++i){ 85 int x=n-1,y=i,sn=0; 86 while(x>=0 && y<m){ 87 sx[sn]=x; sy[sn]=y; 88 str[sn++]=map[x][y]; 89 --x; ++y; 90 } 91 str[sn]=0; 92 match('B'); 93 } 94 //two 95 for(int i=0; i<n; ++i){ 96 for(int j=0; j<m; ++j){ 97 sx[j]=i; sy[j]=j; 98 str[j]=map[i][j]; 99 } 100 str[m]=0; 101 match('C'); 102 } 103 //three 104 for(int i=0; i<m; ++i){ 105 int x=0,y=i,sn=0; 106 while(x<n && y<m){ 107 sx[sn]=x; sy[sn]=y; 108 str[sn++]=map[x][y]; 109 ++x; ++y; 110 } 111 str[sn]=0; 112 match('D'); 113 } 114 for(int i=1; i<n; ++i){ 115 int x=i,y=0,sn=0; 116 while(x<n && y<m){ 117 sx[sn]=x; sy[sn]=y; 118 str[sn++]=map[x][y]; 119 ++x; ++y; 120 } 121 str[sn]=0; 122 match('D'); 123 } 124 //four 125 for(int i=0; i<m; ++i){ 126 for(int j=0; j<n; ++j){ 127 sx[j]=j; sy[j]=i; 128 str[j]=map[j][i]; 129 } 130 str[n]=0; 131 match('E'); 132 } 133 //five 134 for(int i=0; i<m; ++i){ 135 int x=0,y=i,sn=0; 136 while(x<n && y>=0){ 137 sx[sn]=x; sy[sn]=y; 138 str[sn++]=map[x][y]; 139 ++x; --y; 140 } 141 str[sn]=0; 142 match('F'); 143 } 144 for(int i=1; i<n; ++i){ 145 int x=i,y=m-1,sn=0; 146 while(x<n && y>=0){ 147 sx[sn]=x; sy[sn]=y; 148 str[sn++]=map[x][y]; 149 ++x; --y; 150 } 151 str[sn]=0; 152 match('F'); 153 } 154 //six 155 for(int i=0; i<n; ++i){ 156 for(int j=0; j<m; ++j){ 157 sx[m-j-1]=i; sy[m-j-1]=j; 158 str[m-j-1]=map[i][j]; 159 } 160 str[m]=0; 161 match('G'); 162 } 163 //seven 164 for(int i=0; i<m; ++i){ 165 int x=n-1,y=i,sn=0; 166 while(x>=0 && y>=0){ 167 sx[sn]=x; sy[sn]=y; 168 str[sn++]=map[x][y]; 169 --x; --y; 170 } 171 str[sn]=0; 172 match('H'); 173 } 174 for(int i=0; i<n-1; ++i){ 175 int x=i,y=m-1,sn=0; 176 while(x>=0 && y>=0){ 177 sx[sn]=x; sy[sn]=y; 178 str[sn++]=map[x][y]; 179 --x; --y; 180 } 181 str[sn]=0; 182 match('H'); 183 } 184 for(int i=1; i<=w; ++i){ 185 printf("%d %d %c\n",ansx[i],ansy[i],anso[i]); 186 } 187 return 0; 188 }
