結果

問題 No.731 等差数列がだいすき
ユーザー phspls
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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