結果

問題 No.731 等差数列がだいすき
コンテスト
ユーザー phspls
提出日時 2022-12-08 00:55:24
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,695 bytes
コンパイル時間 12,300 ms
コンパイル使用メモリ 376,664 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-14 03:43:29
合計ジャッジ時間 19,855 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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

ソースコード

diff # 

const INF: f64 = (1usize << 60) as f64;
const ITER: usize = 300;

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