【BZOJ 2561】最小生成树 最小割

拿到题再一看数据范围怎么也是思考生成树把，居然是网络流qaq。。。

其实只要是换一个角度思考还是可以的，一棵树满足什么条件，至少就是任意两个节点之间有一条简单路径，而现在题目要求有不联通，那么再最优的情况下，至少应该有两个集合满足互相之间不联通，这样就能想到网络流的最小割模型了。

至于网络流怎么构建模型，这条边（最小生成树为例）可能用上的情况在于：当且只有当比它小的边无法构成一颗最小生成树，即无法将这条边所链接的两个节点联通，他才可能用上，所以先把所有比它小的边建图，然后以这条边两边的节点为源点和汇点，跑最大流（最小割）。至于最大生成树的话同理，然后又由于一部分是大于一部分是小于没有交集所以直接暴力将两部分的答案直接加起来就好了。

#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 400021
#define inf 1000000000
using namespace std;
int n,m,ans,head[maxn],last[maxn],q[maxn*2],tot,s,h[maxn],t,len;
struct edge{int v,w,next;}e[2*maxn];
void adde(int a,int b,int c){
	e[tot].v=b,e[tot].w=c,e[tot].next=head[a];head[a]=tot++;
	e[tot].v=a,e[tot].w=0,e[tot].next=head[b];head[b]=tot++;
}
struct data{
	int a,b,c;
}nod[maxn*2];
bool bfs(){  
    for(int i=0;i<=n;i++)h[i]=-1;  
    h[s]=0;int l=0,r=1;q[0]=s;  
    while(l<r){  
        int u=q[l++];  
        for(int v,i=head[u];i!=-1;i=e[i].next){  
            if(h[v=e[i].v]==-1&&e[i].w){  
                h[v]=h[u]+1;  
                q[r++]=v;  
            }  
        }  
    }  
    return h[t]!=-1;  
}
int dfs(int u,int f){
	if(u==t||f==0)return f;
	int used=0,w;
	for(int v,i=last[u];i!=-1;i=e[i].next)
		if(e[i].w&&h[v=e[i].v]==h[u]+1){
			last[u]=i;
			w=min(f-used,e[i].w);
			w=dfs(v,w);
			used+=w;e[i].w-=w,e[i^1].w+=w;
			if(used==f)return f;
		}
	if(!used)h[u]=-1;
	return used;
}
void dinic(){
	while(bfs()){
		for(int i=1;i<=n;i++)last[i]=head[i];
		ans+=dfs(s,inf);
	}
}
void build(int pos){
	memset(head,-1,sizeof(head));
	tot=0;
	for(int i=1;i<=m;i++){
		if(pos==1){
			if(nod[i].c<len)
			adde(nod[i].b,nod[i].a,1),
			adde(nod[i].a,nod[i].b,1);
		}else{
			if(nod[i].c>len)
			adde(nod[i].b,nod[i].a,1),
			adde(nod[i].a,nod[i].b,1);
		}
	}
	dinic();
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)scanf("%d%d%d",&nod[i].a,&nod[i].b,&nod[i].c);
	scanf("%d%d%d",&s,&t,&len);
	build(0);
	build(1);
	printf("%d",ans);
	return 0;
}