AcWing 158. 项链(字符串最小表示法)

题目描述

有一天，达达捡了一条价值连城的宝石项链，但是，一个严重的问题是，他并不知道项链的主人是谁！
在得知此事后，很多人向达达发来了很多邮件，都说项链是自己的，要求他归还（显然其中最多只有一个人说了真话）。
达达要求每个人都写了一段关于自己项链的描述： 项链上的宝石用数字0至9来标示。
一个对于项链的表示就是从项链的某个宝石开始，顺指针绕一圈，沿途记下经过的宝石，比如项链： 0-1-2-3 ，它的可能的四种表示是0123、1230、2301、3012。
达达现在心急如焚，于是他找到了你，希望你能够编写一个程序，判断两个给定的描述是否代表同一个项链（注意，项链是不会翻转的）。
也就是说给定两个项链的表示，判断他们是否可能是一条项链。
输入格式
输入文件只有两行，每行一个由字符0至9构成的字符串，描述一个项链的表示（保证项链的长度是相等的）。
输出格式
如果两个对项链的描述不可能代表同一个项链，那么输出’No’，否则的话，第一行输出一个’Yes’，第二行输出该项链的字典序最小的表示。
数据范围
设项链的长度为L，1≤L≤1000000
输入样例：
2234342423
2423223434
输出样例：
Yes
2234342423

字符串最小表示法

#include <iostream>
#include <cstring>
using namespace std;

const int N = 2000010;
char a[N], b[N];
int n;

int get_min(char f[])
{
    int i = 0, j = 1, k;
    while(i < n && j < n)
    {
        k = 0;
        while(k < n && f[i + k] == f[j + k]) k++;
        if(k == n) break;
        if(f[i + k] > f[j + k]) 
        {
            i += k + 1;
            if(i == j) i++;
        }
        else 
        {
            j += k + 1;
            if(j == i) j++;
        }
    }
    
    int res = min(i, j);
    
    f[res + n] = 0;
    
    return res;
}

int main()
{
    scanf("%s%s", a, b);
    n = strlen(a);
    memcpy(a + n, a, n);
    memcpy(b + n, b, n);
    
    int ia = get_min(a), ib = get_min(b);
    
    if (strcmp(a + ia, b + ib)) puts("No");
    else
    {
        puts("Yes");
        puts(a + ia);
    }
    
    return 0;
}