結果

問題 No.1690 Power Grid
コンテスト
ユーザー emak
提出日時 2021-09-24 23:06:21
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 158 ms / 3,000 ms
コード長 3,591 bytes
コンパイル時間 14,788 ms
コンパイル使用メモリ 379,652 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-05 11:13:39
合計ジャッジ時間 15,229 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#![allow(unused_macros, unused_imports)]
macro_rules! dbg {
    ($($xs:expr),+) => {
        if cfg!(debug_assertions) {
            std::dbg!($($xs),+)
        } else {
            ($($xs),+)
        }
    }
}

fn main() {
    input!{
        n: usize,
        m: usize,
        k: u32,
        a: [u64; n],
        es: [(Usize1, Usize1, u64); m],
    }

    let inf = 1u64 << 60;
    let mut dist = vec![vec![inf; n]; n];
    for (x, y, z) in es {
        dist[x][y] = z;
        dist[y][x] = z;
    }
    for i in 0..n {
        dist[i][i] = 0;
    }

    for k in 0..n {
        for i in 0..n {
            for j in 0..n {
                dist[i][j] = dist[i][j].min(dist[i][k] + dist[k][j]);
            }
        }
    }

    let mut dp = vec![inf; 1<<n];

    for i in 0..n {
        dp[1<<i] = a[i];
    }

    for s in 1..1<<n {
        for i in 0..n {
            if s>>i&1 == 1 {
                continue;
            }

            let mut k = None;

            for j in 0..n {
                if s>>j&1 == 0 {
                    continue;
                }

                if let Some(kk) = k {
                    if dist[i][j] < dist[i][kk] {
                        k = Some(j);
                    }
                } else {
                    k = Some(j);
                }
            }

            if let Some(k) = k {
                dp[s|1<<i] = dp[s|1<<i].min(dp[s] + a[i] + dist[i][k]);
            }
        }
    }

    let mut ans = inf;    
    for s in 0usize..1<<n {
        if s.count_ones() == k {
            ans = ans.min(dp[s]);
        }
    }

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

pub mod proconio {
    #[macro_export]
    macro_rules! input{
        (sc=$sc:expr,$($r:tt)*)=>{
            input_inner!{$sc,$($r)*}
        };
        ($($r:tt)*)=>{
            let mut sc=proconio::Scanner::new(std::io::stdin().lock());
            input_inner!{sc,$($r)*}
        };
    }

    #[macro_export]
    macro_rules! input_inner{
        ($sc:expr)=>{};
        ($sc:expr,)=>{};
        ($sc:expr,$var:ident:$t:tt$($r:tt)*)=>{
            let $var=read_value!($sc,$t);
            input_inner!{$sc $($r)*}
        };
    }

    #[macro_export]
    macro_rules! read_value{
        ($sc:expr,($($t:tt),*))=>{
            ($(read_value!($sc,$t)),*)
        };
        ($sc:expr,[$t:tt;$len:expr])=>{
            (0..$len).map(|_|read_value!($sc,$t)).collect::<Vec<_>>()
        };
        ($sc:expr,Chars)=>{read_value!($sc,String).chars().collect::<Vec<char>>()};
        ($sc:expr,Usize1)=>{read_value!($sc,usize)-1};
        ($sc:expr,$t:ty)=>{$sc.next::<$t>()};
    }
    pub struct Scanner {
        s: Box<str>,
        input: std::iter::Peekable<std::str::SplitAsciiWhitespace<'static>>,
    }
    impl Scanner {
        pub fn new<R: std::io::Read>(mut reader: R) -> Self {
            let s = {
                let mut s = String::new();
                reader.read_to_string(&mut s).unwrap();
                s.into_boxed_str()
            };
            let mut sc = Scanner {
                s,
                input: "".split_ascii_whitespace().peekable(),
            };
            use std::mem;
            let s: &'static str = unsafe { mem::transmute(&*sc.s) };
            sc.input = s.split_ascii_whitespace().peekable();
            sc
        }
        #[inline]
        pub fn next<T: std::str::FromStr>(&mut self) -> T
        where
            T::Err: std::fmt::Debug,
        {
            self.input
                .next()
                .unwrap()
                .parse::<T>()
                .expect("Parse error")
        }
    }
}
0