結果

問題 No.921 ずんだアロー
ユーザー sash2104
提出日時 2019-11-08 22:50:04
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 41 ms / 2,000 ms
コード長 1,104 bytes
コンパイル時間 854 ms
コンパイル使用メモリ 77,492 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-15 02:00:18
合計ジャッジ時間 2,343 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;

typedef long long ll;
bool done[100010];

struct Data {
  int l, r, n;
  Data(int l, int r): l(l), r(r), n(r-l+1) {}
};

// ">" のオーバーロード numを基準に大小比較を行う
bool operator < (const Data &d1, const Data &d2){
  if (d1.n == d2.n) return d1.l > d2.l;
  return d1.n < d2.n;
};

int main() {
  int n;
  cin >> n;
  int bef = -1;
  int l = 0, r = 0;
  priority_queue<Data> pq;
  for (int i = 0; i < n; ++i) {
    int a;
    cin >> a;
    if (a == bef) { 
      r = i;
    }
    else if (i > 0) {
      pq.push(Data(l, r));
      l = i;
      r = i;
    }
    bef = a;
  }
  pq.push(Data(l, r));

  ll ans = 0;
  while (!pq.empty()) {
    Data d = pq.top(); pq.pop();
    int add = d.n;
    if (done[d.l]) { --add; }
    if (d.l != d.r && done[d.r]) { --add; }
    if (add == 0) continue;
    ans += add;
    if (d.l > 0) done[d.l-1] = true;
    if (d.r < n-1) done[d.r+1] = true;
    // cerr << d.l << " " << d.r << " " << d.n << " " << ans << endl;
  }
  cout << ans << endl;
}
0