結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー yataro800
提出日時 2023-01-08 10:51:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 355 ms / 2,000 ms
コード長 1,676 bytes
コンパイル時間 4,012 ms
コンパイル使用メモリ 241,744 KB
実行使用メモリ 18,880 KB
最終ジャッジ日時 2024-12-15 13:00:02
合計ジャッジ時間 11,401 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <math.h>

#include <algorithm>
#include <array>
#include <atcoder/all>
#include <bitset>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <sstream>
#include <vector>

using namespace std;
using namespace atcoder;

using Graph = vector<vector<int>>;

using ll = long long;
typedef pair<ll, ll> P_ll;
typedef pair<int, int> P;

const ll INF_ll = 1e17;
const int INF = 1e8;
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 <typename T>
using min_priority_queue = priority_queue<T, vector<T>, greater<T>>;

int main() {
  ll N, A, B;
  cin >> N >> A >> B;
  vector<ll> x(N);
  vector<pair<ll, P_ll>> y;
  for (int i = 0; i < N; i++) {
    cin >> x[i];
    y.push_back({x[i], {1, i}});
    y.push_back({x[i] + A, {0, i}});
  }
  dsu tree(N);
  sort(y.begin(), y.end());
  min_priority_queue<P_ll> pq;
  for (int i = 0; i < y.size(); i++) {
    auto p = y[i];
    // cout << p.first << " " << p.second.first << " " << p.second.second <<
    // endl;
    if (p.second.first == 1) {
      P_ll max_r = {-1, -1};
      while (!pq.empty()) {
        auto pp = pq.top();
        pq.pop();
        if (pp.first >= p.first) {
          tree.merge(p.second.second, pp.second);
          chmax(max_r, pp);
        }
      }
      if (max_r.first > 0) {
        pq.push(max_r);
      }
    } else {
      pq.push({x[p.second.second] + B, p.second.second});
    }
  }
  for (int i = 0; i < N; i++) {
    cout << tree.size(i) << endl;
  }

  return 0;
}
0