题目:
数学老师给小明出了一道等差数列求和的题目。但是粗心的小明忘记了一 部分的数列,只记得其中 N 个整数。
现在给出这 N 个整数,小明想知道包含这 N 个整数的最短的等差数列有几项?
输入描述
输入的第一行包含一个整数 N。
第二行包含 N 个整数A1,A2,⋅⋅⋅,AN。(注意A1 ∼ AN 并不一定是按等差数列中的顺序给出)
其中,2≤N≤10^5,0≤Ai≤10^9。
输出描述
输出一个整数表示答案。
输入输出样例
示例
输入
5
2 6 4 10 20
输出
10
样例说明: 包含 2、6、4、10、20 的最短的等差数列是 2、4、6、8、10、12、14、16、 18、20。
运行限制
最大运行时间:1s
最大运行内存: 256M
代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
| import java.util.Arrays; import java.util.Scanner; public class Main { public static void main(String[] args) { Scanner sc=new Scanner(System.in); int n; n=sc.nextInt(); int[] A=new int[n]; for (int i=0;i<n;i++){ A[i]=sc.nextInt(); } Arrays.sort(A); int min=Integer.MAX_VALUE; for (int i=0;i<n-1;i++){ min=Integer.min(min,A[i+1]-A[i]); } if(min==0){ System.out.println(n); }else{ int a=0,c=0; for (int i=n-1;i>=0;i--) { if ((A[i] - A[0]) % min != 0) { int[] b=new int[min]; c=min; for (int x=2;x<=min/2;x++){ if (min%x==0){ b[x]=x; } } for (int y=0;y<min;y++) { for (int j = n - 1; j >= 0; j--) { if (b[y]!=0&&(A[j] - A[0]) % b[y] == 0){ c=b[y]; } } if (c==min){ a=A[n - 1] - A[0]+1; } else{ a = (A[n - 1] - A[0]) / c + 1; } } break; } } if (a==0) { a = (A[n - 1] - A[0]) / min + 1; } System.out.println(a); } } }
|
思路:
因为无序所以先对其进行排序,用java自带的sort方法进行升序排序,然后找最小差值为暂时公差,如果最小差值是0,则为特殊情况1(n个整数都一样)直接输出n。
再利用等差数列定义判断min是否为公差:任意一个数Ai-最小的那个数A0)%(问题二求得的最小公差d)=0则为公差,
这里会出现第二种特殊情况:如果两个不同的公差有公约数就会算错,比如输入数列5 15 30,输出的结果是26而不是6。
所以我们将min的全部约数求出来再进行一次判断如果有约数可以满足公差的要求,则最大满足要求的约数为新的公差进行计算。
a = (A[n - 1] - A[0]) / c + 1;
如果没有满足要求的约数则公差为1,
a=A[n - 1] - A[0]+1;
如果min满足等差数列的定义,则min为真正的公差。
a = (A[n - 1] - A[0]) / min + 1;