LeetCode 30: Substring with Concatenation of All Words

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).

解题思路

使用一个 Hash Table 存放 words 中的单词（记为 wordMap），其中 key 为单词， value 为单词出现的次数。

枚举符合条件的子串可能的起始位置 $i$。首先获得以 $s[i]$ 为起点的子串的第一个单词；然后查看在 wordMap 中是存在这个单词：如果不处在，指针 $i$ 直接后移匹配下一个位置开始的子串；如果存在，我们把这个单词加入到一个新的 Hash Table（记为 curMap） 中，同时递增该单词出现的次数，然后判断这个单词的出现次数是否小于等于 wordMap 中的出现次数，如果大于说明匹配失败，指针 $i$ 后移匹配下一个位置开始的子串；如果从 $i$ 位置开始我们成功匹配了 words.size() 个单词则记录 $i$。

代码如下：

class Solution {
public:
    vector<int> findSubstring(string s, vector<string>& words) {
        int wordNum = words.size();
        int wordLen = words[0].size();

        unordered_map<string, int> wordMap;
        for (int i = 0; i< wordNum; ++i) {
            wordMap[words[i]]++;
        }

        vector<int> result;
        int end = s.size() - wordNum * wordLen;
        for (int i = 0; i <= end; ++i) {
            int j;
            unordered_map<string, int> curMap;

            for (j = 0; j < wordNum; ++j) {
                string word = s.substr(i+j*wordLen, wordLen);

                if (wordMap.find(word) == wordMap.end()) break;

                curMap[word]++;

                if (curMap[word] > wordMap[word]) break;
            }

            if (j == wordNum) {
                result.push_back(i);
            }
        }

        return result;
    }
};