最小表示法 + KMP - String Problem - HDU - 3374

最小表示法 + KMP - String Problem - HDU - 3374

题意:

$$\begin{gathered} 多组测试用例，每组包括一个字符串。 \\ 要求这个字符串的最小表示和最大表示第一次出现的位置以及出现的次数。 \end{gathered}$$

Sample Input:
abcder
aaaaaa
ababab
Sample Output:
1 1 6 1
1 6 1 6
1 3 2 3

数据范围：

$$\begin{gathered} 字符串长度n<=10^6。 \\ Timelimit：1000ms，Memorylimit：32768kB \end{gathered}$$

题解:

$$\begin{gathered} 难点在于求最大/最小表示出现的次数，原串复制一遍再将最大/最小表示逐一匹配似乎会TLE。 \\ 事实上，因为整个字符串被视为一个环，我们可以把任何位置的字符视作开头或结尾， \\ 因此，任何一个位置开始的字符串出现的次数都是相等的。 \\ 那么我们可以将问题转化为求任意位置开始的字符串循环节的个数，每个循环节对应一个相同字符串的起点。 \end{gathered}$$

---

代码：

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

using namespace std;

const int N=1e6+10;

char str[N];
int ne[N],n,m,ans1,ans2,cnt;

void get_next(char *str)
{
    int i=0,j=-1,n=strlen(str);
    ne[i]=j;
    while(i<n){
        if(j==-1 || str[i]==str[j]) ne[++i]=++j;
        else j=ne[j];
    }
}

int get_M(int flag)
{
    int p1=0,p2=1,k=0;
    while(p1<n&&p2<n&&k<n)
    {
        int tt=(str[(p1+k)%n]-str[(p2+k)%n]);
        if(tt==0) k++;
        else
        {
            if(!flag)
            {
                if(tt<0) p2+=k+1;
                else p1+=k+1;
            }
            else
            {
                if(tt<0) p1+=k+1;
                else p2+=k+1;
            }
            if(p1==p2) p2++;
            k=0;
        }
    }
    return min(p1,p2);
}

int main()
{
    while(~scanf("%s",str))
    {
        ans1=ans2=cnt=0;
        n=strlen(str);
        get_next(str);
        ans1=get_M(0)+1,ans2=get_M(1)+1;
        if(n%(n-ne[n])==0) cnt=n/(n-ne[n]);
        else cnt=1;
        printf("%d %d %d %d\n",ans1,cnt,ans2,cnt);
    }
    return 0;
}