题意:
给一个数组,求出数组中有多少对(arr[i] + arr[j] == i - j)
数组最大的大小为1e5,数组中的元素 1 <= a <= 1e9。//i,j表示在数组中的位置
思路:
首先暴力不用说了。然后我想了想看你要预处理标记一下,使得减少计算机操作的次数,以降低时间。然后,看了看这个等式,发现我要同时标记 i 和 j才有可能,但是这样时间复杂度是降不下来的。我们必须要找到只一次只用标记一个数的办法。
显而易见,我们可以把上面的等式转化一下,变成:
arr[i] - i == -1*arr[j] - j . 然后还要注意一点,只有当i大于j的时候才可能成立,因为数组俩个元素加起来不可能为负数。强调这点是因为,这个等式的顺序也是有严格要求的,详情代码解释
AC代码
#include <bits/stdc++.h>
using namespace std;
#define NewNode (ListNode *)malloc(sizeof(ListNode))
#define Mem(a,b) memset(a,b,sizeof(a))
const int N = 1e5 + 5;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const unsigned long long mod = 998244353;
const double II = 3.1415926535;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> piil;
map<int,int> mp;
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
ll ans = 0,n;
scanf("%lld",&n);
for(int i = 1;i <= n;i++)
{
ll a;//这时候的i这个数组中最大的,避免重复和错误的计算
//只有当i最大的时候,才能用上面那个公式arr[i] - i == -1 * arr[j] - j
scanf("%lld",&a);
ans += mp[a-i];
mp[-1*a-i]++;//过了这个数之后,肯定是比后面的数组小标小,所以标记的是-1*a-i
}
printf("%lld\n",ans);
//用c是因为这里的输入输出大,c的输出输入时间少
}