[编程题] 有趣的数字
时间限制:1秒
空间限制:32768K
小Q今天在上厕所时想到了这个问题:有n个数,两两组成二元组,相差最小的有多少对呢?相差最大呢?
输入描述:
输入包含多组测试数据。
对于每组测试数据:
N - 本组测试数据有n个数
a1,a2…an - 需要计算的数据
保证:
1<=N<=100000,0<=ai<=INT_MAX.
输出描述:
对于每组数据,输出两个数,第一个数表示差最小的对数,第二个数表示差最大的对数。
输入例子1:
6
45 12 45 32 5 6
输出例子1:
1 2
题目链接(牛客网)
思路:用简单的暴力时间复杂度是O(n^2),明显超时,正确方法是先排序,然后分多种情况,最后再用双指针统计个数。
代码实现有点复杂,有点绕,用伪代码大概描述一下
伪代码:
if(所有的数字都相等)
{
syso(n*(n-1)/2,n*(n-1)/2)
//这个可以理解吧,最大值最小值都是0,即从n个数中取两个数有多少种组合。
}
else
{
//求最小值
if(最小值为0)
{
将所有值相等的子序列的组合加起来(双指针操作)
//有点抽象,举个栗子,1 1 2 3 3 3 4 6 6 6,将分别将1,3,6的组合情况加起来就是了,也就是8种情况。
}
else
{
一个循环直接统计arr[i+1]-arr[i]==最小值的次数
//这是因为最小值出现的情况只能是两个相邻的元素之间,最小值不为0,所以不可能会出现其他情况。
}
}
代码:
import java.util.Arrays;
import java.util.Scanner;
public class Main
{
public static void main(String[] args) throws Exception
{
Scanner sc=new Scanner(System.in);
while(sc.hasNext())
{
int n=sc.nextInt();
int[] arr=new int[n];
for(int i=0;i<n;i++)
arr[i]=sc.nextInt();
Arrays.sort(arr);
if(arr[n-1]==arr[0])
System.out.println((n*(n-1))/2+" "+(n*(n-1))/2);
else
{
System.out.println(getMinCnt(arr)+" "+getMaxCnt(arr));
}
}
}
private static int getMinCnt(int[] arr)
{
int sum=0;
int cnt=1;
int min=Integer.MAX_VALUE;
for(int i=0;i<arr.length-1;i++)
min=Math.min(min, arr[i+1]-arr[i]);
if(min==0)
{
int i=0;
int j=1;
while(j<arr.length)
{
if(arr[i]==arr[j])
{
j++;
cnt++;
}
else
{
if(cnt>1)
{
i=j;
j++;
sum+=cnt*(cnt-1)/2;
cnt=1;
}
else
{
i++;j++;
}
}
}
sum+=cnt*(cnt-1)/2;
return sum;
}
else
{
for(int i=0;i<arr.length-1;i++)
if(arr[i+1]-arr[i]==min)
sum++;
return sum;
}
}
private static int getMaxCnt(int[] arr)
{
int cnt1=1;
int cnt2=1;
for(int i=1;i<arr.length;i++)
{
if(arr[i]==arr[0])
cnt1++;
else
break;
}
for(int i=arr.length-2;i>=0;i--)
{
if(arr[i]==arr[arr.length-1])
cnt2++;
else
break ;
}
return cnt1*cnt2;
}
}```