結果

問題 No.731 等差数列がだいすき
ユーザー kaito_tateyama
提出日時 2018-09-07 22:06:45
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 1,281 bytes
コンパイル時間 1,933 ms
コンパイル使用メモリ 158,496 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-29 13:56:06
合計ジャッジ時間 2,838 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//---------------------------
using namespace std;
//---------------------------
#define REP(i, n) for (int i = 0; i < (n); i++)
#define P(x) cout << (x) << "\n"
#define fcout cout << fixed << setprecision(18)

#define MOD 1000000007 // 1e9+7
#define PI 3.1415926535
/*
3.1415926535897932384626433832795028841971693993751058209749445923078164062862089986280348253421170679
*/
#define ll long long int         // 10^18
#define INF 1000000001           // 1e9+1
#define LINF 1000000000000000001 // 1e18+1

int dx[4] = {1, -1, 0, 0};
int dy[4] = {0, 0, 1, -1};
//---------------------------
int a[2000];
int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);

    // ifstream in("input.txt");
    // cin.rdbuf(in.rdbuf());
    
    int n;cin>>n;
    ll fi_s=0;
    ll se_s=0;
    ll th_s=0;
    ll fo_s=0;
    for(int i=0;i<n;i++){
        cin>>a[i];
        fi_s+=i;
        se_s+=i*i;
        th_s+=i*a[i];
        fo_s+=a[i];
    }
    ll mo=n * se_s - fi_s * fi_s;
    double aa = (double)(n * th_s - fi_s * fo_s) / mo;
    double b = (double)(se_s * fo_s - th_s * fi_s) / mo;
    fcout<<b<<" "<<aa<<"\n";
    double cost=0;
    for(int i=0;i<n;i++){
        cost+=(a[i]-aa*i-b)*(a[i]-aa*i-b);
    }
    fcout<<cost<<endl;
    return 0;
}
0