Atcoder Regular Contest 093 C - Bichrome Spanning Tree

給定一張圖,對圖上邊黑白染色,使得同時選擇了兩種顏色邊的最小生成樹邊權和爲X,求染色方案數。

先求出圖的\(mst\)大小,然後分三類討論:

1.\(X<mst\) 無解

2.\(X==mst\)

我們求出可以構成最小生成樹的邊集大小\(sumst\)

可以發現,在這個邊集裏,只要不是所有邊顏色相同,就一定能構造出有雙色邊的原圖\(mst\)。邊補集則可以任意染色 ;w;

方案數是\(2^{m-sumst}*(2^{sumst}-2)\)

3.\(X>mst\)

我們考慮在\(mst\)上強制加一條邊對\(mst\)的貢獻。

畫個有點醜的樹()這是某個原圖的一個\(mst\)

現在考慮強制連一條邊\((3,9)\)\(w=13\)

要讓其重新變成一棵樹,就要在\((3,9)\)這條鏈上刪去一條邊()顯然是應該刪去最大的那條,即\((1,2)\)\(w=9\)

草,搞這麼多不就一句話QAQ

一條邊(u,v)的貢獻就是\(v[i]=w[u,v]-maxw(u,v)\)

這樣我們就可以用樹剖+RMQ求得每條邊對mst的貢獻()

這樣我們可以統計出對mst的貢獻\(v[i]=X-mst\)的邊數\(sum1\)

這樣我們要使得mst邊集邊全部同色,sum1邊集至少有一邊異色,剩餘的邊補集任意染色。實際操作的時候我通過判斷w[i]-v[i]>X-mst統計了邊補集的大小sum2

方案數爲\(2^{sum2}*(2^{sum1}*2-2)\)

以及第二個分類中的sumst其實是=sum1+n-1的(很顯然吧x)

因爲ST表我不大會寫,還是寫了線段樹來着(

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#define MAXN (int)(1e3+7)
#define MAXM (int)(2e3+14)
#define mod (int)(1e9+7)
using namespace std;
int n,m;
long long X;
struct edge
{
	int x,y,z;	
}a[MAXM];
struct qwq
{
	int nex,to,w;
}e[MAXN<<1];
int h[MAXN],tot=0;
inline void add(int u,int v,int w)
{
	e[++tot].to=v;
	e[tot].nex=h[u];
	e[tot].w=w;
	h[u]=tot;
}
inline long long power(long long a,long long b)
{
	long long answer=1,base=a;
	while (b)
	{
		if (b&1)
		{
			answer*=base;
			answer%=mod;
		}
		b>>=1;
		base*=base;
		base%=mod;
	}
	return answer;
}
inline bool cmp(edge aa,edge bb) { return aa.z<bb.z; }
int fa[MAXN];
long long mst=0;
int sum3=0;
inline void INIT1() { for (int i=1;i<=n;i++) fa[i]=i; }
inline void INIT2() { for (int i=1;i<=n;i++) fa[i]=0; }
int found(int x) { if (x==fa[x]) return x; return fa[x]=found(fa[x]); }
bool book[MAXM];
inline void MST()
{
	INIT1();
	sort(a+1,a+m+1,cmp);
	for (int i=1,fx,fy;i<=m;i++)
	{
		fx=found(a[i].x); fy=found(a[i].y);
		if (fx!=fy)
		{
			fa[fx]=fy;
			add(a[i].x,a[i].y,a[i].z);
			add(a[i].y,a[i].x,a[i].z);
			mst+=a[i].z;
			sum3++;
			book[i]=1;
		}
	}
	INIT2();
}

int ans[MAXN<<2];
#define leftson cur<<1
#define rightson cur<<1|1
#define mid ((l+r)>>1)
#define push_up ans[cur]=max(ans[leftson],ans[rightson])
int ww[MAXN];
void build(int cur,int l,int r)
{
	if (l==r)
	{
		ans[cur]=ww[l];
		return;
	}
	build(leftson,l,mid);
	build(rightson,mid+1,r);
	push_up;
}
int query(int ql,int qr,int cur,int l,int r)
{
	if (ql<=l&&r<=qr) return ans[cur];
	int answ=0;
	if (ql<=mid) answ=query(ql,qr,leftson,l,mid);
	if (qr>mid) answ=max(answ,query(ql,qr,rightson,mid+1,r));
	return answ;
}
int son[MAXN],dep[MAXN],top[MAXN],siz[MAXN],id[MAXN],cnt=0;
void dfs1(int x)
{
	siz[x]=1;
	for (int i=h[x],y;i;i=e[i].nex)
	{
		y=e[i].to;
		if (y==fa[x]) continue;
		fa[y]=x;
		dep[y]=dep[x]+1;
		dfs1(y);
		siz[x]+=siz[y];
		if (siz[y]>siz[son[x]]) son[x]=y;
	}
}
void dfs2(int x,int tp)
{
	id[x]=++cnt;
	top[x]=tp;
	if (!son[x]) return;
	dfs2(son[x],tp);
	for (int i=h[x],y;i;i=e[i].nex)
	{
		y=e[i].to;
		if (y==fa[x]) continue;
		if (y==son[x]) { ww[id[son[x]]]=e[i].w; continue; }
		dfs2(y,y);
		ww[id[y]]=e[i].w;
	}
}
inline int query_tree(int x,int y)
{
	int answ=0;
	while (top[x]!=top[y])
	{
		if (dep[top[x]]<dep[top[y]]) swap(x,y);
		answ=max(answ,query(id[top[x]],id[x],1,1,n));
		x=fa[top[x]];
	}
	if (dep[x]>dep[y]) swap(x,y);
	return max(answ,query(id[x]+1,id[y],1,1,n));
}

int main()
{
	scanf("%d%d%lld",&n,&m,&X);
	for (int i=1;i<=m;i++) scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
	MST();
//	printf("MST:%lld\n",mst);
	int sum1=0,sum2=0;
	if (mst>X) { printf("0\n"); return 0; }
	dfs1(1);
	dfs2(1,1);
	build(1,1,n);
	for (int i=1,W;i<=m;i++)
	{
		if (book[i]) continue;
		W=query_tree(a[i].x,a[i].y);
//		printf("W:%d %d\n",a[i].z,W);
		if (a[i].z-W==X-mst) sum1++;
		else if (a[i].z-W>X-mst) sum2++;
	}
//	printf("sum1:%d sum2:%d sum3:%d\n",sum1,sum2,sum3);
	if (mst==X) printf("%lld\n",(power(2,sum2)*((power(2,sum1+sum3)-2+mod)%mod))%mod);
	else printf("%lld\n",((power(2,sum2)*((2*power(2,sum1)-2+mod)%mod))%mod)%mod);
	return 0;
}
/*
8 10
48
4 6 10
8 4 11
5 8 8
1 8 10
3 8 128773450
7 8 10
4 2 4
3 4 1
3 1 13
5 2 2
*/
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章