LeetCode每天刷day22:Substring with Concatenation of All Words

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

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

题目链接:Substring with Concatenation of All Words
C++:

class Solution {
public:
    vector findSubstring(string s, vector& words) {
        multiset wordSet;
        vector ret;
        int cnt = words.size();
        if(cnt == 0 || s.size() == 0)
            return ret;
        int aWordLen = words[0].size();
        for(auto itr : words){
            wordSet.insert(itr);
        }
        int len = cnt * aWordLen;
        for(int i = 0; i < s.size() - len + 1; i++){
            string tmp = s.substr(i, len);
            auto j = wordSet.begin();
            while(j != wordSet.end()){
                int flag = tmp.find(*j);
                while(flag % aWordLen != 0 && flag != tmp.npos){
                    flag = tmp.find(*j, flag + 1);
                }
                if(flag != tmp.npos)
                    tmp.erase(flag, aWordLen);
                else
                    break;
                j++;
            }
            if(tmp.empty())
                ret.push_back(i);
        }
        return ret;
    }
};

你可能感兴趣的:(Leetcode)