Number of Arithmetic Triplets LeetCode solution in Python

1
class Solution:
2
    def arithmeticTriplets(self, nums: List[int], diff: int) -> int:
3

4
        ans = 0
5
        n = len(nums)
6
        for i in range(n):
7
            if nums[i] + diff in nums and nums[i] + 2 * diff in nums:
8
                ans += 1
9

10
        return ans