牛客多校__Explorer

https://ac.nowcoder.com/acm/contest/888/E

題意:給你一個無向圖,其中每條邊都有限制的大小區間可以通過,詢問從1到n總共有多少個數字可以通過(即從1到n)

題解:由於m的範圍很大,所以我不能單純的跑圖,所以我們可以用時間分治的思想,先把每個區間離散化,然後加入線段樹中,然後跑線段樹,跑的過程做並查集(即將相連的邊歸爲一類)然後就可以查詢答案了,跑完一個子樹後,要把子樹的並查集還原,不然會影響其他不同的區間的答案(並查集也要優化下,即連向最多有最多入度的類結點)

#include <bits/stdc++.h>

using namespace std;
typedef pair<int,int>P;

const int maxn = 1e5+9;

struct rt{
    int u,v,l,r;
}arr[maxn];

int h[maxn<<2];
int mp[maxn<<2];

vector<P>dat[maxn<<2];
vector<int>s[maxn<<2];
int f[maxn<<1];
int ans=0;
int n,m;
int siz[maxn<<2];

int Find(int x){
    while(x!=f[x]){
        x=f[x];
    }
    return x;
}

void Union(int k,int x,int y){
    x=Find(x);y=Find(y);
    if(x==y)return;
    if(siz[x]>siz[y])swap(x,y);
    f[x]=y;
    siz[y]++;
    s[k].push_back(x);
}

void solve(int k){
    for(int i=0;i<dat[k].size();i++){
        Union(k,dat[k][i].first,dat[k][i].second);
    }
}

void del(int k){
    for(int i=0;i<s[k].size();i++){
        f[s[k][i]]=s[k][i];
    }
}


void dfs(int l,int r,int k){
    solve(k);
    if(l==r){
        if(Find(1)==Find(n)){
//            cout<<"r+1 "<<mp[r+1]<<" "<<"l "<<mp[l]<<endl;
//            cout<<mp[r+1]-mp[l]<<endl;
            ans+=mp[r+1]-mp[l];
        }
        del(k);
        return;
    }
    int mid=(l+r)>>1;
    dfs(l,mid,k<<1);
    dfs(mid+1,r,k<<1|1);
    del(k);
}

void updata(int a,int b,int l,int r,int k,int u,int v){
    if(a<=l&&r<=b){
        dat[k].push_back(P(u,v));
        return;
    }
    int mid=(l+r)>>1;
    if(a<=mid)updata(a,b,l,mid,k<<1,u,v);
    if(b>mid)updata(a,b,mid+1,r,k<<1|1,u,v);
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)f[i]=i;
    int tot=0;
    for(int i=1;i<=m;i++){
        scanf("%d%d%d%d",&arr[i].u,&arr[i].v,&arr[i].l,&arr[i].r);
        h[tot++]=arr[i].l;
        h[tot++]=arr[i].r+1;
    }
    sort(h,h+tot);
    int num=unique(h,h+tot)-h;
    for(int i=0;i<num;i++){
        mp[i+1]=h[i];
    }
    for(int i=1;i<=m;i++){
        int l=lower_bound(h,h+num,arr[i].l)-h+1;
        int r=lower_bound(h,h+num,arr[i].r+1)-h+1;
        updata(l,r-1,1,num-1,1,arr[i].u,arr[i].v);
    }
    dfs(1,num-1,1);
    printf("%d\n",ans);
    return 0;
}

 

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