Submission #67374070

Source Code Expand

use itertools::iproduct;
use itertools::Itertools;
use proconio::input;

fn main() {
    input! {
        q: usize,
    }
    for _ in 0..q {
        input! {
            n: usize,
            mut a: [i64; n],
        }
        a.sort_unstable();
        let mut b = a.iter().map(|&x| x.abs()).collect::<Vec<_>>();
        b.sort_unstable();
        let ans = b.iter().tuple_windows().all(|(x, y, z)| x * z == y * y)
            && iproduct!([1, -1], [1, -1]).any(|(s, t)| {
                let mut b = b
                    .iter()
                    .zip(std::iter::successors(Some(1), |&x| Some(t * x)))
                    .map(|(x, t)| s * t * x)
                    .collect::<Vec<_>>();
                b.sort_unstable();
                a == b
            });
        println!("{}", if ans { "Yes" } else { "No" });
    }
}

Submission Info

Submission Time
Task D - Make Geometric Sequence
User ngtkana
Language Rust (rustc 1.70.0)
Score 425
Code Size 862 Byte
Status AC
Exec Time 37 ms
Memory 8708 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 1 ms 1800 KiB
01_random_01.txt AC 17 ms 2892 KiB
01_random_02.txt AC 25 ms 5840 KiB
01_random_03.txt AC 22 ms 6232 KiB
01_random_04.txt AC 36 ms 4140 KiB
01_random_05.txt AC 31 ms 4704 KiB
01_random_06.txt AC 10 ms 2368 KiB
01_random_07.txt AC 12 ms 8708 KiB
01_random_08.txt AC 36 ms 4024 KiB
01_random_09.txt AC 36 ms 3972 KiB
01_random_10.txt AC 21 ms 3172 KiB
01_random_11.txt AC 26 ms 5580 KiB
01_random_12.txt AC 36 ms 3960 KiB
01_random_13.txt AC 36 ms 4128 KiB
01_random_14.txt AC 36 ms 4004 KiB
01_random_15.txt AC 35 ms 4296 KiB
01_random_16.txt AC 36 ms 3996 KiB
01_random_17.txt AC 37 ms 3988 KiB
01_random_18.txt AC 27 ms 3404 KiB
01_random_19.txt AC 14 ms 8652 KiB
01_random_20.txt AC 17 ms 3020 KiB
01_random_21.txt AC 26 ms 5832 KiB
01_random_22.txt AC 36 ms 3984 KiB
01_random_23.txt AC 22 ms 3196 KiB
01_random_24.txt AC 21 ms 6580 KiB
01_random_25.txt AC 36 ms 4028 KiB
01_random_26.txt AC 36 ms 3948 KiB
01_random_27.txt AC 28 ms 3900 KiB
01_random_28.txt AC 26 ms 5880 KiB
01_random_29.txt AC 36 ms 4020 KiB
01_random_30.txt AC 16 ms 3096 KiB
02_handmade_31.txt AC 1 ms 2016 KiB