2020杭电多校A Very Easy Graph Problem（图论）

题意：给n个点，m条边的图，第i条边的长度是2的i次方，求

题目

由题意，边的长度是依次递增的，根据等比数列的前n项和公式，前面n项和一定小于2的n+
1次方，所以，如果假如的两个点已经连接，就不需要建新边，这里用并查集维护一颗最小生成树。对于每一条边，我们只需要计算它左边0、1的个数和右边的相乘，即计算这一条边做的贡献。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
const int N=200005;
typedef long long ll;
const ll mod=1e9+7;
int fir[N],tot,n,m,cnt0,cnt1,a[N],f[N],sum0[N],sum1[N];
ll p[N],ans;
struct node{
	int ne,to,w;
}e[N];
int getf(int u){
	if(f[u]==u)return u;
	return f[u]=getf(f[u]);
}
void add(int u,int v,int w){
	e[tot].to=v;
	e[tot].w=w;
	e[tot].ne=fir[u];
	fir[u]=tot++;
}
void dfs(int x,int fa){
	if(a[x]==0)sum0[x]++;
	if(a[x]==1)sum1[x]++;
	for(int i=fir[x];i!=-1;i=e[i].ne){
		int v=e[i].to,w=e[i].w;
		if(v==fa)continue;
		dfs(v,x);
		sum0[x]+=sum0[v];
		sum1[x]+=sum1[v];
		ans=(ans+p[w]*sum0[v]%mod*(cnt1-sum1[v])%mod+p[w]*sum1[v]%mod*(cnt0-sum0[v])%mod)%mod;
	}
}
int main(){
	int t;
	scanf("%d",&t);
	p[0]=1;
	for(int i=1;i<=N;i++){
		p[i]=(p[i-1]*2)%mod;
	}
	while(t--){
		
		tot=0;
		cnt0=0;
		cnt1=0;
		ans=0;
		memset(sum0,0,sizeof(sum0));
		memset(sum1,0,sizeof(sum1));
		memset(fir,-1,sizeof(fir));
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++){
			f[i]=i;
		}
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			if(a[i]==0)cnt0++;
			else cnt1++;
		}
		int u,v;
		for(int i=1;i<=m;i++){
			scanf("%d%d",&u,&v);
			int t1=getf(u);
			int t2=getf(v);
			if(t1==t2)continue;
			f[t1]=t2;
			add(u,v,i);
			add(v,u,i);
		}
		dfs(1,0);
		printf("%lld\n",ans);
	}
	return 0;
}