POJ 2778 ac自动机+矩阵;
先给两个链接:POJ 2778http://poj.org/problem?id=2778 matrix67 的矩阵十题:http://www.matrix67.com/blog/archives/276/
矩阵这是一种很神奇的算法,神奇的矩阵的应用。很早以前看过MATRIX67的矩阵十道时 无比ORZ。现在又看到了这里,感触良多啊。
此题的关键在于如何建模,即将一个字符串匹配的种数问题 变成一个矩阵连乘问题。
主要算法是将每一个TRIE【】节点做为一种状态,如果TRIE【I】.NEXT【J】有能走。就在矩阵中将MAP[I][J]变为1;然后对矩阵进行N次幂就得到可以走的结果了。如:最开始在根节点,可以走MAP 【0】【0】;MAP【 0】【2】 ;MAP【0】【5】;MAP【 0】【7】这四个点,分别到达0 2 5 7 四个点;
然后从 A开始走可以到达 MAP 【2】【0】;MAP 【2】【7】;即T,G两个点;因为A 不能到A与T;
……
这样矩阵就建好了。有了AC自动机的基础知识和矩阵的快速幂这题就能解了。
最终结果等于所得矩阵的第一行相加。即从根节点出发,走N次,所能到达的各个节点的次数总和。
这个题的关键思路在于下面这两个图: 本人表示没看懂MATRIX的第十题说明,因此用图将其正确用意表示出来。
1:如果某一个点没有出现就回到0这个节点。
2:之所以建立一个没有用的TRIE[1]是因为当输入的N为0时,需要一个过渡的节点使其能正常地匹配与进行幂运算,结果得到4^N
//对于二分我还不太会运用,需要不断地练习练习。
//HDOJ2243:此题的加强版: http://acm.hdu.edu.cn/showproblem.php?pid=2243要用两次二分。
POJ 2778
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
#define maxn 2000000000
#define mod 100000
typedef long long LL;
int n, m, state;
struct node {
int next[4];
int fail;
int flag;
void init() {
memset(next, -1, sizeof (next));
flag = 0;
fail = -1;
}
} trie[120];
struct matrix {
LL v[120][120];
}g;
int get_num(char c) {
if (c == 'A')return 0;
if (c == 'C')return 1;
if (c == 'G')return 2;
return 3;
}
void inline insert( char* s )
{
int root= 0;
while( *s )
{
if(trie[root].flag)break;
int t= get_num(*s);
if( trie[root].next[t]==-1 )
{
trie[++state].init();
trie[root].next[t]=state;//
}
root= trie[root].next[t];
s++;
}
trie[root].flag++;
}
queue<int>q;
void bfs_ac() {
q.push(0);
while (!q.empty()) {
int p = q.front();q.pop();
for (int i = 0; i < 4; i++)if (trie[p].next[i] != -1) {
if (p == 0)trie[trie[p].next[i]].fail = 0;
else {
int fail = trie[p].fail;
while (fail != -1 &&trie[fail].next[i]==-1) {
fail = trie[fail].fail;
}
if (fail == -1) {
trie[trie[p].next[i]].fail = 0;
}
else
{
trie[trie[p].next[i]].fail=trie[fail].next[i];
if(trie[trie[fail].next[i]].flag)trie[trie[p].next[i]].flag++;
}
}
q.push(trie[p].next[i]);
}
}
}
//*
//*/
matrix mul(matrix trie, matrix b) {
matrix p;
memset(p.v, 0, sizeof (p.v));
for (int i = 0; i < state; i++)
for (int j = 0; j < state; j++)
for (int k = 0; k < state; k++)
p.v[i][j] = (p.v[i][j] + trie.v[i][k] * b.v[k][j]) % mod;
return p;
}
matrix fen(matrix m,int n) {
matrix p;
memset(p.v, 0, sizeof (p.v));
for (int i = 0; i < state; i++)p.v[i][i] = 1;
while (n) {
if (n & 1)
p = mul(p, m);
n >>= 1;
m = mul(m, m);
}
return p;
}
int main() {
int i, j;
char s[30];
while(scanf("%d%d", &m, &n)!=EOF){
trie[0].init();
state = 1;
for (i = 0; i < m; i++) {
scanf("%s", s);
insert(s);
}
bfs_ac();
memset(g.v, 0, sizeof (g.v)); //state--;
//cout<<"FAIL: ";for(int i=0;i<=state;i++) cout<<trie[i].fail<<" ";cout<<endl;
//cout<<"flag: ";for(int i=0;i<=state;i++) cout<<trie[i].flag<<" ";cout<<endl;
for (i = 0; i < state; i++)
if(!trie[i].flag)
{//cout<<"i= "<<i<<endl;
for (j = 0; j < 4; j++)
{//cout<<"j="<<j<<endl;
if (trie[i].next[j] != -1 && !trie[trie[i].next[j]].flag&&trie[i].next[j]!=0&&trie[i].next[j]!=0)
{
// cout<<"a: "<<i<<" "<<trie[i].next[j]<<endl;
g.v[i][trie[i].next[j]]++;}
else if (trie[i].next[j] == -1)
{
if (i == 0) g.v[0][0]++;
else
{
int fail =i;
while (fail!=-1&& trie[fail].next[j] == -1)
{
fail = trie[fail].fail;
}
if (fail==-1)
{
g.v[i][0]++;
//cout<<"b: "<<i<<" "<<"0"<<endl;
}
else if(!trie[trie[fail].next[j]].flag )
{
g.v[i][trie[fail].next[j]]++;
//cout<<"c: "<<i<<" "<<trie[fail].next[j]<<" "<<fail<<endl;
}
}
}
}
}
//for(int i=0;i<=state;i++)cout<<g.v[i][1]<<" "<<g.v[1][i]<<" ";cout<<endl;
//for(int i=0;i<=state;i++){ for(int j=0;j<=state;j++) cout<<g.v[i][j]<<" ";cout<<endl;}cout<<endl;
matrix c = fen(g,n);
//for(int i=0;i<=state;i++){ for(int j=0;j<=state;j++) cout<<c.v[i][j]<<" ";cout<<endl;}
LL ans=0;
for (int i = 0; i < state; i++)if (!trie[i].flag)
{
ans = ans + c.v[0][i];
if(ans>=mod)ans-=mod;
}
printf("%lld\n", ans);}
//system("pause");
}
zealot
2011/7/11
zealot是一种精神,那种为了信仰,勇往直前,永远冲锋的精神,也许别人会嘲笑zealot那近乎傻的执着,但zealot的力量却是谁也不敢小看的,他们的力量来源于对高贵的Protoss的信仰,来源于每一天沉默的苦炼中。