LeetCode 单词子集(字符串处理)

我们给出两个单词数组 A 和 B。每个单词都是一串小写字母。

现在,如果 b 中的每个字母都出现在 a 中,包括重复出现的字母,那么称单词 b 是单词 a 的子集。 例如,“wrr” 是 “warrior” 的子集,但不是 “world” 的子集。

如果对 B 中的每一个单词 b,b 都是 a 的子集,那么我们称 A 中的单词 a 是通用的。

你可以按任意顺序以列表形式返回 A 中所有的通用单词。

示例 1:

输入:A = ["amazon","apple","facebook","google","leetcode"], B = ["e","o"]
输出:["facebook","google","leetcode"]

示例 2:

输入:A = ["amazon","apple","facebook","google","leetcode"], B = ["l","e"]
输出:["apple","google","leetcode"]

示例 3:

输入:A = ["amazon","apple","facebook","google","leetcode"], B = ["e","oo"]
输出:["facebook","google"]

示例 4:

输入:A = ["amazon","apple","facebook","google","leetcode"], B = ["lo","eo"]
输出:["google","leetcode"]

示例 5:

输入:A = ["amazon","apple","facebook","google","leetcode"], B = ["ec","oc","ceo"]
输出:["facebook","leetcode"]

提示:

1 <= A.length, B.length <= 10000
1 <= A[i].length, B[i].length <= 10
A[i] 和 B[i] 只由小写字母组成。
A[i] 中所有的单词都是独一无二的,也就是说不存在 i != j 使得 A[i] == A[j]。

思 路 分 析 : \color{blue}思路分析: 这道题没啥转弯的地方,只需要知道如何判断一个单词是否能够满足B中所有单词构造所需要的字母即可。

首先我们需要知道一个并集思想,假设B中的word1需要3个'a',2个'b',4个'c',word2需要4个'a',1个'b',5个'd',则word1、word2共需要4个'a',2个'b',4个'c',5个'd'。注意字母可以复用!

因此我们需要确定B中构造所有单词需要的最多字母种类数和对应的个数,接着我们判断A中的单词是否满足要求即可。

class Solution {
public:
    vector<string> wordSubsets(vector<string>& A, vector<string>& B) {
        vector<int> needVec(26, 0);//找到B中所有单词需要的总字母及个数
        for (const auto &word : B){
            vector<int> tempVec(26, 0);//获取word需要的字母以及个数
            for (const auto &ch : word){
                ++tempVec[ch - 'a'];
            }
            //更新B需要的最大容量字母及个数
            for (int i = 0; i < 26; ++i){
                if (needVec[i] < tempVec[i]){
                    needVec[i] = tempVec[i];
                }
            }
        }
        vector<string> resStrVec;
        for (const auto &word : A){//扫描A
            if (cherk(needVec, word)){
                //如果word含有需要的所有字母,则放入结果
                resStrVec.push_back(word);
            }
        }
        return resStrVec;
    }
    //判断单词word中是否含有needVec所需要的单词
    bool cherk(const vector<int> &needVec, const string &word){
        vector<int> tempVec(26, 0);
        for (const auto &ch : word){//确定word中含有的单词以及个数
            ++tempVec[ch - 'a'];
        }
        for (int i = 0; i < 26; ++i){
            if (needVec[i] > tempVec[i]){//如果字母'a' + i不够
                return false;
            }
        }
        return true;
    }
};

LeetCode 单词子集(字符串处理)_第1张图片

你可能感兴趣的:(LeetCode,字符串)