#72-【Dinic模板*2】Total flow

最新推荐文章于 2021-05-23 06:17:34 发布

原创最新推荐文章于 2021-05-23 06:17:34 发布 · 244 阅读

0 ·

CC 4.0 BY-SA版权

刷题同时被 2 个专栏收录

220 篇文章

订阅专栏

gdgzoi刷题

200 篇文章

订阅专栏

博客介绍了一道水管流量计算的模板题。输入方面，首行输入N，后续N行每行用前两个英文字母表示水管两端、后一个整数表示流量，流量不超1000；输出为总流量。还提及这是真·模板题并给出代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description

Input

第1行输入N，之后N行每行描述一条水管，前两个英文字母表示水管的两端（大小写字母是不一样的），后一个整数表示水管的流量，流量不会超过1000.

Output

一个整数，表示总流量．

Sample Input

5
A B 3
B C 3
C D 5
D Z 4
B Z 6

Sample Output

真·模板题

代码：

#include <iostream> // 模板就不用说了
#include <cstring>
#include <queue>

#define SIZE 501
#define INF 2e+09

using namespace std;

struct edge
{
	int to, cap, reverse;
};

vector<edge> graph[SIZE];
int depth[SIZE];

bool bfs(void)
{
	int u, i, v;
	queue<int> q;
	
	memset(depth, -1, sizeof (depth));
	depth['A'] = 0;
	q.push('A');
	while (!q.empty())
	{
		u = q.front();
		if (u == 'Z')
		{
			return true;
		}
		q.pop();
		for (i = 0; i < graph[u].size(); ++i)
		{
			v = graph[u][i].to;
			if ((graph[u][i].cap) && (depth[v] == -1))
			{
				depth[v] = depth[u] + 1;
				q.push(v);
			}
		}
	}
	
	return false;
}

int dfs(int u, int flow)
{
	int i, v, ret = 0, delta;
	
	if ((!flow) || (u == 'Z'))
	{
		return flow;
	}
	for (i = 0; i < graph[u].size(); ++i)
	{
		edge &temp = graph[u][i];
		v = temp.to;
		if ((depth[v] == depth[u] + 1) && (temp.cap > 0))
		{
			delta = dfs(v, min(flow - ret, temp.cap));
			if (delta > 0)
			{
				ret += delta;
				temp.cap -= delta;
				graph[v][temp.reverse].cap += delta;
				if (ret == flow)
				{
					return ret;
				}
			}
		}
	}
	
	return ret;
}

int dinic(void) // 日常Dinic
{
	int ret = 0, delta;
	
	while (bfs())
	{
		delta = dfs('A', INF);
		if (!delta)
		{
			return ret;
		}
		ret += delta;
	}
	
	return ret;
}

int main(int argc, char** argv)
{
	int m, cap;
	char u, v;
	
	scanf("%d", &m);
	while (m--)
	{
		cin >> u >> v; // 不知道为什么scanf字符会出问题,只得用cin输入
		scanf("%d", &cap);
		graph[u].push_back({v, cap, graph[v].size()});
		graph[v].push_back({u, 0, graph[u].size() + 1});
	}
	
	printf("%d", dinic());
	
	return 0;
}