[leetCode] 2367. Number of Arithmetic Triplets (Python)

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

https://leetcode.com/problems/number-of-arithmetic-triplets/

Number of Arithmetic Triplets - LeetCode

---

Arithmetic Triplets : 산술 삼중항 

 

값이 증가하는 정수를 담는 리스트 nums와 정수 diff가 주어진다.

nums[0] 부터 숫자 3개를 뽑는데

직전 숫자와의 차이가 diff와 같은 경우의 수를 반환하는 문제.

class Solution:
    def arithmeticTriplets(self, nums: List[int], diff: int) -> int:
        count = 0
        for num in nums:
            if num + diff in nums and num + diff * 2 in nums:
                count += 1
        return count

and 연산자와 *2로 간단하게 해결 가능했는데 복잡하게 짜다가 꼬여서 결국 discuss 보고 해결.

나중에 잊어버릴 때 쯤 다시 풀어봐야겠다.