程序員面試金典 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算法在長串中判斷就好了。