Submission #67316706

Source Code Expand

def solve_same(A):
    x = abs(A[0])
    p = A.count(x)
    m = A.count(-x)
    if abs(p - m) <= 1 or p == 0 or m == 0:
        print("Yes")
    else:
        print("No")

def solve():
    N = int(input())
    A = list(map(int, input().split()))
    s = set([abs(A[i]) for i in range(N)])
    if N <= 2:
        print("Yes")
    elif len(s) == 1:
        solve_same(A)
    else:
        A.sort(key=lambda x: abs(x))
        for i in range(N - 2):
            if A[i] * A[i + 2] != A[i + 1] ** 2:
                return print("No")
        print("Yes")


T = int(input())
for _ in range(T):
    solve()

Submission Info

Submission Time
Task D - Make Geometric Sequence
User mn_7545
Language Python (PyPy 3.10-v7.3.12)
Score 425
Code Size 628 Byte
Status AC
Exec Time 287 ms
Memory 144860 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 425 / 425
Status
AC × 1
AC × 32
Set Name Test Cases
Sample 00_sample_00.txt
All 00_sample_00.txt, 01_random_01.txt, 01_random_02.txt, 01_random_03.txt, 01_random_04.txt, 01_random_05.txt, 01_random_06.txt, 01_random_07.txt, 01_random_08.txt, 01_random_09.txt, 01_random_10.txt, 01_random_11.txt, 01_random_12.txt, 01_random_13.txt, 01_random_14.txt, 01_random_15.txt, 01_random_16.txt, 01_random_17.txt, 01_random_18.txt, 01_random_19.txt, 01_random_20.txt, 01_random_21.txt, 01_random_22.txt, 01_random_23.txt, 01_random_24.txt, 01_random_25.txt, 01_random_26.txt, 01_random_27.txt, 01_random_28.txt, 01_random_29.txt, 01_random_30.txt, 02_handmade_31.txt
Case Name Status Exec Time Memory
00_sample_00.txt AC 56 ms 76580 KiB
01_random_01.txt AC 174 ms 84744 KiB
01_random_02.txt AC 283 ms 128292 KiB
01_random_03.txt AC 287 ms 138192 KiB
01_random_04.txt AC 219 ms 84568 KiB
01_random_05.txt AC 235 ms 100724 KiB
01_random_06.txt AC 149 ms 84824 KiB
01_random_07.txt AC 83 ms 115852 KiB
01_random_08.txt AC 218 ms 84728 KiB
01_random_09.txt AC 220 ms 85292 KiB
01_random_10.txt AC 183 ms 85400 KiB
01_random_11.txt AC 260 ms 118384 KiB
01_random_12.txt AC 222 ms 85060 KiB
01_random_13.txt AC 220 ms 85292 KiB
01_random_14.txt AC 227 ms 85496 KiB
01_random_15.txt AC 225 ms 88952 KiB
01_random_16.txt AC 224 ms 85104 KiB
01_random_17.txt AC 221 ms 85268 KiB
01_random_18.txt AC 197 ms 84564 KiB
01_random_19.txt AC 88 ms 115568 KiB
01_random_20.txt AC 171 ms 85172 KiB
01_random_21.txt AC 263 ms 122876 KiB
01_random_22.txt AC 222 ms 84896 KiB
01_random_23.txt AC 188 ms 85564 KiB
01_random_24.txt AC 285 ms 144860 KiB
01_random_25.txt AC 223 ms 84584 KiB
01_random_26.txt AC 221 ms 84984 KiB
01_random_27.txt AC 209 ms 91308 KiB
01_random_28.txt AC 278 ms 128260 KiB
01_random_29.txt AC 217 ms 84820 KiB
01_random_30.txt AC 180 ms 89424 KiB
02_handmade_31.txt AC 56 ms 76240 KiB