小红的环形字符串

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
?

小红拿到了一个环形字符串sss。所谓环形字符串，指首尾相接的字符串。
小红想顺时针截取其中一段连续子串正好等于ttt，一共有多少种截法？

输入描述:

第一行输入字符串 s。
第二行输入字符串 t。
1≤len(t)≤len(s)≤10001\leq len(t) \leq len(s) \leq 10001≤len(t)≤len(s)≤1000

输出描述:

环形字符串 s 截取一段连续子串等于字符串 t 的方案数。

示例1

输入

ababab
aba

输出

3

#include <stdio.h>
#include <string.h>
int main()
{
    char s[3000] = {0};
    char list[1001] = {0};
    int arr[3000] = {0};
    scanf("%s",s);
    int len = strlen(s);
    getchar();
    scanf("%s",list);
    strncat(s,s,len);
    int count = 0;
    int p1,p2;
    p1 = p2 = 0;
    while (1)
    {
        if (p1 >= len)
        {
            if (arr[p1 - len] != 0) break;
        }
        else
            if (arr[p1] != 0)  break;
        if (s[p1] == list[p2])
        {
            int temp = p1;
            temp++;
            p2++;
            while (list[p2])
            {
                if (s[temp] == list[p2])
                {
                    temp++;
                    p2++;
                }
                else
                    break;
            }
            if (list[p2] != '\0')
            {
                p1++;
                p2 = 0;
            }
            else
            {
                arr[p1]++;
                count++;
                p1++;
                p2 = 0;
            }
        }
        else
            p1++;
    }
    printf("%d",count);
    return 0;
}