#967763 (Rust) No.2709 1975 Powers

• 問題: No.2709 1975 Powers
• ユーザー: 👑 KA37RI
• 提出日時: 2024-03-31 16:13:34
• 言語: Rust (1.77.0)
• 結果: AC
• 実行時間: 289 ms / 2,000 ms
• コード長: 2,167 bytes
• コンパイル時間: 1,781 ms
• コンパイル使用メモリ: 189,124 KB
• 実行使用メモリ: 6,676 KB
• 最終ジャッジ日時: 2024-03-31 16:13:40
• 合計ジャッジ時間: 5,130 ms
• ジャッジサーバーID （参考情報）: judge14 / judge11

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,676 KB
testcase_01	AC	1 ms; 6,676 KB
testcase_02	AC	3 ms; 6,676 KB
testcase_03	AC	144 ms; 6,676 KB
testcase_04	AC	239 ms; 6,676 KB
testcase_05	AC	162 ms; 6,676 KB
testcase_06	AC	51 ms; 6,676 KB
testcase_07	AC	1 ms; 6,676 KB
testcase_08	AC	47 ms; 6,676 KB
testcase_09	AC	162 ms; 6,676 KB
testcase_10	AC	1 ms; 6,676 KB
testcase_11	AC	5 ms; 6,676 KB
testcase_12	AC	14 ms; 6,676 KB
testcase_13	AC	169 ms; 6,676 KB
testcase_14	AC	24 ms; 6,676 KB
testcase_15	AC	28 ms; 6,676 KB
testcase_16	AC	128 ms; 6,676 KB
testcase_17	AC	1 ms; 6,676 KB
testcase_18	AC	17 ms; 6,676 KB
testcase_19	AC	236 ms; 6,676 KB
testcase_20	AC	2 ms; 6,676 KB
testcase_21	AC	45 ms; 6,676 KB
testcase_22	AC	281 ms; 6,676 KB
testcase_23	AC	282 ms; 6,676 KB
testcase_24	AC	282 ms; 6,676 KB
testcase_25	AC	289 ms; 6,676 KB
testcase_26	AC	280 ms; 6,676 KB

ソースコード

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")
  };
}