解法一——双指针
首先我们可以先将原数组进行排序,由于题目要求每个A对应有多少个B使得等式A-B=C成立,我们可以将这个式子换一种形式,改为B=A+C,C为定值,那么确定了A就确定了B的值,因此可以枚举数组中每一个元素的值找到使等式成立的值,若用双重循环的话时间复杂度大为O(n*2),但是由于我们将数组进行排序,对于每一个A能够使等式成立的B一定在一个连续的区间内,此时我们可以用双指针来确定这个区域
int l = 0, r = 0; while(l < n && a[l] < a[i] + c) l++; //将l指针移动到满足要求的B的前一位 while(r < n && a[r] <= a[i] + c)r++; //将r指针移动到满足要求的最后一个B
那么在这个区间内所有元素都满足题目的要求,ans += r - l
解法二——map暴力
由于数组中的数字毫无关联,但是要找对应数学关系的树的个数,这个时候就可以考虑STL库中的map类,定义一个map数组,将这些数作为下标存入,将对应数字出现的次数作为这个数组中存的数,那么就可以直接枚举数组中的每一个数,通过这些数字,找到符合要求的B并在数组中找到下标为B所对应元素的出现次数,那么出现次数之和就是所有满足题目条件的总个数
for (int i = 0; i < n; i++) { cin >> a[i]; ds[a[i]]++; }