HDU 3336

 思路:KMP,但要对其进行变形,当找到失败位置时,要继续考察该位置,一直向前找到字符串首不能再向前找,因为我们不只要计算该子串本身,我们还要计算该子串包含的其他子串,因为这些子串都是原串的子串,这是显然的。


#include<stdio.h>
#include<string.h>
int fail[200005];
int sum[200005]; 
char str[200005]; 
int T, m, max;  
void getfail()
{
    fail[0] = -1;
    int i, j, temp; 
    for(i = 1, j = -1; i < m; i ++)
    {
        while(j >= 0 && str[j + 1] != str[i])
        {
            j = fail[j]; 
        }
        if(str[j + 1] == str[i])
            j ++; 
        fail[i] = j;
        if(j >= 0)
        {
            sum[j] ++; 
            if(max < j)
                max = j; 
            temp = j; 
            while(temp >= 0)
            {
                temp = fail[temp]; 
                sum[temp] ++; 
            }
        }
    }
}
    int main(int argc, char const *argv[]) 
    {
        int i, cnt; 
        scanf("%d", &T); 
        while(T--)
        {
            memset(str, 0, sizeof(str)); 
            memset(sum, 0, sizeof(sum)); 
            scanf("%d", &m); 
            cnt = m % 10007; 
            max = 0; 
            getchar(); 
            fgets(str, m+1,stdin);  
            getfail(); 
            for(i = 0; i <= max; i ++)
            {
                /* printf("OK\n"); */
                sum[i] %= 10007; 
                cnt += sum[i]; 
                cnt %= 10007; 
            }
            printf("%d\n", cnt);
        }
        return 0; 
    }


posted on 2014-04-30 18:36  wangzhili  阅读(88)  评论(0)    收藏  举报