LeetCode（30） Substring with Concatenation of All Words

题目

You are given a string, s, and a list of words, words, that are all of the same length. Find all starting indices of substring(s) in s that is a concatenation of each word in words exactly once and without any intervening characters.

For example, given:

s: “barfoothefoobarman”

words: [“foo”, “bar”]

You should return the indices: [0,9].

(order does not matter).

分析

解决该问题的关键是理解清楚要求。

给定一个目标字符串s，一个单词集合words。

要求使得words集合中所有元素连续出现在s中的首位置组成的集合（元素顺序不考虑）。

正如所给实例，目标字符串s: “barfoothefoobarman”

对比单词集合words: [“foo”, “bar”]

我们发现，在pos=0 ~ 5时“barfoo”恰好匹配，则0压入结果vector；

在pos=9 ~ 14时“foobar”恰好匹配，则9压入结果vector；

在理清楚题意后，便可入手程序实现。

AC代码

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        if (words.empty())
            return vector<int>();

        vector<int> ret;
        //记录所给words中每个单词的出现次数
        map<string, int> word_count;

        //每个单词的长度相同
        int word_size = strlen(words[0].c_str());
        int word_nums = words.size();
        //所给匹配字符串的长度
        int s_len = strlen(s.c_str());

        for (int i = 0; i < word_nums; i++)
            ++word_count[words[i]];

        int i, j;
        map<string, int> temp_count;
        for (i = 0; i < s_len - word_nums*word_size + 1; ++i)
        {
            temp_count.clear();
            for (j = 0; j < word_nums; j++)
            {
                //检验当前单词是否属于words以及出现的次数是否一致
                string word = s.substr(i + j*word_size, word_size);
                if (word_count.find(word) != word_count.end())
                {
                    ++temp_count[word];
                    //如果出现的次数与words不一致，则返回错误
                    if (temp_count[word] > word_count[word])
                        break;
                }//if
                else{
                    break;
                }//else
            }//for
            //所有words内的单词,在i起始位置都出现，则将下标i存入结果的vector中
            if (j == word_nums)
            {
                ret.push_back(i);
            }//if
        }//for
        return ret;
    }
};

GitHub测试程序源码