結果

問題 No.731 等差数列がだいすき
コンテスト
ユーザー vjudge1
提出日時 2026-03-23 11:27:41
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 870 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,713 ms
コンパイル使用メモリ 94,992 KB
実行使用メモリ 7,720 KB
最終ジャッジ日時 2026-03-23 11:27:48
合計ジャッジ時間 4,940 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <iomanip>
using namespace std;

int main() {
    int N;
    cin >> N;
    vector<long long> a(N);
    for (int i = 0; i < N; ++i) {
        cin >> a[i];
    }
    long long sum_x = 0, sum_x2 = 0, sum_a = 0, sum_xa = 0;
    for (int k = 0; k < N; ++k) {
        sum_x += k;
        sum_x2 += (long long)k * k;
        sum_a += a[k];
        sum_xa += (long long)k * a[k];
    }
    double D = (double)N * sum_x2 - (double)sum_x * sum_x;
    double d = ((double)N * sum_xa - (double)sum_x * sum_a) / D;
    double b1 = ((double)sum_a * sum_x2 - (double)sum_x * sum_xa) / D;
    double cost = 0.0;
    for (int k = 0; k < N; ++k) {
        double diff = a[k] - (b1 + d * k);
        cost += diff * diff;
    }
    cout << fixed << setprecision(15);
    cout << b1 << " " << d << endl;
    cout << cost << endl;
    return 0;
}
0