結果

問題 No.731 等差数列がだいすき
ユーザー とばりとばり
提出日時 2018-09-09 03:52:36
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 1,224 bytes
コンパイル時間 1,595 ms
コンパイル使用メモリ 169,332 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-08-24 17:15:15
合計ジャッジ時間 2,689 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 2 ms
4,376 KB
testcase_09 AC 2 ms
4,380 KB
testcase_10 AC 1 ms
4,380 KB
testcase_11 AC 2 ms
4,380 KB
testcase_12 AC 2 ms
4,376 KB
testcase_13 AC 2 ms
4,376 KB
testcase_14 AC 2 ms
4,376 KB
testcase_15 AC 2 ms
4,380 KB
testcase_16 AC 1 ms
4,380 KB
testcase_17 AC 2 ms
4,376 KB
testcase_18 AC 1 ms
4,376 KB
testcase_19 AC 2 ms
4,380 KB
testcase_20 AC 2 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using int64 = long long;
using uint64 = unsigned long long;

double mean(vector<int64>& seq)
{
    double sum = 0.0;
    for (auto e : seq)
    {
        sum += e;
    }
    return sum / seq.size();
}

double sd(vector<int64>& seq)
{
    double sum = 0.0;
    for (auto e : seq)
    {
        sum += e * e;
    }
    return sqrt(sum / seq.size() - pow(mean(seq), 2));
}

double cov(vector<int64>& x, vector<int64>& y)
{
    double sum = 0.0,
           xmean = mean(x),
           ymean = mean(y);
    for (int i = 0; i < (int)x.size(); i++)
    {
        sum += (x[i] - xmean) * (y[i] - ymean);
    }
    return sum / x.size();
}

int main()
{
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    int N;
    cin >> N;

    vector<int64> x(N), y(N);
    for (int i = 0; i < N; i++)
    {
        cin >> y[i];
        x[i] = i;
    }

    double A = cov(x, y) / pow(sd(x), 2),
           B = mean(y) - A * mean(x),
           error = 0.0;
    
    for (int i = 0; i < N; i++)
    {
        error += pow(y[i] - (B + A * i), 2);
    }

    cout << fixed << setprecision(10) << B << " " << A << endl;
    cout << fixed << setprecision(10) << error << endl;

    return 0;
}
0