30.与所有单词相关联的字串

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

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

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

示例 2:
输入:
s = "wordgoodstudentgoodword",
words = ["word","student"]
输出: []

思路
使用一个map，键为字符串数组中的字符串，值为该字符串在字符串数组中出现的次数。遍历字符串s，寻找和字符串数组中的字符串相同的字串，找到后map中的值减一，否则重新初始化map，从下一个字符开始遍历。如果map中所有的值都为0，则找到了一个符合条件的子串，索引压入数组。

#include 
#include
#include 
using namespace std;
class Solution {
public:
    void initializeMAP(map& map, vector& words)
    {
        for (int i = 0; i < words.size(); i++)
        {
            if (map.count(words[i]) == 0) map[words[i]] = 1;
            else { map[words[i]] += 1; }
        }
    }

    vector findSubstring(string s, vector& words) {
        vector result;
        if (s.empty() || words.empty()) return result;
        map mapOfVec;
        int singleStringLen = words[0].length(), wordLen = words.size(), sLen = s.length();
        bool countChang = false;//判断是否改变过map中的值，如果没变则无需重新初始化
        int count, j;
        count = wordLen; //一个计数器表示还需要找到的字串个数
        initializeMAP(mapOfVec, words);
        for (int i = 0; i < sLen - wordLen * singleStringLen; i++)
        {
            string substr = s.substr(i, singleStringLen);
            j = i;
            while (mapOfVec.count(substr) != 0 && mapOfVec[substr] != 0 && j + singleStringLen <= sLen)//当该字串存在于map中且值大于0，并且j不越界的情况下
            {
                mapOfVec[substr] -= 1;
                count--;
                countChang = true;
                j += singleStringLen;
                substr = s.substr(j, singleStringLen);
                if(mapOfVec.count(substr) == 0) break;
            }
            if (count == 0) result.push_back(i);//找齐所有字符串数组中的字串后把该索引压入；
            if (countChang)//若未找到而且改变了map的值需要重新初始化map和count
            {
                mapOfVec.clear();
                initializeMAP(mapOfVec, words);
                count = wordLen;
                countChang = false;
            }
        }
        return result;
    }
};

int main(int argc, char* argv[])
{
    string s = "barfoothefoobarman";
    vector words = { "foo", "bar" };
    auto res = Solution().findSubstring(s, words);
    return 0;
}

30.与所有单词相关联的字串

你可能感兴趣的:(30.与所有单词相关联的字串)