結果

問題 No.1233 割り切れない気持ち
ユーザー manta1130manta1130
提出日時 2022-03-22 22:29:15
言語 Rust
(1.77.0)
結果
TLE  
実行時間 -
コード長 5,523 bytes
コンパイル時間 2,068 ms
コンパイル使用メモリ 189,624 KB
実行使用メモリ 16,064 KB
最終ジャッジ日時 2024-04-18 23:20:50
合計ジャッジ時間 9,078 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
13,880 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 1 ms
6,940 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 0 ms
6,944 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 AC 1 ms
6,940 KB
testcase_07 AC 6 ms
6,940 KB
testcase_08 AC 10 ms
6,940 KB
testcase_09 AC 21 ms
8,716 KB
testcase_10 AC 23 ms
8,544 KB
testcase_11 AC 5 ms
6,944 KB
testcase_12 AC 26 ms
10,412 KB
testcase_13 AC 25 ms
9,772 KB
testcase_14 AC 27 ms
9,124 KB
testcase_15 AC 26 ms
10,020 KB
testcase_16 AC 25 ms
9,252 KB
testcase_17 AC 6 ms
6,944 KB
testcase_18 AC 23 ms
9,248 KB
testcase_19 AC 12 ms
7,588 KB
testcase_20 AC 6 ms
6,940 KB
testcase_21 AC 8 ms
6,944 KB
testcase_22 AC 8 ms
6,944 KB
testcase_23 AC 7 ms
6,940 KB
testcase_24 AC 8 ms
6,944 KB
testcase_25 AC 7 ms
6,940 KB
testcase_26 AC 8 ms
6,944 KB
testcase_27 AC 179 ms
9,388 KB
testcase_28 AC 165 ms
7,800 KB
testcase_29 AC 146 ms
9,432 KB
testcase_30 AC 133 ms
7,844 KB
testcase_31 AC 125 ms
7,716 KB
testcase_32 AC 117 ms
7,588 KB
testcase_33 AC 107 ms
7,712 KB
testcase_34 AC 101 ms
7,468 KB
testcase_35 AC 99 ms
7,456 KB
testcase_36 AC 97 ms
7,456 KB
testcase_37 AC 1 ms
6,944 KB
testcase_38 TLE -
testcase_39 -- -
testcase_40 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable does not need to be mutable
  --> main.rs:11:9
   |
11 |     let mut a = input_vector::<usize>();
   |         ----^
   |         |
   |         help: remove this `mut`
   |
   = note: `#[warn(unused_mut)]` on by default

warning: variable does not need to be mutable
  --> main.rs:12:9
   |
12 |     let mut s = a.iter().sum::<usize>();
   |         ----^
   |         |
   |         help: remove this `mut`

warning: 2 warnings emitted

ソースコード

diff #

#[allow(unused_imports)]
use std::io::{stdout, BufWriter, Write};

fn main() {
    let out = stdout();
    let mut out = BufWriter::new(out.lock());

    inputv! {
        n:usize,
    }
    let mut a = input_vector::<usize>();
    let mut s = a.iter().sum::<usize>();

    let mut ans = n * s;
    let max = *a.iter().max().unwrap();
    let mut arr = vec![0; max * 2 + 1];
    for &i in &a {
        arr[i] += 1;
    }
    let sumarr = [0]
        .iter()
        .cloned()
        .chain(arr.iter().scan(0, |state, &x| {
            *state += x;
            Some(*state)
        }))
        .collect::<Vec<_>>();

    for i in a {
        let mut buf = 0;
        for j in 1.. {
            if (j + 1) * i > max * 2 + 1 {
                break;
            }
            buf += j * (sumarr[(j + 1) * i] - sumarr[j * i]);
            // dbg!(j, j * (sumarr[(j + 1) * i] - sumarr[j * i]));
        }

        //dbg!(i * buf);
        ans -= i * buf;
    }
    writeln!(out, "{}", ans).unwrap();
}

//https://github.com/manta1130/competitive-template-rs

use input::*;

pub mod input {
    use std::cell::RefCell;
    use std::io;
    pub const SPLIT_DELIMITER: char = ' ';
    pub use std::io::prelude::*;

    thread_local! {
        pub static INPUT_BUFFER:RefCell<std::collections::VecDeque<String>>=RefCell::new(std::collections::VecDeque::new());
    }

