def n_par(n):
if n==0:
return 0
else:
return n*(n+1)//2
def void():
n=int(input())
lst=list(map(int,input().split()))
dic={}
for i in range(n):
dic[lst[i]-i]=dic.get(lst[i]-i,0)+1
sm=0
for _,v in dic.items():
sm+=n_par(v-1)
print(sm)
void()
这个还是简单的,关键还是思路,算了解释不清看代码吧



京公网安备 11010502036488号