ac自动机模板

const int Maxn = 1e6;
const int Maxkind = 26;

struct Ac_Automata {
    int cnt; 
    int End[Maxn + 5];
    int Tire[Maxkind + 5][Maxn + 5];
    int dout[Maxn + 5];
    void Init () {
    	memset (dout, 0, sizeof dout);
    	memset (Tire, 0, sizeof Tire);
    	memset (End, 0, sizeof End);
	}
    void Insert (char *str) {
        int len = strlen (str + 1);
        int p = 0;
        rep (i, 1, len) {
            int ch = str[i] - 'a' + 1;
            if (Tire[ch][p] == 0) Tire[ch][p] = ++cnt;
            p = Tire[ch][p];
        }
        End[p]++;
    }
    void Get_Next () {
        queue <int> q;
        rep (i, 1, Maxkind) if (Tire[i][0]) q.push (Tire[i][0]);
        while (q.size ()) {
            int u = q.front (); q.pop ();
            rep (ch, 1, Maxkind) {
                int v = Tire[ch][u];
                if (v != 0) {
                    Tire[0][v] = Tire[ch][Tire[0][u]];
                    dout[Tire[ch][Tire[0][u]]]++;
                    q.push (v);
                }
                else {
                	Tire[ch][u] = Tire[ch][Tire[0][u]];
				}
            }
        }
    }
    bool vis[Maxn + 5];
    int Query (char *str) {
        int len = strlen (str + 1);
        int p = 0;
        rep (i, 1, len) {
            int ch = str[i] - 'a' + 1;
            p = Tire[ch][p];
            vis[p] = 1;
        }
        
        int res = 0;
        queue <int> q;
        rep (i, 1, cnt)
        	if (dout[i] == 0)
        		q.push (i);
        while (q.size ()) {
        	int u = q.front (); q.pop ();
        	res += vis[u] * End[u];
        	vis[Tire[0][u]] |= vis[u];
        	dout[Tire[0][u]]--;
        	if (dout[Tire[0][u]] == 0) {
        		q.push (Tire[0][u]);
			}
		}
        return res;
    }
}Tree;
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值