結果
問題 | No.1825 Except One |
ユーザー |
|
提出日時 | 2024-02-08 21:37:02 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 84 ms / 3,000 ms |
コード長 | 2,317 bytes |
コンパイル時間 | 14,721 ms |
コンパイル使用メモリ | 378,592 KB |
実行使用メモリ | 103,552 KB |
最終ジャッジ日時 | 2024-09-28 12:51:49 |
合計ジャッジ時間 | 15,573 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 31 |
ソースコード
#![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,a:[usize;n],}let mut a = a;a.sort();let suma = a.iter().sum::<usize>();let mut dp = vec![vec![vec![0usize; suma + 1]; n + 1]; n + 1];dp[0][0][0] = 1;let mut ans = 0;for i in 0..n {for j in 0..=i {for k in 0..=suma {if dp[i][j][k] == 0 {continue;}dp[i + 1][j][k] += dp[i][j][k];dp[i + 1][j + 1][k + a[i]] += dp[i][j][k];if j != 0 && (k + a[i]) % j == 0 && a[i] * j <= k + a[i] {ans += dp[i + 1][j + 1][k + a[i]];}}}}println!("{}", ans);}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")};}}