非严格次小生成树+严格次小生成树

非严格次小生成树+严格次小生成树

非严格次小生成树:首先使用最小生成树算法将最小生成树求出来,将生成树建图,然后用倍增维护树上的最大值。 考虑将所有的非树边一次加入最小生成树,并将新边与最小生成树形成的环上最大的边给删了,记录答案,然后答案最小值即为非严格次小生成树。

这个代码并没有求非严格次小生成树,但是除了没有记录答案以外,其他都一样

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstring>
#define int long long 
using namespace std;
const int maxn=2e5+10;
const int mod=1e9+7;
int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int t,n,m,fir[maxn],nxt[maxn<<1],to[maxn<<1],val[maxn<<1],tot;//此处邻接表存储最小生成树
int f[maxn],fa[maxn][25],maxx[maxn][25],sum,k,X,dep[maxn];
struct node{
	int x,y,val,flag;
}e[maxn]; 
inline bool cmpl(node a,node b){
	return a.val<b.val;
}
int get(int x){
	if(x==f[x]) return x;
	f[x]=get(f[x]);
	return f[x];
}
void add(int x,int y,int z){
	nxt[++tot]=fir[x];fir[x]=tot;to[tot]=y;val[tot]=z;
	nxt[++tot]=fir[y];fir[y]=tot;to[tot]=x;val[tot]=z;
}
void dfs(int x){
	for(int i=0;i<=20;i++)
		fa[x][i+1]=fa[fa[x][i]][i];
	for(int i=0;i<=20;i++)
		maxx[x][i+1]=max(maxx[x][i],maxx[fa[x][i]][i]);
	for(int i=fir[x];i;i=nxt[i]){
		int y=to[i];if(y==fa[x][0]) continue;
		fa[y][0]=x;dep[y]=dep[x]+1;
		maxx[y][0]=val[i];dfs(y);
	}
}
int query(int x,int y){
	int shawn=0;
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=21;i>=0;i--){
		if(dep[fa[x][i]]>=dep[y]){
			shawn=max(shawn,maxx[x][i]);
			x=fa[x][i];
		}
	}
	if(x==y) return shawn;
	for(int i=21;i>=0;i--){
		if(fa[x][i]!=fa[y][i]){
			shawn=max(shawn,max(maxx[x][i],maxx[y][i]));
			x=fa[x][i];y=fa[y][i];
		}
	}
	return max(shawn,max(maxx[x][0],maxx[y][0]));
}
int ksm(int x,int y){
	int tot=1;
	while(y){
		if(y&1) tot=(tot*x)%mod;
		x=x*x%mod;y>>=1;
	}
	return tot%mod;
}
signed main(){
	freopen("4.in","r",stdin);
	t=read();
	while(t--){
		memset(fir,0,sizeof(fir));
		memset(fa,0,sizeof(fa));
		memset(dep,0,sizeof(dep));
		memset(maxx,0,sizeof(maxx));
		n=read();m=read();X=read();
		for(int i=1;i<=m;i++){
			e[i].x=read();e[i].y=read();e[i].val=read();e[i].flag=0;
		}
		sort(e+1,e+1+m,cmpl);
		k=0;sum=0;
		for(int i=1;i<=n;i++) f[i]=i;
		for(int i=1,x,y,fx,fy;i<=m;i++){
			x=e[i].x;y=e[i].y;
			fx=get(x);fy=get(y);
			if(fx!=fy){
				f[fx]=fy;sum+=e[i].val;k++;
				e[i].flag=1;
				add(x,y,e[i].val);
			}
			if(k==n-1) break;
		}
		dfs(1);int now,equ=0,les=0;
		for(int i=1;i<=m;i++){
			if(!e[i].flag){
				now=sum+e[i].val-query(e[i].x,e[i].y);
				if(now==X) equ++;
				else if(now<X)les++;
			}
		}
		if(X==sum){
			printf("%lld\n",((ksm(2,m)-2*ksm(2,m-n+1-equ))%mod+mod)%mod);
		}
		else{
			printf("%lld\n",(2*(ksm(2,m-n+1-les)-ksm(2,m-n+1-les-equ))%mod)%mod);
		}
	}
	return 0;
}

严格次小生成树

与非严格次小生成树的唯一区别就是求了最小生成树上的次大值,倍增维护。

现在唯一的难点就是如何倍增维护次大值,用 d i s 1 [ i ] [ x ] dis1[i][x] dis1[i][x] 表示最大值, d i s 2 [ i ] [ x ] dis2[i][x] dis2[i][x] 表示次大值, f a [ i ] [ x ] fa[i][x] <

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值