結果

問題 No.2709 1975 Powers
ユーザー 👑 KA37RIKA37RI
提出日時 2024-03-31 16:13:34
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 260 ms / 2,000 ms
コード長 2,167 bytes
コンパイル時間 13,954 ms
コンパイル使用メモリ 401,884 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-09-30 21:04:03
合計ジャッジ時間 17,535 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 3 ms
6,816 KB
testcase_03 AC 133 ms
6,820 KB
testcase_04 AC 218 ms
6,816 KB
testcase_05 AC 148 ms
6,820 KB
testcase_06 AC 48 ms
6,816 KB
testcase_07 AC 1 ms
6,820 KB
testcase_08 AC 43 ms
6,816 KB
testcase_09 AC 147 ms
6,820 KB
testcase_10 AC 1 ms
6,816 KB
testcase_11 AC 5 ms
6,816 KB
testcase_12 AC 12 ms
6,816 KB
testcase_13 AC 154 ms
6,816 KB
testcase_14 AC 23 ms
6,820 KB
testcase_15 AC 25 ms
6,820 KB
testcase_16 AC 117 ms
6,820 KB
testcase_17 AC 1 ms
6,816 KB
testcase_18 AC 16 ms
6,820 KB
testcase_19 AC 215 ms
6,816 KB
testcase_20 AC 2 ms
6,820 KB
testcase_21 AC 42 ms
6,816 KB
testcase_22 AC 257 ms
6,820 KB
testcase_23 AC 258 ms
6,816 KB
testcase_24 AC 259 ms
6,816 KB
testcase_25 AC 260 ms
6,816 KB
testcase_26 AC 258 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

fn main() {
  input! {
    n: usize,
    p: usize,
    q: usize,
    a: [usize; n],
  }
  let mut ans = 0;
  let mut b = a.clone();
  b.sort();
  let pow10: Vec<usize> = b.iter().map(|&bi| pow(10, bi, p)).collect();
  let pow9: Vec<usize> = b.iter().map(|&bi| pow(9, bi, p)).collect();
  let pow7: Vec<usize> = b.iter().map(|&bi| pow(7, bi, p)).collect();
  let pow5: Vec<usize> = b.iter().map(|&bi| pow(5, bi, p)).collect();
  for i in 0..n {
    for j in i+1..n {
      for k in j+1..n {
        for l in k+1..n {
          if (pow10[i] + pow9[j] + pow7[k] + pow5[l]) % p == q {
            ans += 1;
          }
        }
      }
    }
  }
  println!("{}", ans);
}

fn pow(x: usize, n: usize, p: usize) -> usize {
  let mut k = n;
  let mut res = 1;
  let mut x = x;
  while k > 0 {
    if k & 1 == 1 {
      res = res * x % p;
    }
    x = x * x % p;
    k >>= 1;
  }
  res
}

#[macro_export]
macro_rules! input {
  (source = $s:expr, $($r:tt)*) => {
    let mut iter = $s.split_whitespace();
    let mut next = || { iter.next().unwrap() };
    input_inner!{next, $($r)*}
  };
  ($($r:tt)*) => {
    let stdin = std::io::stdin();
    let mut bytes = std::io::Read::bytes(std::io::BufReader::new(stdin.lock()));
    let mut next = move || -> String{
      bytes
        .by_ref()
        .map(|r|r.unwrap() as char)
        .skip_while(|c|c.is_whitespace())
        .take_while(|c|!c.is_whitespace())
        .collect()
    };
    input_inner!{next, $($r)*}
  };
}

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

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

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

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

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

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

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