LeetCode 1773. 统计匹配检索规则的物品数量(5行核心代码)

Posted 2022-11-27 Tisfy

【LetMeFly】1773.统计匹配检索规则的物品数量(5行核心代码)

力扣题目链接：https://leetcode.cn/problems/count-items-matching-a-rule/

给你一个数组 items ，其中 items[i] = [typei, colori, namei] ，描述第 i 件物品的类型、颜色以及名称。

另给你一条由两个字符串 ruleKey 和 ruleValue 表示的检索规则。

如果第 i 件物品能满足下述条件之一，则认为该物品与给定的检索规则 匹配 ：

• ruleKey == "type" 且 ruleValue == typei 。
• ruleKey == "color" 且 ruleValue == colori 。
• ruleKey == "name" 且 ruleValue == namei 。

统计并返回 匹配检索规则的物品数量 。

 

示例 1：

输入：items = [["phone","blue","pixel"],["computer","silver","lenovo"],["phone","gold","iphone"]], ruleKey = "color", ruleValue = "silver"
输出：1
解释：只有一件物品匹配检索规则，这件物品是 ["computer","silver","lenovo"] 。

示例 2：

输入：items = [["phone","blue","pixel"],["computer","silver","phone"],["phone","gold","iphone"]], ruleKey = "type", ruleValue = "phone"
输出：2
解释：只有两件物品匹配检索规则，这两件物品分别是 ["phone","blue","pixel"] 和 ["phone","gold","iphone"] 。注意，["computer","silver","phone"] 未匹配检索规则。

 

提示：

• 1 <= items.length <= 104
• 1 <= typei.length, colori.length, namei.length, ruleValue.length <= 10
• ruleKey 等于 "type"、"color" 或 "name"
• 所有字符串仅由小写字母组成

方法一：遍历

由参数“ruleKey”可以得到我们关注的是数组中的第几个元素：

• 若为“type”则我们只关注下标0
• 若为“color”则我们只关注下标1
• 若为“name”则我们只关注下标2

这是因为“item”的存放顺序为“[type, color, name]”

知道了我们关注的是第几个元素之后，我们只需要遍历一遍“items”数组，将每个“item”的对应元素与“ruleValue”比对，若相同则答案数量加一。

最终只需要返回答案即可。

• 时间复杂度$O(n)$
• 空间复杂度$O(1)$

AC代码

C++

class Solution 
public:
    int countMatches(vector<vector<string>>& items, string& ruleKey, string& ruleValue) 
        int ans = 0;
        int th = (ruleKey == "type") ? 0 : (ruleKey == "color" ? 1 : 2);
        for (auto& item : items) 
            ans += item[th] == ruleValue;
        
        return ans;
    
;

同步发文于CSDN，原创不易，转载请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/127582940