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

代码GO

func numJewelsInStones(jewels string, stones string) int {
	ans := 0
	// 1. 将jewels映射到数组中
	var arr [52]int
	for _, val := range jewels {
		if val >= 65 && val <= 91 {
			arr[val-65] = 1
		} else {
			arr[val-97+26] = 1
		}
	}
	// 2. 将stones映射到数组中
	for _, val := range stones {
		if val >= 97 && arr[val-97+26] == 1 {
			ans++
		}
		if val <= 91 && arr[val-65] == 1 {
			ans++
		}
	}

	return ans
}