uva 11865 - Stream My Contest（二分+有向最小生成树）

题目链接：uva 11865 - Stream My Contest

二分带宽，然后判断最小生成树是否小于cost值。

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int maxn = 105;

struct DMST {
	struct Edge {
		int u, v, d, b;
		Edge(int u = 0, int v = 0, int d = 0, int b = 0): u(u), v(v), d(d), b(b) {}
		bool operator < (const Edge& u) const { return d < u.d; }
	};

	int in[maxn], pre[maxn], id[maxn], vis[maxn];
	vector<Edge> edges, saves;


	void init () { saves.clear(); }
	void addEdge(int u, int v, int d, int b) { saves.push_back(Edge(u, v, d, b)); }

	bool directedMST(int root, int n, int cost, int bandwith) {
		ll ans = 0;
		edges = saves;
		while (true) {
			// 删除自环+找最小入边
			//memset(in, inf, sizeof(in) * n);
			memset(in, inf, sizeof(in));
			for (int i = 0; i < edges.size(); i++) {
				int u = edges[i].u, v = edges[i].v;
				if (edges[i].d < in[v] && u != v && edges[i].b >= bandwith) {
					pre[v] = u;
					in[v] = edges[i].d;
				}
			}

			// 判断是否可以到达所有节点
			for (int i = 0; i < n; i++) {
				if (i == root) continue;
				if (in[i] == inf) return false;
			}

			int cntnode = 0;
			//memset(id, -1, sizeof(int) * n);
			//memset(vis, -1, sizeof(int) * n);
			memset(id, -1, sizeof(id));
			memset(vis, -1, sizeof(vis));
			in[root] = 0;

			for (int i = 0; i < n; i++) {
				ans += in[i];
				int v = i;
				while (vis[v] != i && id[v] == -1 && v != root) {
					vis[v] = i;
					v = pre[v];
				}
				if (v != root && id[v] == -1) {
					for (int u = pre[v]; u != v; u = pre[u])
						id[u] = cntnode;
					id[v] = cntnode++;
				}
			}

			// 没有找到环, 终止
			if (cntnode == 0) break;

			// 缩点，重新标记
			for (int i = 0; i < n; i++) {
				if (id[i] == -1)
					id[i] = cntnode++;
			}

			for (int i = 0; i < edges.size(); i++) {
				int v = edges[i].v;
				edges[i].u = id[edges[i].u];
				edges[i].v = id[edges[i].v];
				if (edges[i].u != edges[i].v)
					edges[i].d -= in[v];
			}
			n = cntnode;
			root = id[root];
		}
		return ans <= cost;
	}
}solver;


int main () {
	int cas, n, m, cost, u, v, c, b;
	scanf("%d", &cas);
	while (cas--) {
		int l = inf, r = 0;
		solver.init();
		scanf("%d%d%d", &n, &m, &cost);
		while (m--) {
			scanf("%d%d%d%d", &u, &v, &b, &c);
			solver.addEdge(u, v, c, b);
			r = max(r, b);
			l = min(l, b);
		}
		if (!solver.directedMST(0, n, cost, l))
			printf("streaming not possible.\n");
		else {
			while (l < r - 1) {
				int mid = (l + r) >> 1;
				if (!solver.directedMST(0, n, cost, mid))
					r = mid;
				else
					l = mid;
			}
			printf("%d kbps\n", l);
		}
	}
	return 0;
}