[LeetCode] 792. Number of Matching Subsequences 匹配的子序列的个数

 

Given a string s and an array of strings words, return the number of words[i] that is a subsequence of s.

A subsequence of a string is a new string generated from the original string with some characters (can be none) deleted without changing the relative order of the remaining characters.

  • For example, "ace" is a subsequence of "abcde".

 

Example 1:

Input: s = "abcde", words = ["a","bb","acd","ace"]
Output: 3
Explanation: There are three strings in words that are a subsequence of s: "a", "acd", "ace".

Example 2:

Input: s = "dsahjpjauf", words = ["ahjpjau","ja","ahbwzgqnuk","tnmlanowax"]
Output: 2

 

Constraints:

  • 1 <= s.length <= 5 * 104
  • 1 <= words.length <= 5000
  • 1 <= words[i].length <= 50
  • s and words[i] consist of only lowercase English letters.

 

这道题给了一个字符串S,又给了一个单词数组,问数组有多少个单词是字符串S的子序列。注意这里是子序列,而不是子串,子序列并不需要连续。那么只要知道如何验证一个子序列的方法,就可以先尝试暴力搜索法,对数组中的每个单词都验证一下是否是字符串S的子序列。验证子序列的方法就是用两个指针,对于子序列的每个一个字符,都需要在母字符中找到相同的,在母字符串所有字符串遍历完之后或之前,只要子序列中的每个字符都在母字符串中按顺序找到了,那么就验证成功了。很不幸,这种暴力搜索的方法在 C++ 的解法版本中会TLE,貌似 Java 版本的可以通过,感觉 C++ 被 dis 了诶~ However,我们可以进行优化呀,在暴力搜索的基础上稍作些优化,就可以骗过 OJ 啦。下面这种优化的 motivation 是由于看了使暴力搜索跪了的那个 test case,其实是 words 数组里有大量相同的单词,而且字符串S巨长无比,那么为了避免相同的单词被不停的重复检验,用两个 HashSet 来记录验证过的单词,为啥要用两个呢?因为验证的结果有两种,要么通过,要么失败,要分别存在两个 HashSet 中,这样再遇到每种情况的单词时,就知道要不要结果增1了。如果单词没有验证过的话,就用双指针的方法进行验证,然后根据结果的不同,存到相应的 HashSet 中去,参见代码如下:

 

解法一:

class Solution {
public:
    int numMatchingSubseq(string S, vector<string>& words) {
        int res = 0, n = S.size();
        unordered_set<string> pass, out;
        for (string word : words) {
            if (pass.count(word) || out.count(word)) {
                if (pass.count(word)) ++res; 
                continue;
            }
            int i = 0, j = 0, m = word.size();
            while (i < n && j < m) {
                if (word[j] == S[i]) ++j;
                ++i;
            }
            if (j == m) {++res; pass.insert(word);}
            else out.insert(word);
        }
        return res;
    }
};

 

上面的解法已经优化的不错了,但是我们还有更叼的方法。这种解法按照每个单词的首字符进行群组,群组里面保存的是一个 pair 对,由当前字母和下一个位置组成的。然后在遍历字符串S的时候,根据当前遍历到的字母,进入该字母对应的群组中处理,如果群组中某个 pair 的下一个位置已经等于单词长度了,说明该单词已经验证完成,是子序列,结果自增1;否则的话就将下一个位置的字母提取出来,然后将 pair 中的下一个位置自增1后组成的新 pair 加入之前提取出的字母对应的群组中。是不是读到这里已然懵逼了,没关系,博主会举栗子来说明的,就拿题目中的那个例子来说吧:

S = "abcde"

words = ["a", "bb", "acd", "ace"]

那么首先将 words 数组中的单词按照其首字母的不同放入对应的群组中,得到:

a -> {0, 1}, {2, 1}, {3, 1}

b -> {1, 1}

