結果

問題 No.1170 Never Want to Walk
ユーザー kaikeykaikey
提出日時 2020-08-14 21:56:15
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 52 ms / 2,000 ms
コード長 2,408 bytes
コンパイル時間 1,585 ms
コンパイル使用メモリ 171,292 KB
実行使用メモリ 7,808 KB
最終ジャッジ日時 2024-10-10 15:08:23
合計ジャッジ時間 4,460 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#include <random>
using namespace std; typedef unsigned long long _ulong; typedef long long int lint; typedef pair<lint, lint> plint; typedef pair<double long, double long> pld;
#define ALL(x) (x).begin(), (x).end()
#define SZ(x) ((lint)(x).size())
#define FOR(i, begin, end) for(lint i=(begin),i##_end_=(end);i<i##_end_;i++)
#define IFOR(i, begin, end) for(lint i=(end)-1,i##_begin_=(begin);i>=i##_begin_;i--)
#define REP(i, n) FOR(i,0,n)
#define IREP(i, n) IFOR(i,0,n)
#define endk '\n'
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; }
const lint mod = 1e9 + 7, INF = 1e18;
lint dx[8] = { 0, -1, 1, 0, 1, -1, 1, -1 }, dy[8] = { 1, 0, 0, -1, -1, -1, 1, 1 };
typedef pair<double, lint> Pa;

template< typename T >
class UnionFind {
public:
	vector <T> par; // 各元の親を表す配列
	vector <T> siz; // 素集合のサイズを表す配列(1 で初期化)

	// Constructor
	UnionFind(T sz_) : par(sz_), siz(sz_, 1ll) {
		for (T i = 0; i < sz_; ++i) par[i] = i; // 初期では親は自分自身
	}
	void init(T sz_) {
		par.resize(sz_);
		siz.assign(sz_, 1ll);  // resize だとなぜか初期化されなかった
		for (T i = 0; i < sz_; ++i) par[i] = i; // 初期では親は自分自身
	}

	// Member Function
	// Find
	T root(T x) { // 根の検索
		while (par[x] != x) {
			x = par[x] = par[par[x]]; // x の親の親を x の親とする
		}
		return x;
	}

	// Union(Unite, Merge)
	bool merge(T x, T y) {
		x = root(x);
		y = root(y);
		if (x == y) return false;
		// merge technique（データ構造をマージするテク．小を大にくっつける）
		if (siz[x] < siz[y]) swap(x, y);
		siz[x] += siz[y];
		par[y] = x;
		return true;
	}

	bool issame(T x, T y) { // 連結判定
		return root(x) == root(y);
	}

	T size(T x) { // 素集合のサイズ
		return siz[root(x)];
	}
};
lint N, A, B, arr[200000];
int main() {
	cin.tie(0); ios_base::sync_with_stdio(false);
	cin >> N >> A >> B;
	UnionFind<lint> tree(N);
	REP(i, N) {
		cin >> arr[i];
	}
	REP(i, N - 1) {
		auto l = lower_bound(arr + i + 1, arr + N, A + arr[i]) - arr;
		auto r = upper_bound(arr + i + 1, arr + N, B + arr[i]) - arr - 1;
		while (l <= r && !tree.issame(r, i)) {
			tree.merge(r, i);
			r--;
		}
	}
	REP(i, N) {
		cout << tree.size(i) << endk;
	}
}
0