771. 宝石与石头【哈希表】

771. 宝石与石头

771. 宝石与石头

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

字母区分大小写，因此"a"和 "A" 是不同类型的石头。

示例 1：

输入：jewels = "aA", stones = "aAAbbbb"
输出：3

示例 2：

输入：jewels = "z", stones = "ZZ"
输出：0

提示：

• 1 <= jewels.length, stones.length <= 50
• jewels 和 stones 仅由英文字母组成
• jewels 中的所有字符都是 唯一的

解题思路

用一个map存储哪些字符是宝石，然后遍历石头，找出是宝石的数量即可。

Go代码

func numJewelsInStones(jewels string, stones string) int {
    if len(jewels) == 0 {
        return 0
    }
    // 宝石集合
    m := make(map[byte]struct{})
    for i := 0;i < len(jewels);i++ {
        m[jewels[i]] = struct{}{}
    }

    // 统计石头中宝石的数量
    res := 0
    for i := 0;i < len(stones);i++ {
        if _,ok := m[stones[i]];ok {
            res++
        }
    }

    return res
}