結果

問題 No.1150 シュークリームゲーム（Easy）
ユーザー SSRS
提出日時 2020-08-07 21:35:55
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,555 bytes
コンパイル時間 1,943 ms
コンパイル使用メモリ 179,264 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-24 19:27:09
合計ジャッジ時間 4,077 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 16 WA * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int prev(int x, int N){
  if (x == 0){
    return N - 1;
  } else {
    return x - 1;
  }
}
int next(int x, int N){
  if (x == N - 1){
    return 0;
  } else {
    return x + 1;
  }
}
int main(){
  int N;
  cin >> N;
  int s, t;
  cin >> s >> t;
  s--;
  t--;
  vector<int> A(N);
  for (int i = 0; i < N; i++){
    cin >> A[i];
  }
  vector<bool> used(N, false);
  used[s] = true;
  used[t] = true;
  long long X = A[s];
  long long Y = A[t];
  priority_queue<pair<int, int>> pq1, pq2;
  pq1.push(make_pair(A[prev(s, N)], prev(s, N)));
  pq1.push(make_pair(A[next(s, N)], next(s, N)));
  pq2.push(make_pair(A[prev(t, N)], prev(t, N)));
  pq2.push(make_pair(A[next(t, N)], next(t, N)));
  for (int i = 0; i < N - 2; i++){
    if (i % 2 == 0){
      while (used[pq1.top().second]){
        pq1.pop();
      }
      X += pq1.top().first;
      int p = pq1.top().second;
      used[p] = true;
      pq1.pop();
      if (!used[prev(p, N)]){
        pq1.push(make_pair(A[prev(p, N)], prev(p, N)));
      }
      if (!used[next(p, N)]){
        pq1.push(make_pair(A[next(p, N)], next(p, N)));
      }
    }
    if (i % 2 == 1){
      while (used[pq2.top().second]){
        pq2.pop();
      }
      Y += pq2.top().first;
      int p = pq2.top().second;
      used[p] = true;
      pq2.pop();
      if (!used[prev(p, N)]){
        pq2.push(make_pair(A[prev(p, N)], prev(p, N)));
      }
      if (!used[next(p, N)]){
        pq2.push(make_pair(A[next(p, N)], next(p, N)));
      }
    }
  }
  cout << X - Y << endl;
}
0