hdu 5739 Fantasia (2016多校第二场1006)

本人解法是CDQ+并查集。

折半的时候我们用非路径压缩的并查集来维护，但是需要用深度来维护并查集使得复杂度减小很多，接下来就先连后一半的边然后分治前一半，然后撤销后一半的边，然后枚举前一半然后递归后一半

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=200005;
const int mod=1000000007;
typedef  long long LL;
vector<int>g[maxn];
int vis[maxn];
int fa[maxn];
int su[maxn];
int w[maxn];
int ans;
int sum;
int find(int p){
    if(fa[p]==p) return p;
    return find(fa[p]);
}
int deap[maxn];
void merg(int a,int b){
    a=find(a);
    b=find(b);
    if(a==b) return;
    sum-=(su[a]+su[b])%mod;
    sum%=mod;
    if(deap[a]>deap[b]){
        su[a]=((LL)su[b]*su[a])%mod;
        fa[b]=a;
        sum+=su[a];
        deap[a]=max(deap[a],deap[b]+1);
    }
    else{
        su[b]=((LL)su[b]*su[a])%mod;
        fa[a]=b;
        sum+=su[b];
        deap[b]=max(deap[b],deap[a]+1);
    }
    sum%=mod;
}
struct pi{
    int id;
    int su;
    int de;
}pp;
void cdq(int l,int r){
    if(l==r){
        ans+=((LL)l*(sum-w[l]))%mod;
        ans%=mod;
        return;
    }
    int mid=(l+r)/2;
    vector<pi>gg;
    gg.clear();
    int pre=sum;
    for(int i=mid+1;i<=r;i++){
        for(int v:g[i]){
            if(v<l||v>mid){
                int a=find(v);
                int b=find(i);
                if(a!=b){
                    pp.id=a;
                    pp.su=su[a];
                    pp.de=deap[a];
                    gg.push_back(pp);
                    pp.id=b;
                    pp.su=su[b];
                    pp.de=deap[b];
                    gg.push_back(pp);
                    merg(a,b);
                }
            }
        }
    }
    cdq(l,mid);
    int p=gg.size();
    for(int i=p-1;i>=0;i--){
        pp=gg[i];
        fa[pp.id]=pp.id;
        su[pp.id]=pp.su;
        deap[pp.id]=pp.de;
    }
    sum=pre;
    gg.clear();
    for(int i=l;i<=mid;i++){
        for(int v:g[i]){
            if(v<=mid||v>r){
                int a=find(v);
                int b=find(i);
                if(a!=b){
                    pp.id=a;
                    pp.su=su[a];
                    pp.de=deap[a];
                    gg.push_back(pp);
                    pp.id=b;
                    pp.su=su[b];
                    pp.de=deap[b];
                    gg.push_back(pp);
                    merg(a,b);
                }
            }
        }
    }
    cdq(mid+1,r);
    p=gg.size();
    for(int i=p-1;i>=0;i--){
        pp=gg[i];
        fa[pp.id]=pp.id;
        su[pp.id]=pp.su;
        deap[pp.id]=pp.de;
    }
    sum=pre;
    gg.clear();
}
int main()
{
    int t,n,m;
    cin>>t;
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) g[i].clear();
        for(int i=1;i<=n;i++) scanf("%d",&w[i]);
        for(int i=1;i<=m;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            g[y].push_back(x);
        }
        sum=0;
        for(int i=1;i<=n;i++){
            sum+=w[i];
            sum%=mod;
            su[i]=w[i];
            fa[i]=i;
            deap[i]=1;
        }
        ans=0;
        cdq(1,n);
        ans=(ans+mod)%mod;
        printf("%d\n",ans);
    }
}