結果
問題 |
No.1170 Never Want to Walk
|
ユーザー |
![]() |
提出日時 | 2020-08-29 02:30:47 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 90 ms / 2,000 ms |
コード長 | 1,979 bytes |
コンパイル時間 | 3,438 ms |
コンパイル使用メモリ | 223,876 KB |
最終ジャッジ日時 | 2025-01-13 19:54:36 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 37 |
ソースコード
#pragma GCC optimize("O3") #include <bits/stdc++.h> using namespace std; using ll=long long; using P=pair<ll,ll>; template<class T> using V=vector<T>; #define fi first #define se second #define all(v) (v).begin(),(v).end() const ll inf=(1e18); //const ll mod=998244353; const ll mod=1000000007; ll GCD(ll a,ll b) {return b ? GCD(b,a%b):a;} ll LCM(ll c,ll d){return c/GCD(c,d)*d;} struct __INIT{__INIT(){cin.tie(0);ios::sync_with_stdio(false);cout<<fixed<<setprecision(15);}} __init; 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 (a>b) { a=b; return 1; } return 0; } class UF{ public: vector<int> par,num; int find(int v){ return (par[v]==v)? v: (par[v]=find(par[v])); } explicit UF(int N):par(N),num(N,1){ iota(all(par),0); } void unite(int u,int v){ u=find(u),v=find(v); if(u==v)return; if(num[u]<num[v])swap(u,v); num[u]+=num[v]; par[v]=u; } bool same(int u,int v){ return find(u)==find(v); } bool ispar(int v){ return v=find(v); } int size(int v){ return num[find(v)]; } }; int main(){ int n,a,b; cin>>n>>a>>b; UF uf(n); V<ll> d(n); V<P> seg; for(int i=0;i<n;i++)cin>>d[i]; for(int i=0;i<n;i++){ int r=upper_bound(all(d),d[i]+b)-d.begin(); int l=lower_bound(all(d),d[i]+a)-d.begin(); if(r-l>0){ uf.unite(i,l); seg.emplace_back(l,r-1); } } sort(all(seg)); V<P> res; ll L=-1,R=-1; for(int i=0;i<seg.size();i++){ if(R<seg[i].fi){ if(R!=-1){ res.emplace_back(L,R); } L=seg[i].fi; R=seg[i].se; } else chmax(R,seg[i].se); } if(R!=-1)res.emplace_back(L,R); for(int i=0;i<res.size();i++){ for(int j=res[i].fi;j<res[i].se;j++){ uf.unite(j,j+1); } } for(int i=0;i<n;i++){ cout<<uf.size(i)<<"\n"; } }