結果

問題 No.731 等差数列がだいすき
ユーザー 41Toame41Toame
提出日時 2018-09-07 21:41:53
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 1,266 bytes
コンパイル時間 1,494 ms
コンパイル使用メモリ 160,300 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-07 00:41:54
合計ジャッジ時間 2,288 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

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