2019牛客多校第八場E(線段樹+可撤銷並查集)

題目鏈接:https://ac.nowcoder.com/acm/contest/888/E

思路:線段樹維護區間,然後從根節點開始dfs,用並查集維護連通性,回溯時撤銷並查集的操作,注意離散化的時候把右端店++,這樣維護的是一個左閉右開的區間,可以避免很多問題的發生

#pragma GCC optimize(3)
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%lld",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod ll(1000)
#define pb push_back
#define eps 1e-8
#define lc d<<1
#define rc d<<1|1
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
int fa[200008],sz[200008],n,ans,x,y,q,w,m,l[100008],r[100008],c[300008],qw;
int fid(int x) {return (fa[x]^x)?fid(fa[x]):x;}
struct as{
    int l,r;
    vector<int>g;
}tr[200008<<2];
P a[100008];
void build(int d,int l,int r)
{
    tr[d].l=l,tr[d].r=r;
    if(l==r)  return;
    int mid=(l+r)>>1;
    build(lc,l,mid);
    build(rc,mid+1,r);
}
void add(int d,int l,int r,int pos)
{
    if(tr[d].l==l&&tr[d].r==r)  {tr[d].g.pb(pos);return;}
    int mid=(tr[d].l+tr[d].r)>>1;
    if(mid>=r)  add(lc,l,r,pos);
    else if(l>mid)  add(rc,l,r,pos);
    else  add(lc,l,mid,pos),add(rc,mid+1,r,pos);
}
void query(int d)
{
    vector<int>st;
    for(int i=0;i<tr[d].g.size();i++)
    {
        int x=tr[d].g[i];
        int fx=fid(a[x].first),fy=fid(a[x].second);
        if(fx==fy) continue;
        if(sz[fx]<sz[fy]) swap(fx,fy);
        fa[fy]=fx,sz[fx]+=sz[fy];
        st.pb(fy);
    }
    if(fid(1)==fid(n)) ans+=c[tr[d].r+1]-c[tr[d].l];
    else if(tr[d].l!=tr[d].r) query(lc),query(rc);
    for(int i=st.size()-1;i>=0;i--) sz[fa[st[i]]]-=sz[st[i]],fa[st[i]]=st[i];
    st.clear();
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m;
    FOR(i,1,n) fa[i]=i,sz[i]=1;
    FOR(i,1,m)
    {
        si(a[i].first),si(a[i].second),si(l[i]),si(r[i]);
        r[i]++,c[++qw]=l[i],c[++qw]=r[i];
    }
    sort(c+1,c+qw+1);
    qw=unique(c+1,c+qw+1)-c-1;
    build(1,1,qw-1);
    FOR(i,1,m) add(1,lower_bound(c+1,c+qw+1,l[i])-c,lower_bound(c+1,c+qw+1,r[i])-c-1,i);
    query(1);
    cout<<ans<<endl;
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章