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

链接: http://leetcode.com/problems/substring-with-concatenation-of-all-words/

题解:

第一反应是用Trie,像spell checker一样,在Trie中检查这段text是不是word。 (待补充)

第二反应是做一个类似于DFA的状态机。 (待补充)

第三反应是看答案。默默看答案,看到大家都用HashMap,所以也写了用两个HashMap的, 提交就超时了。再想一想,还是要用sliding window, 比如text = abca, 单词为'a','b'和'c',这样0和1都是有效index,要减少重复compare.

 

HashMap:

先把word以及word count放入一个wordMap中,然后对text进行遍历。遍历的时候,每次步长为wordLength,所以对于整个text我们只需要遍历wordLength次pass。 对于每次pass,由于要找到所有复合条件的index,所以我们进行sliding window。为此我们还需要一个current map用来记录当前的window,一个lo变量来记录window的左边界,以及一个。之后对于每次pass,先看当前的单词是否存在于wordMap里,假如不存在则reset curMap,count和lo。假如存在,则把当前单词加入现在的window里。加入完毕后还需要检查重复情况,假如当前window里这个单词的计数大于wordMap里的计数,则从window左边界逐个取出单词,直到当前单词的计数等于wordMap里的计数为止。 假如count == 单词总数,则 lo 是一个解,加入到结果list里,更新lo,count,并且window向右移动一个单词。 代码写得很拖沓,有空要好好refactor。

Time Complexity - O(n), Space Complexity - O(m * l), m为单词数量,l为单词长度。

public class Solution {
    HashMap<String, Integer> wordMap;
    
    public List<Integer> findSubstring(String s, String[] words) {
        List<Integer> res = new ArrayList<>();
        if(s == null ||s.length() == 0 || words == null || words.length == 0)
            return res;
        wordMap = new HashMap<>();
        fillWordMap(words);         //put all words and their frequencey to wordMap
        int wordLen = words[0].length(), wordCount = words.length;
        HashMap<String, Integer> curMap = new HashMap<>();      //sliding window storing current word
        
        for(int i = 0; i < wordLen; i++) {            //we are going to process s word by word, so totally we need "wordLen" passes
            int j = i, lo = i, count = 0;           //lo is the index we need to record and add to result list
            curMap.clear();
            
            while(j <= s.length() - wordLen) {
                String curWord = s.substring(j, j + wordLen);
                if(!wordMap.containsKey(curWord)) {         // intervening characters found
                    curMap.clear();
                    count = 0;
                    lo = j + wordLen;
                } else {
                    if(curMap.containsKey(curWord))                     //put current word into current window
                        curMap.put(curWord, curMap.get(curWord) + 1);
                    else
                        curMap.put(curWord, 1);
                    count++;
                    
                    while(curMap.get(curWord) > wordMap.get(curWord)) {   // remove words from left end of the window until valid    
                        String rmvWord = s.substring(lo, lo + wordLen);
                        curMap.put(rmvWord, curMap.get(rmvWord) - 1);
                        count--;
                        lo += wordLen;
                    }
                    
                    if(count == wordCount) {                    //if target string found
                        res.add(lo);
                        String loWord = s.substring(lo, lo + wordLen);
                        curMap.put(loWord, curMap.get(loWord) - 1);
                        count--;
                        lo += wordLen;
                    }
                }
                
                j += wordLen;
            }
        }
        
        return res;
    }
    
    private void fillWordMap(String[] words) {
        for(String word : words) {
            if(wordMap.containsKey(word))
                wordMap.put(word, wordMap.get(word) + 1);
            else
                wordMap.put(word, 1);
        }
    }
}

 

Trie:

DFA:

Histogram:

二刷:

这里主要还是跟第一遍相同。先建立一个global的wordMap,里面还有单词以及个数。接下来做双重循环,外循环是从0 到 单词的长度,每次递增一个字符,内循环开始前我们clear curMap。内循环是从j = i开始,每次递增一个单词长度L。同时我们维护一个滑动窗口的左边界lo,以及当前复合条件的单词数目 count。 每次我们先求出当前的单词 - s.substring(j, j + wordLen),先判断其是否在wordMap里,假如不在,我们可以直接跳过L - 当前单词,从下一个单词其实为止开始查找 (这里我们要清空curMap以及count,更新lo)。假若当前单词在wordMap里, 那么我们把它加入到curMap中,之后再拿curMap中这个单词的value与wordMap中这个单词的value进行比较。假如curMap.value小,那么我们继续下面的计算。假如curMap.get(curWord) > wordMap.get(curWord),说明我们加入了多余的单词,这里我们要用类似"Sliding Window Maximum"中的方法,使用一个while循环,将这个window前部的单词一个一个poll出去。poll的过程就是先求出前部单词 s.substring(lo, lo + wordLen),然后在curMap中将其value - 1,并且count--,之后再更新lo = lo + wordLen来比较下一个首部单词。直到我们把多加入的单词去掉,使得curMap.get(curWord) <= wordMap.get(curWord)为止。  最后当count == words.length时,这时我们找到了一个解,把这个解的开头index lo加入到结果集中。然后我们要把window首部单词去掉,count--,并且增加lo = lo + wordLen,来继续进行下面的判断。

