[BZOJ2654]Tree问题：二分查找与Kruskal算法结合的优化解决方案

1 #include
2 #include
3 #define rep(i,l,r) for (int i&＃61;l; i<&＃61;r; i&＃43;&＃43;)
4 typedef long long ll;
5 using namespace std;
6
7 const int N&＃61;100100;
8 int n,m,cnt,tot,k,ans,u[N],v[N],w[N],c[N],fa[N];
9 struct E{ int u,v,w,c; }e[N];
10
11 bool operator<(E a,E b){ return a.w&＃61;&＃61;b.w ? a.cb.w; }
12 int find(int x){ return x&＃61;&＃61;fa[x] ? x : fa[x]&＃61;find(fa[x]); }
13
14 bool check(int x){
15 tot&＃61;cnt&＃61;0;
16 rep(i,1,n) fa[i]&＃61;i;
17 rep(i,1,m){
18 e[i].u&＃61;u[i]; e[i].v&＃61;v[i]; e[i].w&＃61;w[i]; e[i].c&＃61;c[i];
19 if(!c[i])e[i].w&＃43;&＃61;x;
20 }
21 sort(e&＃43;1,e&＃43;m&＃43;1);
22 rep(i,1,m){
23 int p&＃61;find(e[i].u),q&＃61;find(e[i].v);
24 if(p!&＃61;q){
25 fa[p]&＃61;q; tot&＃43;&＃61;e[i].w;
26 if (!e[i].c) cnt&＃43;&＃43;;
27 }
28 }
29 return cnt>&＃61;k;
30 }
31
32 int main(){
33 scanf("%d%d%d",&n,&m,&k);
34 rep(i,1,m) scanf("%d%d%d%d",&u[i],&v[i],&w[i],&c[i]),u[i]&＃43;&＃43;,v[i]&＃43;&＃43;;
35 int L&＃61;-105,R&＃61;105;
36 while(L<&＃61;R){
37 int mid&＃61;(L&＃43;R)>>1;
38 if(check(mid)) L&＃61;mid&＃43;1,ans&＃61;tot-k*mid; else R&＃61;mid-1;
39 }
40 printf("%d\n",ans);
41 return 0;
42 }