結果

問題 No.731 等差数列がだいすき
コンテスト
ユーザー daku9640
提出日時 2018-09-07 22:27:13
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 1,500 ms
コード長 2,411 bytes
コンパイル時間 1,687 ms
コンパイル使用メモリ 171,360 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-29 15:32:37
合計ジャッジ時間 2,446 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef vector<int> vi;
typedef vector<double> vd;
typedef vector<long long> vll;
typedef vector<string> vs;
typedef vector<bool> vb;
typedef pair<int, int> pii;
struct cww{cww(){ios::sync_with_stdio(false);cin.tie(0);}}star;
#define all(c) (c).begin(), (c).end()
#define rall(c) (c).rbegin(), (c).rend()
#define vv(type, c, m, n, i) vector<vector<type>> c(m, vector<type>(n, i))
#define mat(type, c, m, n) vector<vector<type>> c(m, vector<type>(n));for(auto& r:c)for(auto& i:r)cin>>i;
#define rep(i,a,n) for(int i=(a), i##_len=(n); i<i##_len; ++i)
#define rrep(i,a,n) for(int i=(a), i##_len=(n); i>i##_len; --i)
#define each(x,y) for(auto &(x):(y))
#define var(type, ...)type __VA_ARGS__;Scan(__VA_ARGS__);
template<typename T> void Scan(T& t) { cin >> t; }
template<typename First, typename...Rest>void Scan(First& first, Rest&...rest) { cin >> first; Scan(rest...); }
#define vec(type, c, n) vector<type> c(n);for(auto& i:c) cin>>i;
#define len(x) ((int)(x).size())
#define mp make_pair
#define eb emplace_back
#define em emplace
#define pb pop_back
#define fi first
#define se second
#define get(a, i) get<i>(a)
#define bit_count __builtin_popcount
#define mt make_tuple
inline string zero(int a,string s,char c='0'){return string(a-len(s),c)+s;}
inline string zero(int a,int t,char c='0'){return zero(a, to_string(t), c);}
inline string zero(int a,ll t,char c='0'){return zero(a, to_string(t), c);}
#define mini min_element
#define maxi max_element
#define sum accumulate
#define chmax(a,b) if(a<b)a=b
#define chmin(a,b) if(a>b)a=b
template <class T> inline void print(T t){cout << t << endl;}
template <class H, class... T> inline void print(H h, T... t){cout << h << " ";print(t...);}
void solve() {
	var(int, n);
	vec(double, v, n);
	double mean_x = n * (n - 1) / 2.0 / n;
	double mean_y = sum(all(v), 0.0) / n;
	vd sd_x(n);
	rep(i, 0, n)
		sd_x[i] = i - mean_x;
	vd sd_y(n);
	rep(i, 0, n)
		sd_y[i] = v[i] - mean_y;
	double var_x = 0;
	rep(i, 0, n)
		var_x += sd_x[i] * sd_x[i];
	var_x /= n;
	double cov_xy = 0;
	rep(i, 0, n)
		cov_xy += (sd_x[i] * sd_y[i]);
	cov_xy /= n;
	double a = cov_xy / var_x;
	double b = mean_y - a * mean_x;
	double cost = 0;
	rep(i, 0, n)
		cost += abs(pow(v[i] - (a * i + b), 2));
	print(b, a);
	print(cost);
}
int main() {
	cout << setprecision(7) << fixed;
	solve();
	return 0;
}
0