結果

問題 No.731 等差数列がだいすき
ユーザー phsplsphspls
提出日時 2022-12-08 00:55:24
言語 Rust
(1.77.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
5,248 KB
testcase_01 AC 5 ms
5,248 KB
testcase_02 AC 6 ms
5,248 KB
testcase_03 AC 122 ms
5,248 KB
testcase_04 AC 330 ms
5,248 KB
testcase_05 AC 123 ms
5,248 KB
testcase_06 WA -
testcase_07 AC 261 ms
5,248 KB
testcase_08 AC 315 ms
5,248 KB
testcase_09 AC 404 ms
5,248 KB
testcase_10 AC 484 ms
5,248 KB
testcase_11 AC 19 ms
5,248 KB
testcase_12 AC 213 ms
5,248 KB
testcase_13 AC 302 ms
5,248 KB
testcase_14 AC 354 ms
5,248 KB
testcase_15 AC 432 ms
5,248 KB
testcase_16 AC 237 ms
5,248 KB
testcase_17 AC 547 ms
5,248 KB
testcase_18 AC 555 ms
5,248 KB
testcase_19 AC 553 ms
5,248 KB
testcase_20 AC 560 ms
5,248 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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