Not Only Algorithm,不仅仅是算法,关注数学、算法、数据结构、程序员笔试面试以及一切涉及计算机编程之美的内容 。。
你的位置:NoAlGo博客 » 题解 » 

Leetcode Substring with Concatenation of All Words

Leetcode algorithms 第 30 题:Substring with Concatenation of All Words。

题目

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

For example, given:
S: “barfoothefoobarman”
L: ["foo", "bar"]

You should return the indices: [0,9].
(order does not matter).

解答

直接枚举所有的起点,判断从该点开始的单词是否都在L中。具体实现时使用map进行保存,方便处理。

具体代码如下:

class Solution {
public:
    vector<int> findSubstring(string S, vector<string> &L) {
        int m = S.size(), n = L.size(), len = L[0].size();
        
        map<string, int> mp, cur;
        for (int i = 0; i < n; i++)
            mp[L[i]]++; //每个词出现的次数
        
        vector<int> ans;
        for (int i = 0, j, k; (k = i + n * len) <= m; i++) //枚举起点
        {
            cur.clear();
            for (j = i; j < k; j += len) //一个单词一个单词前进
            {
                string tem = S.substr(j, len);
                cur[tem]++;
                if (cur[tem] > mp[tem]) break;
            }
            if (j >= k) ans.push_back(i);
        }
        return ans;
    }
};
上一篇: 下一篇:

我的博客

NoAlGo头像编程这件小事牵扯到太多的知识,很容易知其然而不知其所以然,但真正了不起的程序员对自己程序的每一个字节都了如指掌,要立足基础理论,努力提升自我的专业修养。

站内搜索

最新评论