結果

問題 No.1538 引きこもりさんは引き算が得意。
コンテスト
ユーザー Yukino DX.
提出日時 2024-02-14 18:49:34
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 26 ms / 2,000 ms
コード長 2,587 bytes
コンパイル時間 11,191 ms
コンパイル使用メモリ 402,588 KB
実行使用メモリ 8,492 KB
最終ジャッジ日時 2024-09-28 18:58:54
合計ジャッジ時間 12,891 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

diff # 

#![allow(non_snake_case)]
#![allow(unused_imports)]
//use itertools::Itertools;
//use proconio::input;
//use proconio::marker::*;
use std::cmp::Reverse;
use std::collections::*;

fn main() {
    input! {
        n:usize,
        k:i64,
        a:[i64;n],
    }

    let mut ans = a.contains(&k);
    let mut hs0 = HashSet::new();
    let mut hs1 = HashSet::new();
    f(0, 0, &a[..n / 2], false, false, &mut hs0);
    f(0, 0, &a[n / 2..], false, false, &mut hs1);
    ans |= hs0.contains(&(k, true, true)) || hs1.contains(&(k, true, true));
    for &(x, b0, b1) in hs0.iter() {
        for c0 in [false, true] {
            for c1 in [false, true] {
                if hs1.contains(&(k - x, c0, c1)) && (b0 || c0) && (b1 || c1) {
                    ans = true;
                }
            }
        }
    }

    println!("{}", if ans { "Yes" } else { "No" });
}

fn f(id: usize, x: i64, a: &[i64], plus: bool, minus: bool, hs: &mut HashSet<(i64, bool, bool)>) {
    if id == a.len() {
        hs.insert((x, plus, minus));
        return;
    }

    f(id + 1, x + a[id], a, true, minus, hs);
    f(id + 1, x - a[id], a, plus, true, hs);
    f(id + 1, x, a, plus, minus, hs);
}
mod input {
    #[macro_export]
    macro_rules! input {
        (source = $s:expr, $($r:tt)*) => {
            let mut iter = $s.split_whitespace();
            input_inner!{iter, $($r)*}
        };
        ($($r:tt)*) => {
            let s = {
                use std::io::Read;
                let mut s = String::new();
                std::io::stdin().read_to_string(&mut s).unwrap();
                s
            };
            let mut iter = s.split_whitespace();
            input_inner!{iter, $($r)*}
        };
    }

    #[macro_export]
    macro_rules! input_inner {
        ($iter:expr) => {};
        ($iter:expr, ) => {};

        ($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
            let $var = read_value!($iter, $t);
            input_inner!{$iter $($r)*}
        };
    }

    #[macro_export]
    macro_rules! read_value {
        ($iter:expr, ( $($t:tt),* )) => {
            ( $(read_value!($iter, $t)),* )
        };

        ($iter:expr, [ $t:tt ; $len:expr ]) => {
            (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
        };

        ($iter:expr, chars) => {
            read_value!($iter, String).chars().collect::<Vec<char>>()
        };

        ($iter:expr, usize1) => {
            read_value!($iter, usize) - 1
        };

        ($iter:expr, $t:ty) => {
            $iter.next().unwrap().parse::<$t>().expect("Parse error")
        };
    }
}
0