第一行输入整数 。第二行输入 个整数 。
输出一个整数,表示谐距下标对数量。
6 1 2 3 4 5 6
15
from collections import Counter n = int(input()) numbers = list(map(int, input().split())) b = [numbers[i] - (i+1) for i in range(n)] counter = Counter(b) count = 0 for freq in counter.values(): if freq >1: count += freq*(freq - 1) //2 print(count)
这道题你会答吗?花几分钟告诉大家答案吧!
扫描二维码,关注牛客网
下载牛客APP,随时随地刷题