【bzoj2229】[Zjoi2011]最小割

2229: [Zjoi2011]最小割

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 2527  Solved: 898
[Submit][Status][Discuss]

Description

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

Input

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

Output

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

Sample Input

1
5 0
1
0

Sample Output

10

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

HINT

Source

Day1
[Submit][Status][Discuss]

最小割树模板题系列

代码：

#include<cstdio>
#include<vector>
#include<queue>
#include<ctime>
#include<algorithm>
#include<cstdlib>
#include<stack>
#include<cstring>
#include<cmath>
using namespace std;

typedef long long LL;

const int INF = 2147483647;
const int maxm = 30000;
const int maxn = 160;

struct edge{
int to,cap,flow;
}e[maxm];

struct data{
int to,w;
};

vector<int> G[maxn];
vector<data> p[maxn];
int n,m,s,t,N,now,tot,ans;
int dis[maxn],Q[maxn],cur[maxn],vis[maxn];
int fa[maxn],faw[maxn],Vis[maxn];

inline LL getint()
{
LL ret = 0,f = 1;
char c = getchar();
while (c < '0' || c > '9')
{
if (c == '-') f = -1;
c = getchar();
}
while (c >= '0' && c <= '9')
ret = ret * 10 + c - '0',c = getchar();
return ret * f;
}

inline bool bfs()
{
memset(dis,-1,sizeof(dis));
memset(cur,0,sizeof(cur));
int head = 0,tail = 0;
Q[++tail] = s; dis[s] = 0;
++now;
while (head < tail)
{
int u = Q[++head];
vis[u] = now;
for (int i = 0; i < G[u].size(); i++)
{
edge E = e[G[u][i]];
if (dis[E.to] != -1 || E.flow == E.cap) continue;
dis[E.to] = dis[u] + 1;

Q[++tail] = E.to;
}
}
return dis[t] != -1;
}

inline int dinic(int u,int a)
{
if (!a || u == t) return a;
int flow = 0;
for (int &i = cur[u]; i < G[u].size(); i++)
{
edge &E = e[G[u][i]];
if (E.flow == E.cap || dis[E.to] != dis[u] + 1) continue;
int f = dinic(E.to,min(a,E.cap - E.flow));
E.flow += f;
flow += f;
a -= f;
e[G[u][i] ^ 1].flow -= f;
if (!a) break;
}
return flow;
}

inline void init()
{
for (int i = 0; i <= tot; i += 2) e[i].flow = 0;
for (int i = 1; i <= tot; i += 2) e[i].flow = e[i].cap;
}

inline int maxflow(int S,int T)
{
init();
int ret = 0;
s = S; t = T;
while (bfs())
ret += dinic(s,INF);
return ret;
}

inline void add(int u,int v,int cap)
{
e[++tot] = (edge){v,cap,0};
G[u].push_back(tot);
e[++tot] = (edge){u,cap,cap};
G[v].push_back(tot);
}

inline int dfs(int u,int a,int cap)
{
int ret = 0;
Vis[u] = 1;
if (a <= cap) ret++;
for (int i = 0; i < p[u].size(); i++)
{
int v = p[u][i].to,w = p[u][i].w;
if (Vis[v]) continue;
ret += dfs(v,min(a,w),cap);
}
return ret;
}

int main()
{
int test = getint();
while (test--)
{
now = 0; tot = -1;
memset(e,0,sizeof(e));
memset(vis,0,sizeof(vis));
n = getint(); m = getint();
for (int i = 1; i <= n; i++) G[i].clear();
for (int i = 1; i <= m; i++)
{
int u = getint(),v = getint(),w = getint();
add(u,v,w); add(v,u,w);
}
for (int i = 2; i <= n; i++) fa[i] = 1;
for (int i = 2; i <= n; i++)
{
faw[i] = maxflow(i,fa[i]);
for (int j = i + 1; j <= n; j++)
if (fa[i] == fa[j] && vis[i] == vis[j])
fa[j] = i;
}
int q = getint();
for (int i = 1; i <= n; i++) p[i].clear();
for (int i = 2; i <= n; i++)
{
p[i].push_back((data){fa[i],faw[i]});
p[fa[i]].push_back((data){i,faw[i]});
}
for (int j = 1; j <= q; j++)
{
int c = getint();
ans = 0;
for (int i = 1; i <= n; i++)
{
memset(Vis,0,sizeof(Vis));
ans += dfs(i,INF,c);
}
printf("%d\n",ans >> 1);
}
putchar('\n');
}
return 0;
}