锻造兵器（尺取法、三指针）

锻造兵器（尺取法、三指针）

题目描述

小明一共有 n 块锻造石，第 i 块锻造石的属性值为 a_i 。

现在小明决定从这 n 块锻造石中任取两块来锻造兵器。

通过周密计算，小明得出，只有当两块锻造石的属性值的差值等于 C，兵器才能锻造成功。

请你帮小明算算，他有多少种选取锻造石的方案可以使得锻造成功。

输入描述

第一行包含两个整数 n,C，其含义如题所述。

接下来一行包含 n 个整数，分别表示 *a₁,*a₂,⋯,a_n。

1≤N≤2×10⁵，|a_i|≤10⁴，0≤C*≤10⁹。

输出描述

输出共一行，包含一个整数，表示答案。

输入输出样例

示例 1

输入

6 3
8 4 5 7 7 4

输出

5

思路：

先用一个数组存一串数字，次数ans定义为long long，不然会溢出，先将数组进行从小到大排序。因为数组中可能会出现重复数字，所以定义三个指针i、j、k。i就是把数组遍历一遍。j要找到第一个与i相差为c的数。k要找到前一个数与i相差c，也就是a[k-1]与a[i]相差为c，最后判断a[j]与a[i]是否相差c和a[k-1]与a[i]是否相差c且k-1不超出数组，注意最后的次数要是k与j的差，因为 k 与 j 的差是有多少相同的数，而这些相同的数与a[i]都是相差c的所以有k-j种可能

代码：

#include<bits/stdc++.h>
using namespace std;
int a[2000000]; 
int main()
{
	int n,c;
	long long ans=0;
	scanf("%d %d",&n,&c);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	sort(a+1,a+1+n);  //从小到大排序 
	for(int i=1,j=1,k=1;i<=n;i++)
	{
		while(j<=n&&a[j]-a[i]<c)
			j++; //j判断到第一个相差为c的数 
		while(k<=n&&a[k]-a[i]<=c)
			k++;  //k判断到前一个数相差为c的数 
		if(a[j]-a[i]==c&&a[k-1]-a[i]==c&&k-1>=1) 
//如果j和i相差为c，k-1与i相差为c，且k-1不超出数组 
			ans+=k-j; //k-j为多少个相同的数 
	}
	printf("%lld",ans); 
	return 0;
}