BZOJ2229[Zjoi2011]最小割——最小割树

题目描述

小白在图论课上学到了一个新的概念——最小割，下课后小白在笔记本上写下了如下这段话： “对于一个图，某个对图中结点的划分将图中所有结点分成两个部分，如果结点s,t不在同一个部分中，则称这个划分是关于s,t的割。 对于带权图来说，将所有顶点处在不同部分的边的权值相加所得到的值定义为这个割的容量，而s,t的最小割指的是在关于s,t的割中容量最小的割” 现给定一张无向图，小白有若干个形如“图中有多少对点它们的最小割的容量不超过x呢”的疑问，小蓝虽然很想回答这些问题，但小蓝最近忙着挖木块，于是作为仍然是小蓝的好友，你又有任务了。

输入

输入文件第一行有且只有一个正整数T，表示测试数据的组数。 对于每组测试数据， 第一行包含两个整数n,m，表示图的点数和边数。 下面m行，每行3个正整数u,v,c(1<=u,v<=n,0<=c<=106)，表示有一条权为c的无向边(u,v) 接下来一行，包含一个整数q，表示询问的个数 下面q行，每行一个整数x，其含义同题目描述。

输出

对于每组测试数据，输出应包括q行，第i行表示第i个问题的答案。对于点对(p,q)和(q,p)，只统计一次（见样例）。

两组测试数据之间用空行隔开。

样例输入

1
5 0
1
0

样例输出

10

【数据范围】
对于100%的数据 T<=10,n<=150,m<=3000,q<=30，x在32位有符号整数类型范围内。
图中两个点之间可能有多条边

 

最小割树模板题，对于每次询问暴力枚举任意两个点在最小割树上求路径上边权最小值并判断是否$\le x$即可。求路径边权最小值用一个倍增数组即可。

#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<cstdio>
#include<vector>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define pr pair<int,int>
using namespace std;
int head[200];
int to[400];
int val[400];
int next[400];
int dep[200];
int tot;
int f[200][10];
int g[200][10];
int n,m,k;
int u,v,w;
int id[200];
int now;
void add_edge(int u,int v,int w)
{
	next[++tot]=head[u];
	head[u]=tot;
	to[tot]=v;
	val[tot]=w;
}
namespace DINIC
{
	int tot;
	int S,T;
	int q[200];
	int d[200];
	int head[200];
	int next[6010];
	int val[6010];
	int col[200];
	int to[6010];
	int vir[6010];
	void init()
	{
		tot=1;
		memset(head,0,sizeof(head));
		memset(val,0,sizeof(val));
		memset(col,0,sizeof(col));
		memset(vir,0,sizeof(vir));
	}
	void add(int u,int v,int w)
	{
		next[++tot]=head[u];
		head[u]=tot;
		to[tot]=v;
		val[tot]=w;
		vir[tot]=w;
	}
	bool bfs(int S,int T)
	{
		int l=0,r=0;
		memset(d,-1,sizeof(d));
		q[r++]=S;
		d[S]=0;
		while(l<r)
		{
			int now=q[l];
			l++;
			for(int i=head[now];i;i=next[i])
			{
				if(d[to[i]]==-1&&val[i])
				{
					d[to[i]]=d[now]+1;
					q[r++]=to[i];
				}
			}
		}
		if(d[T]==-1)
		{
			return false;
		}
		else
		{
			return true;
		}
	}
	int dfs(int x,int maxflow)
	{
		if(x==T)
		{
			return maxflow;
		}
		int nowflow;
		int used=0;
		for(int i=head[x];i;i=next[i])
		{
			if(d[to[i]]==d[x]+1&&val[i])
			{
				nowflow=dfs(to[i],min(maxflow-used,val[i]));
				val[i]-=nowflow;
				val[i^1]+=nowflow;
				used+=nowflow;
				if(nowflow==maxflow)
				{
					return maxflow;
				}
			}
		}
		if(used==0)
		{
			d[x]=-1;
		}
		return used;
	}
	int dinic()
	{
		for(int i=2;i<=tot;i++)
		{
			val[i]=vir[i];
		}
		int res=0;
		while(bfs(S,T)==true)
		{
			res+=dfs(S,0x3f3f3f3f);
		}
		return res;
	}
	void find(int x)
	{
		col[x]=now;
		for(int i=head[x];i;i=next[i])
		{
			if(col[to[i]]!=now&&val[i])
			{
				find(to[i]);
			}
		}
	}
	void build(int l,int r)
	{
		if(l>=r)
		{
			return ;
		}
		S=id[l],T=id[l+1];
		int cut=dinic();
		now++;
		find(S);
		int L=l,R=r;
		for(int i=l;i<=r;i++)
		{
			if(col[id[i]]==now)
			{
				q[L++]=id[i];
			}
			else
			{
				q[R--]=id[i];
			}
		}
		for(int i=l;i<=r;i++)
		{
			id[i]=q[i];
		}
		add_edge(S,T,cut);
		add_edge(T,S,cut);
		build(l,L-1);
		build(R+1,r);
	}
};
void dfs(int x)
{
	for(int i=1;i<=8;i++)
	{
		f[x][i]=f[f[x][i-1]][i-1];
		g[x][i]=min(g[x][i-1],g[f[x][i-1]][i-1]);
	}
	for(int i=head[x];i;i=next[i])
	{
		if(to[i]!=f[x][0])
		{
			dep[to[i]]=dep[x]+1;
			f[to[i]][0]=x;
			g[to[i]][0]=val[i];
			dfs(to[i]);
		}
	}
}
int lca(int x,int y)
{
	int res=1<<30;
	if(dep[x]<dep[y])
	{
		swap(x,y);
	}
	int d=dep[x]-dep[y];
	for(int i=0;i<=8;i++)
	{
		if(d&(1<<i))
		{
			res=min(res,g[x][i]);
			x=f[x][i];
		}
	}
	if(x==y)
	{
		return res;
	}
	for(int i=8;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			res=min(res,g[x][i]);
			res=min(res,g[y][i]);
			x=f[x][i];
			y=f[y][i];
		}
	}
	res=min(min(g[x][0],g[y][0]),res);
	return res;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		memset(head,0,sizeof(head));
		memset(dep,0,sizeof(dep));
		memset(val,0,sizeof(val));
		memset(f,0,sizeof(f));
		memset(g,0,sizeof(g));
		now=0;
		tot=0;
		DINIC::init();
		scanf("%d%d",&n,&m);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			DINIC::add(u,v,w);
			DINIC::add(v,u,w);
		}
		for(int i=1;i<=n;i++)
		{
			id[i]=i;
		}
		DINIC::build(1,n);
		dfs(1);
		scanf("%d",&k);
		while(k--)
		{
			scanf("%d",&u);
			int ans=0;
			for(int i=1;i<=n;i++)
			{
				for(int j=i+1;j<=n;j++)
				{
					if(lca(i,j)<=u)
					{
						ans++;
					}
				}
			}
			printf("%d\n",ans);
		}
		printf("\n");
	}
}