#648129 (C++17) No.731 等差数列がだいすき

• 問題: No.731 等差数列がだいすき
• ユーザー: Fu_L
• 提出日時: 2021-04-16 12:08:24
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 3 ms / 1,500 ms
• コード長: 722 bytes
• コンパイル時間: 4,316 ms
• コンパイル使用メモリ: 262,000 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-02 14:02:51
• 合計ジャッジ時間: 5,105 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	3 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 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

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<ll,ll> P;
typedef modint998244353 mint;
#define rep(i,a,b) for(ll i=a;i<b;i++)
#define rrep(i,a,b) for(ll i=a;i>=b;i--)
const ll inf=1e18;
int main(void){
    cin.tie(0);
    ios::sync_with_stdio(0);
    int n; scanf("%d",&n);
	vector<double> x(n),y(n);
	rep(i,0,n) x[i]=i, scanf("%lf",&y[i]);

	double sx=0,sy=0,sxx=0,sxy=0;
	rep(i,0,n){
		sx+=x[i];
		sy+=y[i];
		sxx+=x[i]*x[i];
		sxy+=x[i]*y[i];
	}
	double a=(n*sxy-sx*sy)/(n*sxx-sx*sx);
	double b=(sxx*sy-sxy*sx)/(n*sxx-sx*sx);

	double ans=0;
	rep(i,0,n) ans+=(y[i]-(a*x[i]+b))*(y[i]-(a*x[i]+b));
	printf("%.9f %.9f\n%.9f\n",b,a,ans);

}