30. 串联所有单词的子串

Posted 2022-12-05 2016bits

一、题目描述

给定一个字符串 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：

输入：

"barfoofoobarthefoobarman"
["bar","foo","the"]
输出：
[6,9,12]

三、基本思路

本题使用双指针。（用wnum记录单词个数，wlen记录单词长度，len记录子串长度。）

设置左指针left和右指针right分别指向检查子串的第一个和最后一个单词的首字母位置，每次向后检查wlen长度的子串，判断该子串是否是words中的单词，如果是，则将right向后移动wlen长度，同时将单词计数count进行更新。如果已检查的子串（left和right之间的子串）中含有单词数目多于words中对应单词的数目，则需要移除该单词，即向右移动left。

如果count等于单词个数wnum，则已检查子串符合题目要求，将第一个字母的索引left存入结果数组ans

四、代码

class Solution 
public:
    vector<int> findSubstring(string s, vector<string>& words) 
        vector<int> ans;
        if (s == "" || words.size() == 0) 
            return ans;
        
        int wnum = words.size();//单词个数
        int wlen = words[0].length();//单词长度
        int len = wnum * wlen;//子串长度
        if (s.length() < len) 
            return ans;
        
        map<string, int> msi;
        for (int i = 0; i < wnum; i++) 
            msi[words[i]]++;
        
        for (int i = 0; i < wlen; i++) 
            int left = i, right = i;
            int count = 0;//遇到的单词个数
            map<string, int> visited;//记录遇到的单词出现次数
            while (right + wlen <= s.length()) 
                string str = s.substr(right, wlen);//以right为起点的长度为wlen的单词
                int j;
                for (j = 0; j < wnum; j++) //在words中寻找str
                    if (str == words[j]) 
                        right += wlen;
                        visited[str]++;
                        count++;
                        break;
                    
                
                if (j == wnum) //words中没有与之匹配的单词
                    right += wlen;
                    left = right;
                    count = 0;
                    visited.clear();
                    continue;
                
                while (visited[str] > msi[str]) //str多了，需要整体右移
                    string stmp = s.substr(left, wlen);
                    visited[stmp]--;
                    count--;
                    left += wlen;
                
                if (count == wnum) 
                    ans.push_back(left);
                
            
        
        return ans;
    
;