結果

問題 No.1375 Divide and Update
ユーザー chocono2230chocono2230
提出日時 2021-02-05 22:31:57
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 429 ms / 2,000 ms
コード長 2,698 bytes
コンパイル時間 2,729 ms
コンパイル使用メモリ 216,704 KB
実行使用メモリ 61,008 KB
最終ジャッジ日時 2023-09-15 08:57:44
合計ジャッジ時間 9,091 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 1 ms
4,380 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 2 ms
4,380 KB
testcase_10 AC 373 ms
53,824 KB
testcase_11 AC 335 ms
49,304 KB
testcase_12 AC 185 ms
28,340 KB
testcase_13 AC 154 ms
24,272 KB
testcase_14 AC 419 ms
60,840 KB
testcase_15 AC 417 ms
60,736 KB
testcase_16 AC 417 ms
60,844 KB
testcase_17 AC 417 ms
60,740 KB
testcase_18 AC 421 ms
60,856 KB
testcase_19 AC 429 ms
60,740 KB
testcase_20 AC 370 ms
61,008 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(ri,n) for(int ri = (int)(n-1); ri >= 0; ri--)
#define rep2(i,x,n) for(int i = (int)(x); i < (int)(n); i++)
#define rrep2(ri,x,n) for(int ri = (int)(n-1); ri >= (int)(x); ri--)
#define repit(itr,x) for(auto itr = x.begin(); itr != x.end(); itr++)
#define rrepit(ritr,x) for(auto ritr = x.rbegin(); ritr != x.rend(); ritr++)
#define ALL(x) x.begin(), x.end()
using ll = long long;
using namespace std;

int main(){
  ll n, x, y;
  cin >> n >> x >> y;
  vector<ll> a(n), xa(n), ya(n);
  rep(i, n){
    ll in;
    cin >> in;
    a.at(i) = in;
    xa.at(i) = x - in;
    ya.at(i) = y - in;
  }
  vector<ll> sumsl(n+1, 0);
  rep(i, n) sumsl.at(i+1) = sumsl.at(i) + a.at(i);
  vector<ll> sumsr(n+1, 0);
  rrep(i, n) sumsr.at(i) = sumsr.at(i+1) + a.at(i);

  vector<ll> xv(n), yv(n);
  vector dp(n, vector<ll>(2, -1e18));
  vector idxs(n, vector<pair<int, int>>(2));
  dp.at(0).at(0) = dp.at(0).at(1) = xa.at(0);
  idxs.front().at(0) = idxs.front().at(1) = {0, 0};
  rep2(i, 1, n-1){
    if(dp.at(i-1).at(1) + xa.at(i) > xa.at(i)){
      idxs.at(i).at(1) = {idxs.at(i-1).at(1).first, i};
    }else{
      idxs.at(i).at(1) = {i, i};
    }
    dp.at(i).at(1) = max(xa.at(i), dp.at(i-1).at(1) + xa.at(i));
    if(dp.at(i).at(1) > dp.at(i-1).at(0)){
      idxs.at(i).at(0) = idxs.at(i).at(1);
    }else{
      idxs.at(i).at(0) = idxs.at(i-1).at(0);
    }
    dp.at(i).at(0) = max(dp.at(i).at(1), dp.at(i-1).at(0));
  }

  vector dp2(n, vector<ll>(2, -1e18));
  vector idxs2(n, vector<pair<int, int>>(2));
  dp2.back().at(0) = dp2.back().at(1) = ya.back();
  idxs2.back().at(0) = idxs2.back().at(1) = {n-1, n-1};
  rrep2(i, 1, n-1){
    if(dp2.at(i+1).at(1) + ya.at(i) > ya.at(i)){
      idxs2.at(i).at(1) = {i, idxs2.at(i+1).at(1).second};
    }else{
      idxs2.at(i).at(1) = {i, i};
    }
    dp2.at(i).at(1) = max(ya.at(i), dp2.at(i+1).at(1) + ya.at(i));
    if(dp2.at(i).at(1) > dp2.at(i+1).at(0)){
      idxs2.at(i).at(0) = idxs2.at(i).at(1);
    }else{
      idxs2.at(i).at(0) = idxs2.at(i+1).at(0);
    }
    dp2.at(i).at(0) = max(dp2.at(i).at(1), dp2.at(i+1).at(0));
  }

  vector<ll> ansx(n, 0), ansy(n, 0);
  rep2(i, 1, n-1){
    ll nx, nl, nr;
    if(dp.at(i-1).at(0) < dp.at(i-1).at(1)){
      nx = dp.at(i-1).at(1);
      tie(nl, nr) = idxs.at(i-1).at(1);
    }else{
      nx = dp.at(i-1).at(0);
      tie(nl, nr) = idxs.at(i-1).at(0);
    }
    ll add = sumsl.at(i);
    ansx.at(i) = add + nx;
  }
  rrep2(i, 1, n-1){
    ansy.at(i) = sumsr.at(i+1) + max(dp2.at(i+1).at(0), dp2.at(i+1).at(1));
  }
  rep2(i, 1, n-1){
    ll ans = ansx.at(i) + ansy.at(i) + a.at(i);
    cout << ans << endl;
  }
  return 0;
}
0