Snake mall

#![allow(unused)]

use core::num;
use std::io;

fn take_int() -> usize {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    return input.trim().parse::<usize>().unwrap();
}

fn take_vector() -> Vec<i128> {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let arr: Vec<i128> = input
        .trim()
        .split_whitespace()
        .map(|x| x.parse::<i128>().unwrap())
        .collect();
    return arr;
}

fn main() {
    let mut num_shops: usize = take_int();

    let mut spaces: Vec<i128> = take_vector();
    let mut order: Vec<i128> = Vec::new();
    for i in 0..num_shops {
        order.push(spaces[i]);
    }
    let mut shop_entries: Vec<i128> = vec![-10000000; 1000];
    spaces.sort_unstable();

    let mut sum_distance: i128 = 0;
    let mut right_distance: i128 = 0;
    let mut left_distance: i128 = 0;

    for i in 0..num_shops {
        let t_r = right_distance + spaces[i];
        let t_l = left_distance + spaces[i];

        if t_r < t_l {
            // better to move store to the right side of door
            sum_distance += right_distance;
            let index = spaces[i] as usize;
            shop_entries[index] = right_distance;
            shop_entries[index] = 0 - spaces[i] - right_distance;
            right_distance += spaces[i];
        } else {
            sum_distance += left_distance;
            let index = spaces[i] as usize;
            shop_entries[index] = left_distance;
            left_distance += spaces[i];
        }
    }

    let sum_float = sum_distance as f64;
    let num_float = num_shops as f64;

    println!("{}", (sum_float / num_float));

    for i in 0..order.len() {
        let index = order[i] as usize;
        print!("{} ", shop_entries[index]);
    }
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: ACCEPTED

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: WRONG ANSWER

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: WRONG ANSWER

Test 14

Verdict: WRONG ANSWER

Test 15

Verdict: WRONG ANSWER

Test 16

Verdict: ACCEPTED

Test 17

Verdict: WRONG ANSWER

Test 18

Verdict: WRONG ANSWER

Test 19

Verdict: WRONG ANSWER

Test 20

Verdict: WRONG ANSWER

Test 21

Verdict: WRONG ANSWER

Test 22

Verdict: WRONG ANSWER

Test 23

Verdict: WRONG ANSWER

Test 24

Verdict: WRONG ANSWER

Test 25

Verdict: WRONG ANSWER

Test 26

Verdict: WRONG ANSWER

Test 27

Verdict: WRONG ANSWER

Test 28

Verdict: WRONG ANSWER

Test 29

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1060
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 30

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1033
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 31

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1070
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 32

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1214
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 33

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1082
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 34

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1110
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 35

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1077
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 36

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1005
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 37

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1139
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 38

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1045
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 39

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1021
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 40

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1024
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 41

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1070
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 42

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1002
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 43

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1004
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 44

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1009
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 45

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1077
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 46

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1005
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 47

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1004
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 48

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1031
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 49

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1001
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 50

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1020
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 51

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 52

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1002
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 53

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1004
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 54

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1009
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 55

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1044
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 56

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1005
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 57

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1004
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 58

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 59

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 60

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 61

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 62

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 63

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 64

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 65

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 66

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 67

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 68

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 69

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 70

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 71

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:46:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 72

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:52:25:
index out of bounds: the len is 1000 but the index is 1000
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace