Codeforces Round #719 (Div. 3) D. Same Differences

一、算法分析

很经典的做法，把原来的$a_j-a_i=j-i$进行移项，变成$a_j-j=a_i-i$，从而生成一个新的数组$b_i=a_i-i$，对于新数组，只需要统计某个数$b_i$之前有多少个和其相等的数即可。

对于这种题目，要求就是要做得快，代码要一气呵成尽可能不要出错，要思考缜密。比如这道题的$b$数组元素下标就可能小于0，那么就要用到偏移量或者是用map数组。

二、代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#define LL long long
using namespace std;
const int N=400050;
const int base=N/2;                        //偏移量
int s[N];
int n;
LL res;
int main(){
    
    int T;
    scanf("%d",&T);
    while(T--){
        res=0;
        memset(s,0,sizeof(s));
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            int x;
            scanf("%d",&x);
            x+=base-i;
            res+=s[x];
            s[x]++;
        }
        printf("%lld\n",res);
    }
    
    
    
    return 0;
    
}