結果

問題 No.2709 1975 Powers
コンテスト
ユーザー KA37RI
提出日時 2024-03-31 16:13:34
言語 Rust
(1.83.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

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