刷题打卡:字符串操作の 串联所有单词的子串

126 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招系列活动-刷题打卡任务,点击查看活动详情

一、题目描述:

串联所有单词的子串

给定一个字符串 s 和一些 长度相同 的单词 words 。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。

注意子串要与 words 中的单词完全匹配,中间不能有其他字符 ,但不需要考虑 words 中单词串联的顺序。

示例 1:

输入:s = "barfoothefoobarman", words = ["foo","bar"]
输出:[0,9]
解释:
从索引 0 和 9 开始的子串分别是 "barfoo" 和 "foobar" 。
输出的顺序不重要, [9,0] 也是有效答案。

示例 2:

输入:s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
输出:[]

示例 3:

输入:s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
输出:[6,9,12]

二、思路与实现:

整体思路就是滑动窗口+哈希表计数:

遍历words,统计其中每个单词出现的次数记录wm 遍历字符串s,循环words中一个单词的长度awl即可

定义左右指针l,r,出现words中单词的个数count,当前循环单词出现的次数记录win 通过截取ss=substring(r, r+awl),查看ss是否存在于wm

不存在,清空win和count,l移动至r 存在的话,记录win[ss],count+1

当win[ss]>wm[ss]时需要从左边l开始缩小窗口,减去win中s.substring(l,l+awl)出现的次数,且count-1 count=words.length的话,符合条件;

var findSubstring = function(s, words) {
    let wsl = words[0].length,wl=words.length,wm={}
    for(const w of words){
        wm[w] = (wm[w]||0)+1
    }
    let i=0,l=0,r=0,count=0,res=[],win = {}
    while(i<wsl){
        l=r=i,count=0,win = {}
        while(r<=s.length-wsl){
            let ss = s.substring(r, r+wsl)
            r+=wsl
            if(!wm[ss]){
                win = {}
                count = 0
                l=r
            }else{
                win[ss] = (win[ss]||0)+1
                count++
                while(win[ss]>wm[ss]){
                    let sl = s.substring(l,l+wsl)
                    win[sl] = (win[sl]||0)-1
                    count--
                    l+=wsl
                }
                if(count===wl) res.push(l)
            }
        }
        i++
    }
    return res
};

image.png

三、总结:

这道题最大的亮点就是应用了 HashMap ,使得我们不再纠结于子串包含单词的顺序。