数据结构与算法——经典题目(5)锻造兵器

本文介绍了一种算法问题,即帮助角色小明从若干锻造石中找出属性值差为特定值C的石块组合,以便成功锻造兵器。文章提供了一个使用三指针尺取法的C++实现示例。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

锻造兵器


题目描述

小明一共有 n 块锻造石,第 i 块锻造石的属性值为 a_{i} 。现在小明决定从这 n 块锻造石中任取两块来锻造兵器。通过周密计算,小明得出,只有当两块锻造石的属性值的差值等于 C,兵器才能锻造成功。

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

输入描述

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

接下来一行包含 n​ 个整数,分别表示 a_{1},a_{2},a_{3} ... a_{n}​​。

1 ≤ N ≤ 2×10^5​,∣a_{i}​∣ ≤ 10^4​,0 ≤ C ≤ 10^9​​。 

输出描述

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

输入输出样例

输入

6 3
8 4 5 7 7 4

输出

5

运行限制

  • 最大运行时间:1s
  • 最大运行内存:128M

/*
    三指针尺取法
*/

#include <bits/stdc++.h>
using namespace std;

const int N = 2e5 + 10;
int a[N];

int main () {
	int n, c;
	cin >> n >> c;
	for (int i = 1 ; i <= n ; i++){
		cin >> a[i];
    }
	sort(a + 1, a + 1 + n);
	long long ans = 0;  //答案数量超过int,需要用long long
	for (int i = 1, j = 1, k = 1; i <= n ; i++) {
		while (j <= n && a[j] - a[i] < c )
			j++;  //用j、k查找数字相同的区间
		while (k <= n && a[k] - a[i] <= c)
			k++;  //区间[j,k]内所有数字相同
		if (a[j] - a[i] == c && a[k - 1] - a[i] == c && k - 1 >= 1)
			ans += k - j;  //统计数对
	}
	cout << ans << endl;

	return 0;
}

  加油哦! 如有错误和需要改进完善之处,欢迎大家纠正指教。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值