描述:给一个数组长度为n的数组ai,求有多少对(i,j)(1<=i<j<=j),使得ai+aj = i+j
题解:这里可以考虑使用unorder_map容器,也就是hash的思想来解决,将上述公式转换为ai-i = -(aj-j),在hash表中保存每一次输入的差值,可以每次保存ai-i的值在hash表中,这样只要查看对应的i-ai的值是否在其中出现,累加个数即可。
例如:4 3 5 1 2 9
第一次:1 – 4: hash[3]=1;
第二次:2 – 3: hash[1]=1;
第三次:3 – 5:hash[2]=1;
第四次:4 – 1: 此时hash[3]存在,累加到res中,同时设置hash[-3]=1;
第五次:5 – 2:此时hash[3]存在,在累加到res中,同时设置hash[-3]=2;
第六次:6 – 9:此时hash[-3]存在,累加到res中。
代码实现:
#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
// 记录之间的差值.
int main() {
unordered_map<int, int>dp;
int n;
cin >> n;
long long res = 0;
int m;
for(int i = 1;i<=n;i++){
cin >> m;
if(dp[i-m]) res += dp[i-m];
dp[m-i]++;
}
cout << res << endl;
return 0;
}