2019牛客多校第八场E(线段树+可撤销并查集)

题目链接:https://ac.nowcoder.com/acm/contest/888/E

思路:线段树维护区间,然后从根节点开始dfs,用并查集维护连通性,回溯时撤销并查集的操作,注意离散化的时候把右端店++,这样维护的是一个左闭右开的区间,可以避免很多问题的发生

#pragma GCC optimize(3)
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
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
#define P pair
#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;
    vectorg;
}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)
{
    vectorst;
    for(int i=0;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<

 

你可能感兴趣的:(数据结构)