Particle Accelerator II

warning: unused import: `std::cmp::min`
 --> input/code.rs:1:5
  |
1 | use std::cmp::min;
  |     ^^^^^^^^^^^^^
  |
  = note: `#[warn(unused_imports)]` on by default

warning: 1 warning emitted

use std::cmp::min;
use std::io::{self, Read};

macro_rules! input {
    ($it: expr) => {
        $it.next().unwrap().parse().unwrap()
    };
    ($it: expr, $T: ty) => {
        $it.next().unwrap().parse::<$T>().unwrap()
    };
}
/*
mod classes;
mod homework;

fn main() {
    // println!("{}", "-".repeat(20));
    classes::c24::task2();
    // homework::hw9::task1();
    // println!("{}", "-".repeat(20));
}
// */

fn main() {
    let mut buf = String::new();
    io::stdin().read_to_string(&mut buf).unwrap();

    let mut it = buf.split_whitespace();

    let n: usize = input!(it);

    let mut v: Vec<usize> = (0..n).into_iter().map(|_| input!(it)).collect();

    v.sort();

    let i = v.len()/2;


    let mut w1: usize = v[0];
    let mut w2: usize = v[i];

    let mut cost = 0;

    for x in 1..=(i -1) {
        cost += w1 * v[x];
        w1 += v[x];
        cost += w2 * v[i+x];
        w2 += v[i+x];
    }

    cost += w1 * w2;

    println!("{}", cost);
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: WRONG ANSWER

Test 3

Verdict: ACCEPTED

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: ACCEPTED

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: ACCEPTED

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED