結果

問題 No.731 等差数列がだいすき
コンテスト
ユーザー donkorin_
提出日時 2018-09-09 14:55:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 1,500 ms
コード長 1,437 bytes
コンパイル時間 1,648 ms
コンパイル使用メモリ 168,836 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-23 09:46:51
合計ジャッジ時間 2,509 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<(b);++i)
#define erep(i,a,b) for(int i=a;i<=(int)(b);++i)
#define per(i,a,b) for(int i=(a);i>(b);--i)
#define eper(i,a,b) for(int i=(a);i>=b;--i)
#define pb push_back
#define mp make_pair
#define INF (1<<30)-1
#define MOD 1000000007
#define all(x) (x).begin(),(x).end()
#define vii vector<int>
#define vll vector<long long>
using namespace std;
typedef long long ll;
typedef pair<int,int> Pii;
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a = b; return 1; } return 0; }
int dy[]={0, 0, 1, -1};
int dx[]={1, -1, 0, 0};
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int lcm(int a,int b){return a/gcd(a, b)*b;}

int n;
int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
    cin >> n;
    vector<double> a(n), x(n);
    rep(i, 0, n) {
        cin >> a[i];
        x[i] = i;
    }
    double xsum = 0, ysum = 0, xysum = 0, x2sum = 0;
    rep(i, 0, n) {
        xsum += i;
        ysum += a[i];
        xysum += i * a[i];
        x2sum += i * i;
    }
    double d = (n * xysum - xsum * ysum) / (n * x2sum - (xsum * xsum));
    double b1 = (x2sum * ysum - xysum * xsum) / (n * x2sum - (xsum * xsum));
    double cost = 0;
    rep(i, 0, n) {
        cost += (a[i] - (b1 + i * d)) * (a[i] - (b1 + i * d));
    }
    printf("%.15f %.15f\n%.15f\n", b1, d, cost);
    return 0;
}
0