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 中的所有字符都是唯一的

解析

用 Set 存储宝石类型，遍历石头统计即可。

var numJewelsInStones = function (jewels, stones) {
  const set = new Set(jewels);
  let count = 0;
  for (const s of stones) {
    if (set.has(s)) count++;
  }
  return count;
};

时间复杂度 O(m+n)，空间复杂度 O(m)，其中 m 为宝石种类数，n 为石头总数。