HDU 4253 Two Famous Companies（最小生成树+二分）



题意：就是说有A、B两个公司要修路，有m条路，可能是属于A修的，也可能是属于B修的，现在要求所有路都联通的情况下的最小权值，并且A公司必须要修k条路。

思路：一开始看了半天没思路...百度了下发现是一道clj的论文题，

思路是用f(x)表示A公司每条路加上x后图中最小生成树中最多含有的A公司的边数。

用g(x)表示A公司每条路加上x后图中最小生成树中最少含有的A公司的边数。

二分x，如果k在某个f(x)和g(x)之间那么此时这种情况就是我们要求的解，这个可以反证，如果不是最优解那么最优解的每条A边加上x代回这个子问题，一定比求得最小生成树权值和小，这样就推出矛盾。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<set>
#include<ctime>
#define eps 1e-6
#define LL long long
#define pii pair<int, int>
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

const int MAXN = 100010;
struct Edge {
    int u, v, w, id;
    Edge(int u=0, int v=0, int w=0, int id=0) : u(u), v(v), w(w), id(id) {} 
    bool operator < (const Edge& e) const {
		return w < e.w;
    }
} e0[MAXN], e1[MAXN];
int n, m, k, ans, cnt0, cnt1, pa[MAXN];
int Find(int x) {
	if(pa[x] == x) return x;
	return pa[x] = Find(pa[x]);
}
int kruscal(int x) {
	for(int i = 0; i < n; i++) pa[i] = i;
	int len0 = 0, len1 = 0, ret = 0;
	Edge tmp;
	for(int i = 1; i <= m; i++) {
		if(len0>=cnt0 || len1<cnt1&&e1[len1].w<e0[len0].w+x) tmp = e1[len1++];
		else tmp = e0[len0++];
		int p = Find(tmp.u), q = Find(tmp.v);
		if(p != q) {
			ans += tmp.w;
			pa[p] = q;
			if(!tmp.id) ret++, ans+=x;
		}
	}
	return ret;
}
int main() {
    //reopen("input.txt", "r", stdin);
    int kase = 0;
	while(scanf("%d%d%d", &n, &m, &k) == 3) {
		cnt0 = cnt1 = 0;
		for(int i = 0; i < m; i++) {
			int u, v, w, id;
			scanf("%d%d%d%d", &u, &v, &w, &id);
			if(!id) e0[cnt0++] = Edge(u, v, w, id);
			else e1[cnt1++] = Edge(u, v, w, id);
		}
		sort(e0, e0+cnt0);
		sort(e1, e1+cnt1);
		int le = -100, ri = 100;
		while(le <= ri) {
			int mid = le + (ri-le)/2;
			ans = 0; int t1 = kruscal(mid+1);
			ans = 0; int t2 = kruscal(mid);
 			if(t1<=k && t2>=k) {
				ans -= mid*k;
				break;
			}
			else if(t1 > k) le = mid + 1;
			else ri = mid - 1;
		}
		printf("Case %d: %d\n", ++kase, ans);
	}
    return 0;
}