結果
問題 |
No.731 等差数列がだいすき
|
ユーザー |
|
提出日時 | 2022-12-08 00:54:01 |
言語 | Rust (1.83.0 + proconio) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,695 bytes |
コンパイル時間 | 12,922 ms |
コンパイル使用メモリ | 376,556 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-14 03:42:15 |
合計ジャッジ時間 | 16,942 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 WA * 1 |
コンパイルメッセージ
warning: unused variable: `n` --> src/main.rs:32:9 | 32 | let n: usize = n.trim().parse().unwrap(); | ^ help: if this is intentional, prefix it with an underscore: `_n` | = note: `#[warn(unused_variables)]` on by default
ソースコード
const INF: f64 = (1usize << 60) as f64; const ITER: usize = 200; fn calc(d: f64, a: &Vec<f64>) -> (f64, f64) { let mut lower = -INF; let mut upper = INF; for _ in 0..ITER { let lmiddle = (upper + 2. * lower) / 3.; let umiddle = (2. * upper + lower) / 3.; let lucost = calc_c(d, lmiddle, a); let ulcost = calc_c(d, umiddle, a); if lucost > ulcost { lower = lmiddle; } else { upper = umiddle; } } (calc_c(d, upper, a), upper) } fn calc_c(d: f64, x: f64, a: &Vec<f64>) -> f64 { (0..a.len()).map(|i| { (d * i as f64 + x - a[i]) * (d * i as f64 + x - a[i]) }) .sum::<f64>() } 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 mut lower_d = -INF; let mut upper_d = INF; let mut result_d = 0.; let mut result_x = 0.; for _ in 0..ITER { let middle_ld = (upper_d + 2. * lower_d) / 3.; let middle_ud = (2. * upper_d + lower_d) / 3.; let (lucost, lux) = calc(middle_ld, &a); let (ulcost, ulx) = calc(middle_ud, &a); if lucost > ulcost { lower_d = middle_ld; result_d = middle_ud; result_x = ulx; } else { upper_d = middle_ud; result_d = middle_ld; result_x = lux; } } println!("{} {}", result_x, result_d); println!("{}", calc_c(result_d, result_x, &a)); }