結果

問題 No.1170 Never Want to Walk
ユーザー Mr.Spock
提出日時 2020-12-15 14:43:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 457 ms / 2,000 ms
コード長 1,176 bytes
コンパイル時間 1,509 ms
コンパイル使用メモリ 175,684 KB
実行使用メモリ 33,792 KB
最終ジャッジ日時 2024-09-20 01:40:02
合計ジャッジ時間 9,535 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int nax = 2e5 + 7;
vector<int>par(nax);
vector<int>siz(nax, 1);
int get(int x) {
	if (x == par[x])return x;
	return (par[x]) = get(par[x]);
}
void unite(int x, int y) {
	x = get(x);
	y = get(y);
	if (x == y)return ;
	if (siz[y] > siz[x])swap(x, y);
	par[y] = x;
	siz[x] += siz[y];
}
void foo() {
	iota(par.begin(), par.end(), 0);
}
int main() {
#ifndef ONLINE_JUDGE
	freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);
#endif
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	foo();
	int n, a, b;
	cin >> n >> a >> b;
	vector<int>v(n);
	for (int i = 0; i < n; i++) {
		cin >> v[i];
	}
	int lower = 0, upper = 0;
	set<pair<int, int>>s;
	for (int i = 0; i < n; i++) {
		int l = lower_bound(v.begin(), v.end(), v[i] + a) - v.begin();
		lower = max(lower, l);
		upper = upper_bound(v.begin(), v.end(), v[i] + b) - v.begin();
		if (upper != 0)upper--;
		for (int j = lower; j <= upper; j++) {
			s.insert({i, j});
		}
		lower = upper - 1;
	}
	for (auto k : s) {
		int x = k.first;
		int y = k.second;
		unite(x, y);
	}
	for (int i = 0; i < n; i++) {
		cout << siz[get(i)] << endl;
	}
}

0