每日一题(注意string里是char)-771. 宝石与石头

最新推荐文章于 2024-09-26 19:24:27 发布

Bu9_Maker

最新推荐文章于 2024-09-26 19:24:27 发布

阅读量151

点赞数

分类专栏： Leetcode每日一题文章标签： leetcode python c++

本文链接：https://blog.youkuaiyun.com/qq_39225746/article/details/131901496

版权

Leetcode每日一题专栏收录该内容

184 篇文章

订阅专栏

文章讨论了一个简单的算法问题——找出宝石与石头中的宝石数量。通过使用C++的unordered_map和Python的set，可以将时间复杂度优化到O(n)。代码示例分别展示了两种语言的实现方式，强调了不同数据结构在优化循环效率上的作用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

771. 宝石与石头

题解思路

题目本身没啥难度，两个for循环就可以解决
但是如果使用set可以将时间复杂度优化到o(n)

注：之前python写多了，需要注意c++中string类中是char

代码

C++

class Solution {
public:
    int numJewelsInStones(string jewels, string stones) {
        unordered_map<char, int> a;
        for (auto &c : jewels){
            a[c] = 1;
        }
        int ans = 0;
        for (auto &c : stones){
            if (a.count(c)){
                ans++;
            }
        }
        return ans;
    }
};

Python

class Solution:
    def numJewelsInStones(self, jewels: str, stones: str) -> int:
        dic = set()
        ans = 0
        for ch in jewels:
            dic.add(ch)
        for ch in stones:
            if ch in dic:
                ans += 1
        return ans