본문 바로가기
👩‍💻Developer/Leetcode

[Easy][Python] Count Items Matching a Rule

by 블루누들 2022. 10. 22.

 items라는 array가 주어지며 각  items[i] 는 [type_i, color_i, name_i] 의 형태로 ith item의 type, color 그리고 item를 나타낸다. 또한 ruleKey 와 ruleValue라는 두 string형태의 rule이 주어진다.  ith item이 아래 중 한가지와 매치할 경우 룰에 맞는다고 가정한다. 

  • ruleKey == "type" and ruleValue == type_i.
  • ruleKey == "color" and ruleValue == color_i.
  • ruleKey == "name" and ruleValue == name_i

 

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

댓글