程序员面试金典 1.8 翻转子串

题目

假定我们都知道非常高效的算法来检查一个单词是否为其他字符串的子串。请将这个算法编写成一个函数，给定两个字符串s1和s2，请编写代码检查s2是否为s1旋转而成，要求只能调用一次检查子串的函数。
给定两个字符串s1,s2,请返回bool值代表s2是否由s1旋转而成。字符串中字符为英文字母和空格，区分大小写，字符串长度小于等于1000。

我的题解

class ReverseEqual {
public:
    vector<int> func(string s) {
        vector<int> res(s.size(), 0);
        res[0] = -1;
        int i = 0, pos = 0;

        for(i = 1; i < s.size(); ++ i) {
            char cur = s[i - 1];
            if(cur == s[pos]) {
                res[i] = res[i - 1] + 1;
                if(pos != (i - 1))pos ++;
            } else {
                pos = 0;
                if(cur == s[pos]) {
                    res[i] = 1;
                    pos ++;
                } else res[i] = 0;
            }
        }

        return res;
    }

    vector<int> makeNext(string s) {
        vector<int> vec = func(s);
        vector<int> next(s.size());
        next[0] = -1;
        int i = 1, f;
        for(; i < s.size(); ++ i) {
            f = vec[i];
            if(s[f] != s[i]) next[i] = f;
            else next[i] = next[f];
        }

        return next;
    }

    bool kmp(string pattern, string target) {
        vector<int> next = makeNext(pattern);
        int pos = 0, i;
        for(i = 0; i < target.size();) {
            if(target[i] == pattern[pos]) {
                pos ++;
                i ++;
            }else {
                pos = next[pos];
                if(pos == -1) {
                    pos = 0;
                    i ++;
                }
            }
            if(pos == pattern.size())
                return true;
        }
        return false;
    }

    bool checkReverseEqual(string s1, string s2) {
        // write code here
        s1 = s1 + s1;
        return kmp(s2, s1);
    }
};

同一个字符串连着写两次，这样不管怎么翻转，字串都会在这个长串里面。在用kmp算法在长串中判断就好了。