結果

問題 No.731 等差数列がだいすき
ユーザー raven7959raven7959
提出日時 2022-01-18 10:12:14
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 2,154 bytes
コンパイル時間 1,842 ms
コンパイル使用メモリ 198,236 KB
最終ジャッジ日時 2025-01-27 13:01:49
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rrep(i, n) for (int i = (int)(n - 1); i >= 0; i--)
#define all(x) (x).begin(), (x).end()
#define sz(x) int(x.size())
#define yn(joken) cout<<((joken) ? "Yes" : "No")<<endl
#define YN(joken) cout<<((joken) ? "YES" : "NO")<<endl
using namespace std;
using ll = long long;
using vi = vector<int>;
using vl = vector<ll>;
using vs = vector<string>;
using vc = vector<char>;
using vd = vector<double>;
using vvi = vector<vector<int>>;
using vvl = vector<vector<ll>>;
const int INF = 1e9;
const ll LINF = 1e18;
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;
}
template <class T>
vector<T> make_vec(size_t a) {
    return vector<T>(a);
}
template <class T, class... Ts>
auto make_vec(size_t a, Ts... ts) {
    return vector<decltype(make_vec<T>(ts...))>(a, make_vec<T>(ts...));
}
template <typename T>
istream& operator>>(istream& is, vector<T>& v) {
    for (int i = 0; i < int(v.size()); i++) {
        is >> v[i];
    }
    return is;
}
template <typename T>
ostream& operator<<(ostream& os, const vector<T>& v) {
    for (int i = 0; i < int(v.size()); i++) {
        os << v[i];
        if (i < int(v.size()) - 1) os << ' ';
    }
    return os;
}

pair<double,double> LeastSquaresMethod(vector<pair<double,double>> &V){
    int N=sz(V);
    double sx=0,sy=0,sxx=0,sxy=0;
    for(auto [x,y]:V){
        sx+=x;
        sy+=y;
        sxx+=x*x;
        sxy+=x*y;
    }
    return make_pair((N*sxy-sx*sy)/(N*sxx-sx*sx),(sxx*sy-sx*sxy)/(N*sxx-sx*sx));
}

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    int N;
    cin>>N;
    vector<pair<double,double>> V(N);
    rep(i,N){
        double y;
        cin>>y;
        V[i]=make_pair(double(i),y);
    }
    auto [a,b]=LeastSquaresMethod(V);
    cout<<fixed<<setprecision(12)<<b<<" "<<a<<endl;
    double cost=0;
    for(auto [x,y]:V) cost+=(y-a*x-b)*(y-a*x-b);
    cout<<fixed<<setprecision(12)<<cost<<endl;
}
0