結果
問題 |
No.731 等差数列がだいすき
|
ユーザー |
|
提出日時 | 2022-12-08 02:36:18 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 1 ms / 1,500 ms |
コード長 | 805 bytes |
コンパイル時間 | 12,124 ms |
コンパイル使用メモリ | 379,532 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-14 04:48:16 |
合計ジャッジ時間 | 13,179 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
fn main() { let mut n = String::new(); std::io::stdin().read_line(&mut n).ok(); let n: usize = n.trim().parse().unwrap(); let mut a = String::new(); std::io::stdin().read_line(&mut a).ok(); let a: Vec<f64> = a.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let xsum = n * (n-1) / 2; let xsum = xsum as f64; let ysum = a.iter().sum::<f64>(); let x2sum = (0..n).map(|i| i*i).sum::<usize>() as f64; let xysum = (0..n).map(|i| i as f64 * a[i]).sum::<f64>(); let denom = n as f64 * x2sum - xsum * xsum; let d = (n as f64 * xysum - xsum * ysum) / denom; let x = (x2sum * ysum - xysum * xsum) / denom; let cost = (0..n).map(|i| (a[i] - d*i as f64 - x).powf(2.)).sum::<f64>(); println!("{} {}", x, d); println!("{}", cost); }