結果

問題 No.731 等差数列がだいすき
ユーザー SugarDragon5
提出日時 2018-09-07 21:57:16
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 942 bytes
コンパイル時間 1,646 ms
コンパイル使用メモリ 167,032 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-29 13:06:17
合計ジャッジ時間 2,400 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
// 最小二乗法の計算 
void lsm(double x[],double y[], int N, double *a0, double *a1)
{
        int i;
        double A00=0 ,A01=0, A02=0, A11=0, A12=0;
 
        for (i=0;i<N;i++) {
                A00+=1.0;
                A01+=x[i];
                A02+=y[i];
                A11+=x[i]*x[i];
                A12+=x[i]*y[i];
        }
 
        *a0 = (A02*A11-A01*A12) / (A00*A11-A01*A01);
        *a1 = (A00*A12-A01*A02) / (A00*A11-A01*A01);
}
 
int main()
{
    int n;
    cin>>n;
    double x[1000];
    double y[1000];
    for(int i=0;i<n;i++){
        x[i]=i;
        cin>>y[i];
    }
        double a0 = 0,a1 = 0;    
        
        // 最小二乗法の計算
        lsm(x, y,n, &a0, &a1);
        
    cout<<fixed<<setprecision(10)<<a0<<" "<<a1<<endl;
    double sum=0;
    for(int i=0;i<n;i++){
        sum+=pow(y[i]-(a0+a1*i),2);
    }
    cout<<sum<<endl;
        return 0;
}
0