这里,每个pair的第一个数字是该单词在words中的位置,第二个数字是下一个字母的位置。比如 {0, 1} 表示 "a" 在words数组中位置为0,且下一个位置为1(因为当前位置是首字母)。{2, 1} 表示 "acd" 在words数组中位置为2,且下一个位置为1。{3, 1} 表示 "ace" 在words数组中位置为3,且下一个位置为1。{1, 1} 表示 "bb" 在words数组中位置为1,且下一个位置为1。

好,下面来遍历字符串S,第一个遇到的字母是 'a'。

那么群组中a对应了三个 pair,将其提取出来分别进行操作。首先处理 {0, 1},此时发现下一个位置为1,和单词"a"的长度相同了,说明是子序列,结果 res 自增1。然后处理 {2, 1},在 "acd" 中取下一个位置1的字母为'c',则将下一位置自增1后的新 pair {2, 2} 加入c对应的群组。然后处理 {3, 1},在 "ace" 中取下一个位置1的字母为 'c',则将下一位置自增1后的新 pair {3, 2} 加入c对应的群组。则此时的群组为:

b -> {1, 1}

c -> {2, 2}, {3, 2}

好,继续来遍历字符串S,第二个遇到的字母是 'b'。

那么群组中b对应了一个 pair,处理 {1, 1},在 "bb" 中取下一个位置1的字母为 'b',则将下一位置自增1后的新 pair {1, 2} 加入b对应的群组。则此时的群组为:

b -> {1, 2}

c -> {2, 2}, {3, 2}

好,继续来遍历字符串S,第三个遇到的字母是 'c'。

那么群组中c对应了两个 pair,将其提取出来分别进行操作。首先处理 {2, 2},在 "ace" 中取下一个位置2的字母为 'e',则将下一位置自增1后的新 pair {2, 3} 加入e对应的群组。然后处理 {3, 2},在 "acd" 中取下一个位置2的字母为 'd',则将下一位置自增1后的新 pair {3, 3} 加入d对应的群组。则此时的群组为:

b -> {1, 2}

d -> {3, 3}

e -> {2, 3}

好,继续来遍历字符串S,第四个遇到的字母是 'd'。

那么群组中d对应了一个 pair,处理 {3, 3},此时发现下一个位置为3,和单词 "acd" 的长度相同了,说明是子序列,结果 res 自增1。则此时的群组为:

b -> {1, 2}

e -> {2, 3}

好,继续来遍历字符串S,第五个遇到的字母是 'e'。

那么群组中e对应了一个 pair,处理 {2, 3},此时发现下一个位置为3,和单词 "ace" 的长度相同了,说明是子序列,结果 res 自增1。则此时的群组为:

b -> {1, 2}

此时S已经遍历完了,已经没有b了,说明 "bb" 不是子序列,这 make sense,返回结果 res 即可,参见代码如下:

 

解法二:

class Solution {
public:
    int numMatchingSubseq(string S, vector<string>& words) {
        vector<pair<int, int>> all[128];
        int res = 0, n = words.size();
        for (int i = 0; i < n; i++) {
            all[words[i][0]].emplace_back(i, 1);
        }
        for (char c : S) {
            auto vect = all[c];
            all[c].clear();
            for (auto it : vect) {
                if (it.second == words[it.first].size()) ++res;
                else all[words[it.first][it.second++]].push_back(it);
            }
        }
        return res;
    }
};

 

Github 同步地址:

https://github.com/grandyang/leetcode/issues/792

 

类似题目:

Is Subsequence

 

参考资料:

https://leetcode.com/problems/number-of-matching-subsequences/solution/

https://leetcode.com/problems/number-of-matching-subsequences/discuss/117634/Efficient-and-simple-go-through-words-in-parallel-with-explanation

 

LeetCode All in One 题目讲解汇总(持续更新中...)

posted @ 2018-06-19 21:53  Grandyang  阅读(6970)  评论(0编辑  收藏  举报
Fork me on GitHub