hdu 5853 Jong Hyok and String

说出来你可能不信,我把数组开成了char,结果检查了一辈子没检查出来。

这题乍一看,是后缀自动机。但是后缀自动机都忘完了。

于是就想了想,能不能用后缀数组。

首先,分析完题目。其实题目让求的就是和询问串的后缀相同的,而且作为P串的子串的种数相同的有多少个。

于是。

我们可以把P串都接到一块,用不同的字符隔开。

每次来一个串,我们可以通过二分,确定公共子串是当前串的所有后缀串前缀(把所有串翻转一下,就变成了找后缀的前缀,就是子串)有哪些。然后显然能使用的最长长度就是这些串的最长公共长度。而最短长度就是可以使得个数变多的所求串的前缀的长度长度+1。于是减一下就可以了。

注意,当串只有一个的时候需要特殊处理一下。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>

using namespace std;

const int maxn = 200030;

char ch[maxn], ch2[maxn];
int bg[maxn], id[maxn];

struct SuffixArray {
    int r[maxn];
    int sa[maxn], rank[maxn], height[maxn];
    int t[maxn], t2[maxn], c[maxn], n;
    int m;///模板长度
    void init(char* s) {
        n = strlen(s);
        int cnt = 'z' + 1;
        for (int i = 0; i < n; i++) {
            r[i] = int(s[i]);
            if(s[i] == '#') r[i] = cnt ++;
        }
        m = cnt;
    }
    int cmp(int *r, int a, int b, int l) {
        return r[a] == r[b] && r[a + l] == r[b + l];
    }
    /**
    字符要先转化为正整数
    待排序的字符串放在r[]数组中,从r[0]到r[n-1],长度为n,且最大值小于m。
    所有的r[i]都大于0,r[n]无意义算法中置0
    函数结束后,结果放在sa[]数组中(名次从1..n),从sa[1]到sa[n]。s[0]无意义
    **/
    void build_sa() {
        int i, k, p, *x = t, *y = t2;
        r[n++] = 0;
        for (i = 0; i < m; i++) c[i] = 0;
        for (i = 0; i < n; i++) c[x[i] = r[i]]++;
        for (i = 1; i < m; i++) c[i] += c[i - 1];
        for (i = n - 1; i >= 0; i--) sa[--c[x[i]]] = i;
        for (k = 1, p = 1; k < n; k *= 2, m = p) {
            for (p = 0, i = n - k; i < n; i++) y[p++] = i;
            for (i = 0; i < n; i++) if (sa[i] >= k) y[p++] = sa[i] - k;
            for (i = 0; i < m; i++) c[i] = 0;
            for (i = 0; i < n; i++) c[x[y[i]]]++;
            for (i = 1; i < m; i++) c[i] += c[i - 1];
            for (i = n - 1; i >= 0; i--) sa[--c[x[y[i]]]] = y[i];
            swap(x, y);
            p = 1;
            x[sa[0]] = 0;
            for (i = 1; i < n; i++) x[sa[i]] = cmp(y, sa[i - 1], sa[i], k) ? p - 1 : p++;
        }
        n--;
    }
    /**
    height[2..n]:height[i]保存的是lcp(sa[i],sa[i-1])
    rank[0..n-1]:rank[i]保存的是原串中suffix[i]的名次
    **/
    void getHeight() {
        int i, j, k = 0;
        for (i = 1; i <= n; i++) rank[sa[i]] = i;
        for (i = 0; i < n; i++) {
            if (k) k--;
            j = sa[rank[i] - 1];
            while (r[i + k] == r[j + k]) k++;
            height[rank[i]] = k;
        }
    }
    int d[maxn][20];
    ///元素从1编号到n
    void RMQ_init(int A[], int n) {
        for (int i = 1; i <= n; i++) d[i][0] = A[i];
        for (int j = 1; (1 << j) <= n; j++)
            for (int i = 1; i + (1 << (j - 1)) <= n; i++)
                d[i][j] = min(d[i][j - 1], d[i + (1 << (j - 1))][j - 1]);
    }
    int RMQ(int L, int R) {
        int k = 0;
        if(L > R) swap(L, R);
        if(L == R) return n - sa[L];
        L++;
        while ((1 << (k + 1)) <= R - L + 1) k++;
        return min(d[L][k], d[R - (1 << k) + 1][k]);
    }
    void call_fun(char* s) {
        init(s);
        build_sa();
        getHeight();
        RMQ_init(height, n);
    }

    int gao(char* ch2) {
       int len = strlen(ch2);
       int L = 1, R = n, cnt = n + 1, ret = 0;
       for(int i = len - 1, j = 0; i >= 0; i --, j ++) {
            int L1 = L, R1 = R;
            while(L1 <= R1) {
                int mid = (L1 + R1) >> 1;
                if(n - sa[mid] - j <= 0 || r[sa[mid] + j] < ch2[i]) L1 = mid + 1;
                else R1 = mid - 1;
            }
            L = L1;
            L1 = L, R1 = R;
            while(L1 <= R1) {
                int mid = (L1 + R1) >> 1;
                if(r[sa[mid] + j] == ch2[i]) L1 = mid + 1;
                else R1 = mid - 1;
            }
            R = R1;
            if(L > R) return 0;
            if(R - L < cnt) ret = 1, cnt = R - L;
            else if(R - L == cnt) ret ++;
       }
       if(L == R) {
            return ret + bg[id[sa[L]]] - sa[L] - len;
       }
       return RMQ(L, R) - len + ret;
    }
} sol;

int main() {
    int T, n, m, cas = 1;
    scanf("%d", &T);
    while(T --) {
        scanf("%d%d", &n, &m);
        int l = 0;
        for(int i = 0; i < n; i ++) {
            scanf("%s", ch2);
            int l2 = strlen(ch2);
            for(int j = 0; j < l2; j ++) {
                id[l] = i;
                ch[l ++] = ch2[l2 - j - 1];
            }
            bg[i] = l;
            id[l] = i;
            ch[l ++] = '#';
        }
        ch[l] = 0;
        sol.call_fun(ch);
        printf("Case #%d:\n", cas ++);
        for(int i = 0; i < m; i ++) {
            scanf("%s", ch2);
            printf("%d\n", sol.gao(ch2));
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值