day_11 —— LeetCode 771：宝石与石头

Day_11 —— LeetCode练习题：宝石与石头

771. 宝石与石头

题目：

给定字符串J 代表石头中宝石的类型，和字符串 S代表你拥有的石头。 S 中每个字符代表了一种你拥有的石头的类型，你想知道你拥有的石头中有多少是宝石。

J 中的字母不重复，J 和 S中的所有字符都是字母。字母区分大小写，因此"a"和"A"是不同类型的石头。

示例1：

输入: J = “aA”, S = “aAAbbbb”
输出: 3

示例2：

输入: J = “z”, S = “ZZ”
输出: 0

解答：

1. C++解法一（暴力搜索）:

class Solution {
public:
    int numJewelsInStones(string J, string S) {
        int res = 0;
        for (char s : S) {
            for (char j : J) {
                if (s == j) {
                    ++res; break;
                }
            }
        }
        return res;
    }
};

2. C++解法二（用$HashSet$来优化时间复杂度，将珠宝字符串J中的所有字符都放入$HashSet$中，然后遍历石头字符串中的每个字符，到$HashSet$中查找是否存在，存在的话计数器自增1）：

class Solution {
public:
    int numJewelsInStones(string J, string S) {
        int res = 0;
        unordered_set<char> s;
        for (char c : J) s.insert(c);
        for (char c : S) {
            if (s.count(c)) ++res;
        }
        return res;
    }
};

我使用了$HashSet$来检查它是否包含一个元素，时间复杂度为$O(1)$，所以总时间复杂度是$O(J+S)$，而不是暴力搜索的$O(J\ast S)$。

3. python解法（暴力搜索）：

class Solution:
    def numJewelsInStones(self, J, S):
        return sum(s in J for s in S)

4. python解法（$Hashset$）：

只要是涉及到空间换时间，就要往集合靠拢。

class Solution:
    def numJewelsInStones(self, J, S):
        set_J = set(J)
        return sum(s in set_J for s in S)

C++与python执行对比