    #[macro_export]
    macro_rules! input_internal {
        ($x:ident : $t:ty) => {
            INPUT_BUFFER.with(|p| {
                while p.borrow().len() == 0 {
                    let temp_str = input_line_str();
                    let mut split_result_iter = temp_str
                        .split(SPLIT_DELIMITER)
                        .map(|q| q.to_string())
                        .filter(|q| q.len() > 0)
                        .collect::<std::collections::VecDeque<_>>();
                    p.borrow_mut().append(&mut split_result_iter)
                }
            });
            let mut buf_split_result = String::new();
            INPUT_BUFFER.with(|p| buf_split_result = p.borrow_mut().pop_front().unwrap());
            let $x: $t = buf_split_result.parse().unwrap();
        };
        (mut $x:ident : $t:ty) => {
            INPUT_BUFFER.with(|p| {
                while p.borrow().len() == 0 {
                    let temp_str = input_line_str();
                    let mut split_result_iter = temp_str
                        .split(SPLIT_DELIMITER)
                        .map(|q| q.to_string())
                        .filter(|q| q.len() > 0)
                        .collect::<std::collections::VecDeque<_>>();
                    p.borrow_mut().append(&mut split_result_iter)
                }
            });
            let mut buf_split_result = String::new();
            INPUT_BUFFER.with(|p| buf_split_result = p.borrow_mut().pop_front().unwrap());
            let mut $x: $t = buf_split_result.parse().unwrap();
        };
    }

    pub fn input_buffer_is_empty() -> bool {
        let mut empty = false;
        INPUT_BUFFER.with(|p| {
            if p.borrow().len() == 0 {
                empty = true;
            }
        });
        empty
    }

    #[macro_export]
    macro_rules! inputv {
    ($i:ident : $t:ty) => {
        input_internal!{$i : $t}
    };
    (mut $i:ident : $t:ty) => {
        input_internal!{mut $i : $t}
    };
    ($i:ident : $t:ty $(,)*) => {
            input_internal!{$i : $t}
    };
    (mut $i:ident : $t:ty $(,)*) => {
            input_internal!{mut $i : $t}
    };
    (mut $i:ident : $t:ty,$($q:tt)*) => {
            input_internal!{mut $i : $t}
            inputv!{$($q)*}
    };
    ($i:ident : $t:ty,$($q:tt)*) => {
            input_internal!{$i : $t}
            inputv!{$($q)*}
    };
}

    pub fn input_all() {
        INPUT_BUFFER.with(|p| {
            if p.borrow().len() == 0 {
                let mut temp_str = String::new();
                std::io::stdin().read_to_string(&mut temp_str).unwrap();
                let mut split_result_iter = temp_str
                    .split_whitespace()
                    .map(|q| q.to_string())
                    .collect::<std::collections::VecDeque<_>>();
                p.borrow_mut().append(&mut split_result_iter)
            }
        });
    }

    pub fn input_line_str() -> String {
        let mut s = String::new();
        io::stdin().read_line(&mut s).unwrap();
        s.trim().to_string()
    }

    #[allow(clippy::match_wild_err_arm)]
    pub fn input_vector<T>() -> Vec<T>
    where
        T: std::str::FromStr,
    {
        let mut v: Vec<T> = Vec::new();

        let s = input_line_str();
        let split_result = s.split(SPLIT_DELIMITER);
        for z in split_result {
            let buf = match z.parse() {
                Ok(r) => r,
                Err(_) => panic!("Parse Error",),
            };
            v.push(buf);
        }
        v
    }

    #[allow(clippy::match_wild_err_arm)]
    pub fn input_vector_row<T>(n: usize) -> Vec<T>
    where
        T: std::str::FromStr,
    {
        let mut v = Vec::with_capacity(n);
        for _ in 0..n {
            let buf = match input_line_str().parse() {
                Ok(r) => r,
                Err(_) => panic!("Parse Error",),
            };
            v.push(buf);
        }
        v
    }

    pub trait ToCharVec {
        fn to_charvec(&self) -> Vec<char>;
    }

    impl ToCharVec for String {
        fn to_charvec(&self) -> Vec<char> {
            self.to_string().chars().collect::<Vec<_>>()
        }
    }
}
0