結果

問題 No.1170 Never Want to Walk
ユーザー fumofumofuni
提出日時 2021-08-25 01:51:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 455 ms / 2,000 ms
コード長 4,956 bytes
コンパイル時間 2,147 ms
コンパイル使用メモリ 208,908 KB
最終ジャッジ日時 2025-01-24 01:49:24
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,n) for(ll i=0;i<n;i++)
#define repl(i,l,r) for(ll i=(l);i<(r);i++)
#define per(i,n) for(ll i=(n)-1;i>=0;i--)
#define perl(i,r,l) for(ll i=r-1;i>=l;i--)
#define fi first
#define se second
#define pb push_back
#define ins insert
#define pqueue(x) priority_queue<x,vector<x>,greater<x>>
#define all(x) (x).begin(),(x).end()
#define CST(x) cout<<fixed<<setprecision(x)
#define rev(x) reverse(x);
using ll=long long;
using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using pl=pair<ll,ll>;
using vpl=vector<pl>;
using vvpl=vector<vpl>;
const ll MOD=1000000007;
const ll MOD9=998244353;
const int inf=1e9+10;
const ll INF=4e18;
const ll dy[8]={1,0,-1,0,1,1,-1,-1};
const ll dx[8]={0,1,0,-1,1,-1,1,-1};
template <typename T> inline bool chmax(T &a, T b) {
  return ((a < b) ? (a = b, true) : (false));
}
template <typename T> inline bool chmin(T &a, T b) {
  return ((a > b) ? (a = b, true) : (false));
}

struct UnionFind {
    vector<int> par;
    vector<int> edge;
    
    UnionFind(int n) : par(n, -1),edge(n, 0) {}

    int root(int x) {
        if (par[x] < 0) return x;
        else return par[x] = root(par[x]);
    }
    
    bool same(int x, int y) {
        return root(x) == root(y);
    }
    
    bool merge(int x, int y) {
        x = root(x); y = root(y);
        if (x == y) {
            edge[x]++;
            return false;
        }
        if (par[x] > par[y]) swap(x, y);
        par[x] += par[y];
        par[y] = x;
        edge[x] += edge[y]+1;
        return true;
    }
    
    int size(int x) {
        return -par[root(x)];
    }
};


struct iv{
  ll left,right,idx;
  bool operator<(const iv &b) const{
    return left<b.left;
  }
  bool operator==(const iv &b) const{
    return left==b.left&&right==b.right;
  }
};

struct Intervals{//開区間を扱う
	set<iv> st;
	Intervals(){
    iv rt;
    rt.left=-INF,rt.right=-INF;
		st.insert(rt);
    rt.left=INF,rt.right=INF;
		st.insert(rt);
	}
  bool overlap(iv p,iv q){
    //もし[x,y)と[y,z)を統合したくないなら後ろの等号を外す
    return (p.left<=q.left&&q.left<p.right)||(q.left<=p.left&&p.left<q.right);
  }
	ll insert(iv p){//開区間pを挿入し、マージする　帰り値はoverlapするようなiの個数
		auto x=st.lower_bound(p);x--;
    ll ret=0;iv org=p;
		while(true){
			auto q=*x;
			if(overlap(p,q)){
        ret+=min(q.right,org.right)-max(q.left,org.left);
				p.left=min(p.left,q.left);
				p.right=max(p.right,q.right);
				x=st.erase(x);
			}
			else if(q.right<=p.left)x++;
			else break;
		}
		st.insert(p);
    return ret;
	}
	vl split(iv p){//区間pをちょうど取り除く 該当する区間がない場合は何も起こらない　[x,x)を入力すると、そこで区間を分割するが、幅0の区間は捨てられる。
		auto x=st.lower_bound(p);x--;
		vector<iv> cand;
    vector<ll> ids;
		while(true){
			auto q=*x;
			if(overlap(p,q)){
				if(q.left<=p.left){
					if(q.left!=p.left)cand.push_back({q.left,p.left});
				}
				if(p.right<=q.right){
					if(p.right!=q.right)cand.push_back({p.right,q.right});
				}
        ids.push_back(x->idx);
				x=st.erase(x);
			}
			else if(q.right<=p.left)x++;
			else break;
		}
		for(auto q:cand)st.insert(q);
    return ids;
	}
  iv get(ll i){//iを含む区間を返す　もしそのような区間がなければ、{-INF,INF}を返す
		auto x=st.upper_bound({i,INF});x--;
		if(x->left<=i&&i<x->right)return *x;
		else {
      iv ret;ret.left=-INF,ret.right=INF;
      return ret;
    }
	}
	bool same(ll l,ll r){//lとrが同じ区間に属しているか　
    if(get(l).left==-INF)return false;
		else return get(l)==get(r);
	}
  ll mex(iv p){//区間pに含まれ、stに含まれないようなiのうち、もっとも左の座標　区間pが全て埋まっているなら、INFを返す
		ll left=p.left;
		while(left<p.right){
			auto to=get(left);
			if(to.left==-INF)break;
			left=to.right;
		}
		if(left<p.right)return left;
		else return INF;
	}
	void view(){
		for(auto p:st){
			if(abs(p.left)==INF)continue;
			cout << "[" <<p.left <<"," << p.right <<"), ";
		}
		cout << endl;
	}
  vector<iv> enumerate(iv p){//区間pとかぶる区間を列挙
    auto x=st.lower_bound(p);x--;
		vector<iv> cand;
		while(true){
			auto q=*x;
			if(overlap(p,q)){
        cand.push_back(q);
        x++;
			}
			else if(q.right<=p.left)x++;
			else break;
    }
    return cand;
  }
};

int main(){
  ll n,a,b;cin >> n >> a >> b;
  UnionFind uf(n);
  Intervals ivs;
  rep(i,n){
    ll v;cin >> v;
    iv p;p.left=v-b,p.right=v-a+1;
    auto cand=ivs.enumerate(p);
    //for(auto q:cand)cout << q.left <<" " << q.right <<" ";cout << endl;
    for(auto q:cand)uf.merge(i,q.idx);
    for(auto q:cand)ivs.split(q);
    if(cand.size()){
      ivs.insert({cand.front().left,cand.back().right,cand.front().idx});
    }
    ivs.insert({v,v+1,i});
    //ivs.view();
  }
  rep(i,n)cout << uf.size(i) << endl;
}
0