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

题目链接：https://codeforces.com/problemset/problem/1520/D

解题思路： 题目要求j-i=aj-ai，变形可得aj-j=ai-i。即意为差值相同即可。由于map只会计入一次相同元素，此处用map计数不能更合适了。
对每个输入的元素，用map[ai-i]记录差值相等元素的个数。对每一个存在差值相同元素的ai来说，它能组合前面的任意一个元素。则ai的组合对数即为m[ai-i]，计算完ai的组合对数后需对map进行更新，将ai计入进去。因此用ans+=m[ai-i]++即可记录组合对数。

#include<iostream>
#include<unordered_map>
using namespace std;
unordered_map<int,int> m;
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		m.clear();
		int n, a;
		long long ans = 0;
		cin >> n;
		for (int i = 0; i < n; i++)
		{
			cin >> a;
			ans += m[a - i]++;
		}
		cout << ans << endl;
	}
	return 0;
}

标签：map,719,ai,++,Differences,Codeforces,cin,int,ans
来源： https://blog.csdn.net/qq_52930538/article/details/123608993