[leetCode] 1773. Count Items Matching a Rule (Python)

▼ 문제 바로가기 (링크) ▼

https://leetcode.com/problems/count-items-matching-a-rule/

Count Items Matching a Rule - LeetCode

---

3개 원소를 가지는 리스트를 1개 이상 가지는 중첩 리스트items가 주어진다.

원소는 순서대로 type, color, name을 나타낸다.

ruleKey와 ruleValue가 주어졌을 때,

맞는 값이 items에 몇 개 있는 지 반환하는 문제.

class Solution:
    def countMatches(self, items: List[List[str]], ruleKey: str, ruleValue: str) -> int:
        cnt = 0
        for i in items:
            if ruleKey == "type":
                if i[0] == ruleValue:
                    cnt += 1
            elif ruleKey == "color":
                if i[1] == ruleValue:
                    cnt += 1
            else:
                if i[2] == ruleValue:
                    cnt += 1
        return cnt

구하려는 값이 있는지만 확인하면 되니 주어진 key값에 따라 분기처리했다.