题目
小Q今天在上厕所时想到了这个问题:有n个数,两两组成二元组,差最小的有多少对呢?差最大呢?
输入描述
输入包含多组测试数据。
对于每组测试数据:
N - 本组测试数据有n个数
a1,a2…an - 需要计算的数据
保证:
1<=N<=100000,0<=ai<=INT_MAX.
输出描述
对于每组数据,输出两个数,第一个数表示差最小的对数,第二个数表示差最大的对数。
输入例子
6
45 12 45 32 5 6
输出例子
1 2
解答
import java.util.*;
public class Solution {
public static void main(String[] args) {
///输入处理
Scanner sc = new Scanner(System.in);
while (sc.hasNext()) {
int n = sc.nextInt();
int[] num = new int[n];
for (int i = 0; i < n; i++) {
num[i] = sc.nextInt();
}
//先对数组排序,方便找最大最小,也方便求每个数字的次数
Arrays.sort(num);
//如果所有的数字都相等,想想为什么要单独判断?
if (num[n - 1] == num[0]) {
int res = n * (n - 1) / 2;
System.out.println(res + " " + res);
continue; //continue返回,因为在while循环里面
//统计次数
}
//注意,此处用TreeMap,它能自动排序,因为后面
//求最大值时,需要用到
Map<Integer, Integer> map = new TreeMap<>();
for (int val : num) {
if (map.containsKey(val)) {
map.put(val, map.get(val) + 1);
} else {
map.put(val, 1);
}
}
//求最小值
int minCnt = 0;
if (map.size() == n) { //没有重复数字
int minNum = Integer.MAX_VALUE;
for (int i = 1; i < n; i++) { //当需要数组中相邻的数字比较时,尤其注意数组越界的情况
if (num[i] - num[i-1] < minNum) {
minNum = num[i] - num[i-1];
minCnt = 1;
} else if (num[i] - num[i-1] == minNum) {
minCnt ++;
}
// int tmp = num[i] - num[i-1];
// if (tmp == minNum) {
// minCnt ++;
// } else if (tmp < minNum) {
// minNum = tmp;
// minCnt = 1;
// }
}
} else {
for (int val : map.keySet()) {
int value = map.get(val);
if (value > 1) {
minCnt += (value * (value - 1)) / 2;
}
}
}
//求最大值
List<Integer> list = new ArrayList<>(map.keySet());
int max = map.get(list.get(0)) * map.get(list.get(list.size() - 1));
System.out.println(minCnt + " " + max);
}
}
}