結果

問題 No.731 等差数列がだいすき
ユーザー le_panda_noir
提出日時 2020-06-13 19:18:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,015 bytes
コンパイル時間 1,230 ms
コンパイル使用メモリ 78,288 KB
最終ジャッジ日時 2025-01-11 03:54:05
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <iomanip>

using namespace std;
using ll = long long;
using vi = vector<int>;

#define in(v) v; cin >> v;
void ins() {}
template<class T,class... Rest>void ins(T& v,Rest&... rest){cin>>v;ins(rest...);}
#define rep(i,n) for(int i=0,_i=(n);i<_i;++i)

int main() {
  int N;
  ins(N);

  vi A(N);
  rep(i, N) {
    cin >> A[i];
  }
  cout << fixed << setprecision(15);

  ll diff = 0;
  rep(i, N-1) diff += A[i+1] - A[i];

  ll B = 0, B2 = 0;
  rep(i, N) {
    B += A[i] * (N-1) - diff * i;
    B2 += A[i] * N - diff * i;
  }
  double b = 1.0 * B / (N-1) / N,
         d = 1.0 * diff  / (N-1);
  double b2 = 1.0 * B / N / (N+1),
         d2 = 1.0 * diff  / N;

  double ans = 0, ans2 = 0;
  rep(i, N) {
    ans += 1.0 * (A[i] - b - d * i) * (A[i] - b - d * i);
    ans2 += 1.0 * (A[i] - b2 - d2 * i) * (A[i] - b2 - d2 * i);
  }
  if (ans < ans2)
    cout << b << " " << d << endl << ans << endl;
  else
    cout << b2 << " " << d2 << endl << ans2 << endl;


  return 0;
}
0