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
}

在这里插入图片描述

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