139. 单词拆分

原题

https://leetcode.cn/problems/word-break/description/

思路

动态规划

复杂度

时间：O(n²)
空间：O(n)

Python代码

class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        # dp[i] 表示 s 的前 i 位是否可以用 wordDict 中的单词表示。
        n = len(s)
        dp = [False] * (n + 1)
        dp[0] = True
        wordSet = set(wordDict)
        for i in range(n):
            for j in range(i + 1, n + 1):
                if dp[i] and s[i:j] in wordSet:
                    dp[j] = True
               
        return dp[-1]

Go代码

func wordBreak(s string, wordDict []string) bool {
	n := len(s)
	// dp[i] 表示 s 的前 i 位是否可以用 wordDict 中的单词表示。
	dp := make([]bool, n+1)
	dp[0] = true
	// 将切片转换为集合
	set := map[string]bool{}
	for _, word := range wordDict {
		set[word] = true
	}
	for i := 0; i < n; i++ {
		for j := i + 1; j < n+1; j++ {
			if dp[i] && set[s[i:j]] {
				dp[j] = true
			}
		}
	}
	return dp[len(dp)-1]
}