bzoj1398 Necklace

关于最小表示法的模板题。

最小表示法：把一个字符串表示为它的的所有循环同构字符串中的字典序最小者。

直接参见代码中的函数getmin()获取精髓

#include <cstdio>
#include <string>
#include <iostream>

using std::string;
using std::cin;
using std::cout;
using std::endl;

inline string getmin(string x) {
    int n = x.size();
    string s = x + x;
    int i = 0, j = 1;
    while(i <= n && j <= n) {
        int k = 0;
        while(s[i + k] == s[j + k] && k < n) {
            k++;
        }
        if(k == n + 1) {
            return x;
        }
        if(s[i + k] < s[j + k]) {
            j += k + 1;
            if(j == i) {
                j++;
            }
        }
        else if(s[i + k] > s[j + k]) {
            i += k + 1;
            if(i == j) {
                i++;
            }
        }
        else {
            printf("ERROR!\n");
        }
    }
    if(j > i) j = i;
    string ans = "";
    for(i = 0; i < n; i++) {
        ans += s[j + i];
    }
    //cout << ans << endl;
    return ans;
}

int main() {
    string a, b;
    cin >> a >> b;
    //cout << a << endl << b << endl;
    a = getmin(a);
    b = getmin(b);
    //cout << a << endl << b << endl;
    if(a == b) {
        printf("Yes\n");
        cout << a;
    }
    else {
        printf("No");
    }
    return 0;
}

一开始我把 j == i 写成了 j == 1 结果样例都过不了。。。