LeetCode 30. 与所有单词相关联的字串(C++)

题目描述

给定一个字符串 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”]
输出: []

思路

这一题确实有些难度，刚开始比较难想到怎样去解决比较好，这里使用字符串分割、Arrays.sort排序、再比较的方法，依次遍历s分割数组元素个数*元素长度的子串，然后再对这个子串再进行分割，创建一个字符串数组保存得到的结果；对该数组和待检测数组排序后比较（在java中比较两个数组是否相同需要使用Arrays.equal(s1,s2)，而不是s1.equals(s2)，这里需要注意一下），就可以得到所有位置，但是这种方法因为需要遍历所以效率不高。

代码

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        vector<int> re;
        if(s.size()==0 || words.size()==0) return re;
        int slen=words.at(0).size();
        int len1=s.size();
        int len2=words.size()*slen;
        if(len1<len2) return re;
        vector<string> vs;
        sort(words.begin(),words.end());
        for(int i=0;i<=len1-len2;i++)
        {
            for(int j=0;j<words.size();j++)
                vs.push_back(s.substr(i+slen*j,slen));
            sort(vs.begin(),vs.end());
            if(words==vs) re.push_back(i);
            vs.clear();
        }
        return re;        
    }
};