Java:

假如不考虑substring的话,应该是L次遍历,每次遍历 n / L个字符,这样应该算是 Time Complexity:  O(n), Space Complexity - O(L * m), L为单词的长度,m为单词个数。

public class Solution {
    public List<Integer> findSubstring(String s, String[] words) {
        List<Integer> res = new ArrayList<>();
        if (s == null || s.length() == 0 || words == null || words.length == 0) {
            return res;
        }
        Map<String, Integer> wordMap = new HashMap<>();
        for (String word : words) {
            if (!wordMap.containsKey(word)) {
                wordMap.put(word, 1);
            } else {
                wordMap.put(word, wordMap.get(word) + 1);
            }
        }
        int wordLen = words[0].length();
        Map<String, Integer> curMap = new HashMap<>();
        
        for (int i = 0; i < wordLen; i++) {  // start from each char
            int lo = i, count = 0;
            curMap.clear();
            for (int j = i; j <= s.length() - wordLen; j += wordLen) {
                String curWord = s.substring(j, j + wordLen);
                if (!wordMap.containsKey(curWord)) {
                    curMap.clear();
                    count = 0;
                    lo = j + wordLen;
                } else {
                    if (!curMap.containsKey(curWord)) {
                        curMap.put(curWord, 1);
                    } else {
                        curMap.put(curWord, curMap.get(curWord) + 1);
                    }
                    count++;
                    while (curMap.get(curWord) > wordMap.get(curWord)) {        // poll from front
                        String wordToRemove = s.substring(lo, lo + wordLen);
                        curMap.put(wordToRemove, curMap.get(wordToRemove) - 1);
                        lo += wordLen;
                        count--;
                    }
                    if (count == words.length) { // found one solution
                        res.add(lo);
                        String loWord = s.substring(lo, lo + wordLen);
                        curMap.put(loWord, curMap.get(loWord) - 1);
                        lo += wordLen;
                        count--;
                    }
                }
            }
        }
        return res;
    }
}

有的时候HashMap的操作也可以简写,比如

curMap.put(curWord, curMap.get(curWord) == null ? 1 : curMap.get(curWord) + 1);

 

三刷:

跟二刷基本相同。

要注意的是  j的范围是  [i,  s.length() - wordLen],前后都是闭合的。

Java:

public class Solution {
    public List<Integer> findSubstring(String s, String[] words) {
        List<Integer> res = new ArrayList<>();
        if (s == null || words == null || words.length == 0) return res;
        Map<String, Integer> wordsMap = new HashMap<>();
        for (String word : words) {
            if (!wordsMap.containsKey(word)) wordsMap.put(word, 1);
            else wordsMap.put(word, wordsMap.get(word) + 1);
        }
        int wordLen = words[0].length();
        
        for (int i = 0; i < wordLen; i++) {
            Map<String, Integer> curMap = new HashMap<>();
            int lo = i; 
            int count = 0;
            for (int j = i; j <= s.length() - wordLen; j += wordLen) {
                String word = s.substring(j, j + wordLen);
                if (!wordsMap.containsKey(word)) {
                    count = 0;
                    curMap.clear();
                    lo = j + wordLen;
                    continue;
                }
                if (!curMap.containsKey(word)) curMap.put(word, 1);
                else curMap.put(word, curMap.get(word) + 1);
                count++;
                while (curMap.get(word) > wordsMap.get(word)) {
                    String loWord = s.substring(lo, lo + wordLen);
                    curMap.put(loWord, curMap.get(loWord) - 1);
                    lo += wordLen;
                    count--;
                }
                if (count == words.length) {
                    res.add(lo);
                    String loWord = s.substring(lo, lo + wordLen);
                    curMap.put(loWord, curMap.get(loWord) - 1);
                    lo += wordLen;
                    count--;
                }
            }
        }
        return res;
    }
}

 

posted @ 2015-04-17 15:53  YRB  阅读(440)  评论(0编辑  收藏  举报