LeetCode OJ Substring with Concatenation of All Words

https://oj.leetcode.com/discuss/366/better-solution-than-brute-force

看了下面的评论。

思路是：例如 "foobarisbarfoobar" ["foo","bar"]，那么将S划分为：

1、foo, bar, isb, arf, oob, ar;

2、oob, ari, sba, rfo, oba, r;

3、oba, ris, bar, foo, bar;

再从各个划分中逐个匹配。

class Solution {
public:
    map<string, int> shouldBe;  // L中string应有的数量
    map<string, int> now;  // 当前构建的string的数量
    vector<int> findSubstring(string S, vector<string> &L) {
        vector<int> ans;
        for (int i = 0; i < L.size(); i++) {
            if (shouldBe.find(L[i]) != shouldBe.end()) {
                shouldBe[L[i]]++;
            } else {
                shouldBe[L[i]] = 1;
            }
        }
        int alength = L[0].size();  // L中每个string的长度
        int sNum, nNum;  // S中某个string的个数和now中对应string的个数
        int startPos;  // 符合条件的某个substr的起点

        for (int q = 0; q < alength; q++) {
            now.clear();
            for (int i = q; i + alength <= S.size(); i += alength) {

                if (now.empty()) startPos = i;  // 如果now是空的说明是新的开始

                string temp = S.substr(i, alength);

                if (shouldBe.find(temp) != shouldBe.end()) sNum = shouldBe[temp];  // 判断现在的temp是否在S中
                else sNum = 0;

                if (sNum == 0) {  // 如果不在清空now，准备下一次新的开始
                    now.clear();
                } else {  // 如果在S中
                    if (now.find(temp) == now.end()) {  // 如果不在now中
                        now[temp] = 1;
                        nNum = 1;
                    } else {  // 如果在now中
                        nNum = ++now[temp];
                    }
                    if (nNum > sNum) {  // 如果now中的temp数量已经多过了S中，那么就重新开始，注意i的坐标应是当前startPos的下一位
                        now.clear();
                        i = startPos;
                    }
                    if (now == shouldBe) {  // 如果匹配成功
                        ans.push_back(startPos);
                        i = startPos;
                        now.clear();
                    }
                }
            }
        }
        return ans;
    }
};