蓝桥杯第1374题——锻造兵器

Ares陌冄

已于 2024-02-22 12:38:31 修改

阅读量650

点赞数 8

分类专栏： java算法学习笔记文章标签：算法

于 2024-02-22 12:37:33 首次发布

本文链接：https://blog.csdn.net/Ares_moran/article/details/136229699

版权

java算法学习笔记专栏收录该内容

26 篇文章 2 订阅

订阅专栏

本文提供了Java代码，展示了如何通过双指针策略计算给定锻造石属性值满足差值条件的组合数量，适用于编程竞赛或算法学习

摘要由CSDN通过智能技术生成

题目描述

小明一共有n块锻造石，第块锻造石的属性值为ai.
现在小明决定从这n块锻造石中任取两块来锻造兵器
通过周密计算，小明得出，只有当两块锻造石的属性值的差值等于C，兵器才能锻造成功
请你帮小明算算，他有多少种选取锻造石的方案可以使得锻造成功

输入描述

第一行包含两个整数n，C，其含义如题所述
接下来一行包含n个整数，分别表示a1，a2，··，an.
1 < N < 2 x 10^5，|ai| < 10^4，0 < C < 10^9

输出描述

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

输入输出样例

6 3

8 4 5 7 7 4

5

解题思路

这个题是一道典型的双指针题，要控制快指针和慢指针所对应的数据之差为C。

首先，使用排序方法对输入数据进行排序是必要的。

然后快指针优先移动，直到快慢指针数据之差至少为C；接下来慢指针进行移动，如果不是C（就一定比C大）就往后移动，直到快慢指针之差至多为C。

到此就是一组快慢指针的移动，此时可以判断快慢指针数据之差是否为C，如果是，就对ans做更新。这题的关键是ans更新多少，我们思考后不难发现，按照题目提供的例子，两个7的石头可以与两个4的石头分别组成一组，这意味着组合数量是满足要求的两个数值的石头的个数乘积，那么我们只需要对快慢指针分别派生出一个新指针，分别向后步进进行计数即可。

下面给出代码：

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.lang.reflect.Array;
import java.math.BigInteger;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {
        Scanner sc = new Scanner(System.in);
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        String[] temp = in.readLine().split(" ");
        int n = Integer.parseInt(temp[0]);
        int c = Integer.parseInt(temp[1]);
        temp = in.readLine().split(" ");
        int[] data = new int[n];
        for (int i = 0; i < n; i++) {
            data[i] = Integer.parseInt(temp[i]);
        }
        Arrays.sort(data);

        int slow = 0, fast = 0;
        long ans = 0;
        while (fast < n) {
            while (fast < n && data[fast] - data[slow] < c) {
                fast++;
            }
            while (fast < n && data[fast] - data[slow] > c) {
                slow++;
            }

            if (fast < n && data[fast] - data[slow] == c) {
                int oldFast = fast, oldSlow = slow;
                while (fast < n && data[fast] == data[oldFast]) {
                    fast++;
                }
                while (slow < n && data[slow] == data[oldSlow]) {
                    slow++;
                }
                ans += (long) (fast - oldFast) * (slow - oldSlow);
            }
        }
        System.out.println(ans);
    }
}