lintcode-13-字符串查找

字符串查找

對於一個給定的 source 字符串和一個 target 字符串，你應該在 source 字符串中找出 target 字符串出現的第一個位置(從0開始)。如果不存在，則返回 -1。

說明

在面試中我是否需要實現KMP算法？
不需要，當這種問題出現在面試中時，面試官很可能只是想要測試一下你的基礎應用能力。當然你需要先跟面試官確認清楚要怎么實現這個題。

樣例

如果 source = "source" 和 target = "target"，返回 -1。
如果 source = "abcdabcdefg" 和 target = "bcd"，返回 1。

挑戰

O(n2)的算法是可以接受的。如果你能用O(n)的算法做出來那更加好。（提示：KMP）

標簽

基本實現 字符串處理 臉書

方法一，暴力破解

class Solution {
public:
    /**
     * Returns a index to the first occurrence of target in source,
     * or -1  if target is not part of source.
     * @param source string to be scanned.
     * @param target string containing the sequence of characters to match.
     */
    int strStr(const char *source, const char *target) {
        // write your code here
        if(source == NULL || target == NULL)
            return -1;
        if(source[0] == '\0' && target[0] == '\0')
            return 0;
        if(target[0] == '\0')
            return 0;
        int sourceLen = strlen(source), targetLen = strlen(target);
        int i=0, j=0;
        if (sourceLen < targetLen)
            return -1;
        
        while(i < sourceLen) {
            if(source[i] == target[j]) {
                i++;
                j++;
            }
            else {
                i = i-j+1;
                j = 0;
            }
            if(target[j] == '\0')
                return i-j;
        }
        return -1;  
    }
};

方法二：KMP算法

class Solution {
public:
    /**
     * Returns a index to the first occurrence of target in source,
     * or -1  if target is not part of source.
     * @param source string to be scanned.
     * @param target string containing the sequence of characters to match.
     */
    int strStr(const char *source, const char *target) {
        // write your code here
        if(source == NULL || target == NULL)
            return -1;
        if(source[0] == '\0' && target[0] == '\0')
            return 0;
        if(target[0] == '\0')
            return 0;

        int sourceLen = strlen(source), targetLen = strlen(target);
        int *next = getNext(target, targetLen);

        int i=0, j=0;
        for (i=0; i<sourceLen; i++) {
            while (j > 0 && source[i] != target[j])  
                j = next[j];
            
            if (source[i] == target[j])  
                j++;

            if (j == targetLen) {
                return i-j+1;
                j = next[j];
            }  
        }
        return -1;  
    }

    int *getNext(const char *target, int targetLen) {
        int *next = new int[targetLen+1];
        int i=0, j=0;

        next[0] = next[1] = 0; 
          
        for(i=1; i<targetLen; i++) {
            while(j>0 && target[i]!=target[j])
                j = next[j];
            if(target[i] ==target[j])
                j++;  
            next[i+1] = j;  
        }  
          
        return next;  
    }
};