算法--统计匹配检索规则的物品数量

35 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第21天,点击查看活动详情

题目

leetcode 1773. 统计匹配检索规则的物品数量 难度:简单

给你一个数组 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"

所有字符串仅由小写字母组成

通过次数40,430提交次数46,341

题解

方法一: 可以利用哈希表把输入ruleKey 转换为items[i] 的下标,然后再遍历一遍 items,找出符合条件的物品数量

var countMatches = function(items, ruleKey, ruleValue) {
    const index = {"type":0, "color":1, "name":2}[ruleKey];
    let res = 0;
    for (const item of items) {
        if (item[index] === ruleValue) {
            res++;
        }
    }
    return res;
};

方法二: 本题简单模拟,先确定ruleKey,再遍历items即可解题。

class Solution {
    public int countMatches(List<List<String>> items, String ruleKey, String ruleValue) {
        int ans = 0;
        int idx = ruleKey.charAt(0) == 't' ? 0 : (ruleKey.charAt(0) == 'c' ? 1 : 2);
        for(List<String> item : items) {
            if(ruleValue.equals(item.get(idx))) {
                ++ans;
            }
        }
        return ans;
    }
}

代码详解

由于 ruleKey 只可能是 "type"、"color" 或 "name",我们可以直接取 ruleKey 的第一个字符来确定 item 的下标 ii。

然后遍历 items 数组,统计 item[i] == ruleValue 的个数即可。