题解 | 谐距下标对

import java.util.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        int num = in.nextInt();
        int kk = 0;
        int[] arr = new int[num];
        while (in.hasNextInt() && kk < num) { // 注意 while 处理多个 case
            arr[kk] = in.nextInt();
            kk++;
        }
        // System.out.println(Arrays.toString(arr));
        long count = 0;
        Map<Integer, Long> map = new HashMap<>();
        for (int i = 0; i < num; i++) {
            int x = arr[i] - i;
            count += map.getOrDefault(x, 0L);
            map.put(x, map.getOrDefault(x, 0L) + 1);

            // for (int j = i + 1; j < num; j++) {
            //     // System.out.println(i + " " + j + " " + arr[i] + " " +arr[j]);
            //     if(arr[j] - arr[i] == j - i) {
            //         count++;
            //     }
            // }
        }
        System.out.println(count);
    }
}