15-下标数组

描述:给一个数组长度为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;
}
posted @ 2024-07-16 23:18  alone_qing  阅读(18)  评论(0)    收藏  举报