[PYTHON] list 에서 for 문 2개 돌리기 (i, j) (leetcode: number of good pairs)

문제

 

nums 라는 list 가 주어진다. 만약 nums[i] == nums[j] 이고 i < j 라면 good pair 이고, 총 good pair 의 개수를 return 한다. 

 

풀이

 

주어진 list 안에서 i, j 의 for 문을 돌리면서 good pair 에 해당하는지 확인하고 해당하면 답에 + 1 해 나가면서 전체 list 를 확인 하면 된다. 다만 i 의 범위는 0부터 len(nums) 까지이고, j 의 범위는 i를 포함하면 안되기 때문에 i+1 부터 len(nums) 까지 이다. 

 

class Solution:
    def numIdenticalPairs(self, nums: List[int]) -> int:

        solNum = 0

        for i in range(len(nums)):

            for j in range(i + 1, len(nums)):

                if(nums[i] == nums[j]):

                    solNum += 1

        return solNum

 

그리 어렵지 않은 문제이다.