結果
問題 | No.731 等差数列がだいすき |
ユーザー |
|
提出日時 | 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 |
ソースコード
#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_tupleinline 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=btemplate <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;}