Rabin-karp算法实现字符串匹配

最新推荐文章于 2024-03-08 10:50:11 发布

原创最新推荐文章于 2024-03-08 10:50:11 发布 · 642 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#go #rabin-karp #字符串匹配

算法专栏收录该内容

13 篇文章

订阅专栏

本文详细介绍了Rabin-Karp算法的实现过程，这是一种高效字符串匹配算法，通过使用哈希函数来快速查找一个字符串是否包含另一个子串。文章中包含了算法的具体代码实现，包括哈希值的计算和滚动哈希的更新，适用于处理大规模文本搜索问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

// RabinKarp算法实现

// RabinKarp算法实现
const primeRK = 16777619

func hashStr(seq string) (uint32, uint32) {
    hash := uint32(0)
    for _, value := range seq {
        hash = hash * primeRK + uint32(value)
    }
    
    var pow, sq uint32 = 1, primeRK
    for i := len(seq); i > 0; i >>= 1 {
        if i & 1 != 0 {
            pow *= sq
        }
        sq *= sq
    }
    return hash, pow
}

func rabinKarp(s, substr string) int {
    hashss, pow := hashStr(substr)
    n := len(substr)
    var h uint32
    for i := 0; i < n; i++ {
		h = h*primeRK + uint32(s[i])
	}
	if h == hashss && s[:n] == substr {
		return 0
	}
    for i := n; i < len(s); {
		h *= primeRK
		h += uint32(s[i])
		h -= pow * uint32(s[i-n])
		i++
		if h == hashss && s[i-n:i] == substr {
			return i - n
		}
	}
	return -1
}

func strStr(haystack string, needle string) int {
    n := len(needle) 
    switch {
        case n == 0:
            return 0
        case n == len(haystack):
            if haystack == needle {
                return 0
            }
            return -1
        case n > len(haystack):
            return -1
    }
    return rabinKarp(haystack, needle)
}