BZOJ 4349

最小树形图模板

#include<bits/stdc++.h>
#define N 550
#define M 10100
using namespace std;
int n,m,f,len;
int t[N],done[N];
int rs[N],pla[N];
int l[M],r[M];
double va[M],zx[N],ans;
inline void Add(int a,int b,double c){
	l[++len]=a,r[len]=b,va[len]=c;
}
double zxsxt(){
	double tot=0,use;
	int cnt=n;
	while(1){
		memset(done,0,sizeof(done));
		memset(rs,0,sizeof(rs));
		for(int i=1;i<=cnt;i++) zx[i]=1008610086;
		for(int i=1;i<=len;i++){
			if(va[i]<zx[r[i]]&&l[i]!=r[i]){
				zx[r[i]]=va[i];
				pla[r[i]]=l[i];
			}
		}
		use=0;
		rs[1]=++use;
		for(int i=2;i<=cnt;i++){
			tot+=zx[i];
			int p=i;
			while(p!=1&&done[p]!=i&&!rs[p]){
				done[p]=i;
				p=pla[p];
			}
			if(done[p]==i){
				f=1;
				int w=p;++use;
				do{
					rs[w]=use;
					w=pla[w];
				}while(w!=p);
			}
		}
		if(use==1) break;
		for(int i=2;i<=cnt;i++)
			if(!rs[i]) rs[i]=++use;
		for(int i=1;i<=len;i++){
			if(rs[l[i]]!=rs[r[i]]) va[i]-=zx[r[i]];
			l[i]=rs[l[i]],r[i]=rs[r[i]];
		}
		cnt=use;
	}
	return tot;
}
int main(){
	scanf("%d",&n);
	n++;
	for(int i=1;i<=n;i++) rs[i]=i;
	for(int i=2;i<=n;i++){
		double a;
		scanf("%lf%d",&a,&t[i]);
		Add(1,i,a);
		t[i]--;
	}
	scanf("%d",&m);
	for(int i=0,a,b;i<m;i++){
		double c;
		scanf("%d%d%lf",&a,&b,&c);
		if(a==b) continue;
		Add(a+1,b+1,c);
	}
	for(int i=1;i<=n;i++) zx[i]=1008610086;
	for(int i=1;i<=len;i++) zx[r[i]]=min(zx[r[i]],va[i]);
	for(int i=2;i<=n;i++)
		if(t[i]>0) ans+=zx[i]*t[i];
	printf("%.2lf\n",ans+zxsxt());
	return 0;
}