結果
問題 | No.800 四平方定理 |
ユーザー |
|
提出日時 | 2020-08-18 21:49:49 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 112 ms / 2,000 ms |
コード長 | 1,767 bytes |
コンパイル時間 | 13,333 ms |
コンパイル使用メモリ | 385,384 KB |
実行使用メモリ | 53,628 KB |
最終ジャッジ日時 | 2024-10-12 03:15:59 |
合計ジャッジ時間 | 15,589 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
コンパイルメッセージ
warning: variable does not need to be mutable --> src/main.rs:10:13 | 10 | let mut s = { | ----^ | | | help: remove this `mut` ... 62 | / input!{ 63 | | n:usize, d:usize 64 | | } | |_____- in this macro invocation | = note: `#[warn(unused_mut)]` on by default = note: this warning originates in the macro `input` (in Nightly builds, run with -Z macro-backtrace for more info)
ソースコード
#![allow(dead_code)]#![allow(non_snake_case)]macro_rules! input {(source = $s:expr, $($r:tt)*) => {let mut iter = $s.split_whitespace();input_inner!{iter, $($r)*}};($($r:tt)*) => {let mut 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_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_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")};}fn gcd(a:i32, b:i32) -> i32 {if b == 0 {a} else {gcd(b, a%b)}}fn main() {input!{n:usize, d:usize}let m = n * n * 2 + 1;let mut f = vec![0; m];let mut g = vec![0; m];for x in 1..n+1 {for y in 1..n+1 {f[x * x + y * y] += 1;if x * x < y * y + d && y * y + d < m + x * x{g[y * y + d - x * x] += 1;}}}let ans = f.iter().zip(g).fold(0, |acc, (x, y)| acc + x * y);println!("{}", ans);}