BestCoder Round #54 (div.2)HDU 5429 Geometric Progression(1003)

1003 ( HDU 5429 ) Geometric Progression

 Time Limit: 2000/1000 MS (Java/Others)

 

 Memory Limit: 65536/65536 K (Java/Others)

问题描述

判断一个数列是否为等比数列。

在数学中，等比数列，是一个数列，这个数列中的第一项之后的每一项是前一项乘上一个固定的非零实数（我们称之为公比）。比如，数列 2, 6, 18, 54, ... 是一个公比为3的等比数列。 类似的，10,5,2.5,1.25，...是一个公比为0.5的等比数列。
等比数列的一般形式是：
a,ar,ar^2,ar^3,ar^4,...a,ar,ar​2​​,ar​3​​,ar​4​​,...
其中r!=0，r为公比，a是首项（a可以是任何实数）

输入描述

第一行一个整数T，表示数据组数。$T\le 20$
对于每一个组，第一行一个整数$n(1\le n\le 100)$，接下来第二行$n$个数允许前导零的非负整数A_iA​i​​，表示数列。保证A_iA​i​​位数$\le 100$。

输出描述

对于每一个组，输出Yes或者No。

输入样例

4
1
0
3
1 1 1
3
1 4 2
5
16 8 4 2 1

输出样例

Yes
Yes
No
Yes

分析：判断是否为等比数列，可以检验对所有

$1<i<nA[i-1]\ast A[i+1]=A[i]\ast A[i]$ 

是否都成立。直接高精度也是资词的。

比较简单的方法是选择若干质数（保证乘积大于10^{200}10​200​​），

在模意义下检验。复杂度

$O(k\ast n)$。$k$表示选取的质数个数。

注意：要特判是不是全零 还有当n=1全都是Yes

/*
Pro.ID	5429	Exe.Time	358MS	
Exe.Memory  10228K	Code Len.970 B
*/
import java.util.*;
import java.math.*;
import java.nio.file.*;
import java.io.*;
public class Main
{
    public static void main(String args[])
    {
        BigInteger []a = new BigInteger[111];
        Scanner cin = new Scanner(System.in);
        int t = cin.nextInt();
        while(t-->0)
        {
            int n = cin.nextInt();
            int i,tmp = 0;
            for(i = 0; i < n; i++)
            {
                a[i] = cin.nextBigInteger();
                if(a[i].equals(BigInteger.ZERO)) tmp++; ///tmp有多少个零
            }
            int flag = 1;
            for(i = 1; i < n-1; ++i)
                if(!a[i].multiply(a[i]).equals(a[i-1].multiply(a[i+1])))
                {
                    flag = 0;   ///判断符不符合等比数列
                    break;
                }
            if(tmp!=0&&tmp != n)  flag = 0;     ///特断是不是全零
            if(flag==1) System.out.println("Yes");
            else System.out.println("No");
        }
    }
}