题目:
给你一个数组 items ,其中 items[i] = [typei, colori, namei] ,描述第 i 件物品的类型、颜色以及名称。
另给你一条由两个字符串 ruleKey 和 ruleValue 表示的检索规则。
如果第 i 件物品能满足下述条件之一,则认为该物品与给定的检索规则 匹配 :
ruleKey == “type” 且 ruleValue == typei 。
ruleKey == “color” 且 ruleValue == colori 。
ruleKey == “name” 且 ruleValue == namei 。
统计并返回 匹配检索规则的物品数量 。
思路:
由于 ruleKey 只可能是 “type”、“color” 或 “name”,我们可以直接取 ruleKey 的第一个字符来确定 item 的下标 i。
然后遍历 items 数组,统计 item[i] == ruleValue 的个数即可。
代码:
class Solution {
public:
int countMatches(vector<vector<string>>& items, string ruleKey, string ruleValue) {
int i = ruleKey[0] == 't' ? 0 : (ruleKey[0] == 'c' ? 1 : 2);
return count_if(items.begin(), items.end(), [&](auto& v) { return v[i] == ruleValue; });
}
};