算法題解之滑動窗口

Substring with Concatenation of All Words

尋找所有詞連接的子串

思路：由於該字串是所有詞典中的詞連接的，所以該字串長度固定。因此本題可以看作一個滑動窗口的題。為了去除重復工作，每次滑動一個單詞的長度，因此起始位置就有n種（n為單詞長度）。每種起始位置的滑動策略如下：

　　　如果當前窗口滿足條件，則窗口只往后移動一個單詞，並且下一次只檢查最后一個單詞（中間的單詞肯定滿足條件）。

　　   如果當前窗口不滿足條件，則有兩種情況：

　　　1.當前窗口內不滿足條件的第一個單詞不存在於詞典中，則下一個窗口就直接滑到這個單詞的下一個單詞；

　　　2.當前窗口內不滿足條件的第一個單詞存在於詞典中，但是這個單詞已經被前面用完了，則下一個窗口只往后移動一個單詞，並且下一次之間從這個單詞開始檢查。

public class Solution {
    public List<Integer> findSubstring(String s, String[] words) {
        List<Integer> res = new ArrayList<Integer>();
        int m = words.length;
        int n = words[0].length();
        Map<String, Integer> map = new HashMap<String, Integer>();
        for (String word : words) {
            if (map.containsKey(word)) {
                map.put(word, map.get(word) + 1);
            } else {
                map.put(word, 1);
            }
        }
        for (int p = 0; p <= n - 1; p++) {
            int start = p;
            while (start + n * m - 1 < s.length()) {
                int i = 1;
                boolean moveOneWord = true;
                Map<String, Integer> copy = new HashMap<String, Integer>(map);
                while (moveOneWord && start + n * m - 1 < s.length()) {
                    for(; i <= m; i++) {
                        String cut = s.substring(start + (i - 1) * n, start + i * n);
                        if (copy.containsKey(cut)) {
                            if (copy.get(cut) == 0) {
                                moveOneWord = true;
                                break;
                            } else {
                                copy.put(cut, copy.get(cut) - 1);
                            }
                        } else {
                            start += i * n;
                            i = 1;
                            moveOneWord = false;
                            break;
                        }
                    }
                    
                    if (i == m + 1) {
                        res.add(start);
                        moveOneWord = true;
                    }
                    if (moveOneWord) {
                        String firstword = s.substring(start, start + n);
                        copy.put(firstword, copy.get(firstword) + 1);
                        start += n;
                        i--;
                    }
                }
            } 
        }
        return res;
    }
}