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;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值