2021-05-22:假设所有字符都是小写字母, 大字符串是str,arr是去重的单词表, 每个单词都不是空字符串且可以使用任意次。使用arr中的单词有多少种拼接str的方式。 返回方法数。

2021-05-22:假设所有字符都是小写字母, 大字符串是str,arr是去重的单词表, 每个单词都不是空字符串且可以使用任意次。使用arr中的单词有多少种拼接str的方式。 返回方法数。

福大大 答案2021-05-22:

前缀树。时间复杂度O(N2)。传统方法的时间复杂度是O(N3)。

代码用golang编写。代码如下:

package main

import "fmt"

func main() {
    s := "aaab"
    arr := []string{"a", "aa", "aaa", "ab", "b"}
    ret := ways4(s, arr)
    fmt.Println(ret)
}

type Node struct {
    end   bool
    nexts []*Node
}

func NewNode() *Node {
    ans := &Node{}
    ans.nexts = make([]*Node, 26)
    return ans
}

func ways4(s string, arr []string) int {
    if len(s) == 0 || len(arr) == 0 {
        return 0
    }
    root := NewNode()
    for _, str := range arr {
        node := root
        index := 0
        for i := 0; i < len(str); i++ {
            index = int(str[i] - 'a')
            if node.nexts[index] == nil {
                node.nexts[index] = NewNode()
            }
            node = node.nexts[index]
        }
        node.end = true
    }

    N := len(s)
    dp := make([]int, N+1)
    dp[N] = 1
    for i := N - 1; i >= 0; i-- {
        cur := root
        for end := i; end < N; end++ {
            path := int(s[end] - 'a')
            if cur.nexts[path] == nil {
                break
            }
            cur = cur.nexts[path]
            if cur.end {
                dp[i] += dp[end+1]
            }
        }
    }
    return dp[0]
}

执行结果如下:
图片


左神java代码

posted @   福大大架构师每日一题  阅读(15)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· AI 智能体引爆开源社区「GitHub 热点速览」
· Manus的开源复刻OpenManus初探
· 写一个简单的SQL生成工具
点击右上角即可分享
微信分享提示