求n个正整数中距离为k的数量

原创于 2018-03-25 12:06:18 发布 · 160 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#机试

机试专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种算法，用于解决给定一组正整数和一个距离k的情况下，计算数组中相距为k的数值对数量的问题。通过使用HashSet去除重复元素并进行排序，再采用双循环遍历的方法来实现这一目标。

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

题目大意：

输入一个正整数n, k, 以及n个正整数，求这n个正整数中距离为k的数量

如输入: 5 1 2 1 5 4 1

输出: 2

import java.util.Arrays;
import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		System.out.println("intput n ,k ,number:");
		String str = sc.nextLine();
		int n = Integer.parseInt(str.split(" ")[0]) ;
		int k = Integer.parseInt(str.split(" ")[1]);
		Set<Integer> set = new HashSet<Integer>(); 
	    for (int i=0; i<n; i++) { 
	      set.add(Integer.parseInt(str.split(" ")[i+2])); 
	    } 
	    Integer[] num = set.toArray(new Integer[1]);
		sc.close();
		Arrays.sort(num);
		int count = 0;
		for(int i=0;i<num.length-1;i++) {	
			for(int j = i+1;j<num.length && ((num[i]+k) >= num[j]);j++) {
				count += ((num[i]+k)==num[j])?1:0;
			}
		}
		System.out.println(count);
	}
}