結果

問題 No.731 等差数列がだいすき
コンテスト
ユーザー 41Toame
提出日時 2018-09-07 21:41:53
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 1,266 bytes
コンパイル時間 1,508 ms
コンパイル使用メモリ 159,216 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-29 11:59:54
合計ジャッジ時間 2,657 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define ll long long int
#define rep(i,n) for( int i = 0; i < n; i++ )
#define rrep(i,n) for( int i = n; i >= 0; i-- )
#define REP(i,s,t) for( int i = s; i <= t; i++ )
#define RREP(i,s,t) for( int i = s; i >= t; i-- )
#define dump(x)  cerr << #x << " = " << (x) << endl;
#define INF 2000000000
#define mod 1000000007
#define INF2 1000000000000000000

int N;

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

double cost(double x[], double b, double d) {
	double ret = 0;
	double num = b;
	rep(i, N) {
		double tmp = x[i] - num;
		ret += tmp * tmp;
		num += d;
	}
	return ret;
}


int main(void)
{
    cin.tie(0);
    ios::sync_with_stdio(false);
	cin >> N;
	double a[1010];
	double b[1010];
	rep(i, N) cin >> b[i];
	rep(i, N) a[i] = (double)i;
	double a0 = 0, a1 = 0;
	lsm(a, b, N, &a0, &a1);
	cout << fixed;
	cout << setprecision(12);
	cout << a0 << " " << a1 << endl;
	cout << cost(b, a0, a1) << endl;

    return 0;
}
0