Number of Arithmetic Triplets LeetCode solution in Java

1
class Solution {
2
  public int arithmeticTriplets(int[] nums, int diff) {
3
    int result = 0;
4
    int[] map = new int[201];
5

6
    for (int num : nums) {
7
      map[num] = 1;
8

9
      if (num - diff >= 0) {
10
        map[num] += map[num - diff];
11
      }
12

13
      if (map[num] >= 3) result += 1;
14
    }
15

16
    return result;
17
  }
18
}