2021-05-08

Codeforces Round #720 (Div. 2) D. Same Differences

传送门

题目大意：给你n个数，让你找出其中符合i<j且aj-ai=j-i的对数。

思路：本来想的是暴力每个点然后遍历求符合条件的最后将结果加起来，但是这样就一定会TLE，然后看了别人的题解（大佬就是大佬），再稍加思索变得出了一个比较巧妙的方法：aj-ai=j-i，经过变形可得 aj-j=ai-i，这个时候我们只需要求出 ai-i出现的次数然后-1就可以了（因为题目要求的是对数）；

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<cmath>
#include<map> //用map记录 ai-i出现的次数
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
map<int ,int>p;
int main()
{
	int t;
	cin>>t;
	while(t--){
		int n;
		p.clear();
		cin>>n;
		long long ans=0;
		for(int i=1;i<=n;i++){
			int x;
			cin>>x;
			ans+=p[x-i];
			p[x-i]++;
		}
		
		cout<<ans<<'\n';
	}
}
 

遇到问题还是要先好好思考一下，多去想想其它办法，而不是一味的去暴力解决问题。