Submission #67316562

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from collections import Counter
T=int(input())
for _ in range(T):
    N=int(input())
    A=list(map(int, input().split()))
    if N <= 2:
        print("Yes")
        continue
    B=sorted(abs(a) for a in A)
    ok=True
    for i in range(N-2):
        if B[i+1] * B[i+1] != B[i] * B[i+2]:
            ok=False
            break
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

from collections import Counter

T=int(input())
for _ in range(T):
    N=int(input())
    A=list(map(int, input().split()))


    if N <= 2:
        print("Yes")
        continue


    B=sorted(abs(a) for a in A)


    ok=True
    for i in range(N-2):
        if B[i+1] * B[i+1] != B[i] * B[i+2]:
            ok=False
            break
    if not ok:
        print("No")
        continue


    cntA=Counter(A)
    f=False


    for b in (B, B[::-1]):
        for s in (1, -1):
            cand=[s*v for v in b]
            if Counter(cand)==cntA:
                f=True
                break
        if f:
            break

        for st in (1, -1):
            cand=[]
            s=st
            for v in b:
                cand.append(s*v)
                s*=-1
            if Counter(cand)==cntA:
                f= True
                break
        if f:
            break

    print("Yes" if f else "No")

Submission Info

Judge Result