結果
問題 | No.731 等差数列がだいすき |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
#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 1000000000000000000int 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;}