HDU 4009 不定根最小树形图

讲一下建图过程，首先建立一个超级源点S，对于这个源点，向每个HOUSE连一条有向边，权值为该HOUSE建立WELL的费用，即高度*X。

然后每个可以连边的WELL之间，费用为曼哈顿距离*Y，然后考虑两边的高度，如果需要连接PUMB，则在该费用上+Z。

这样建图之后，以S为根，跑一遍最小树形图算法即可。

CODE：

#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Max 2505
#define FI first
#define SE second
#define ll long long
#define PI acos(-1.0)
#define inf 0x3fffffff
#define LL(x) ( x << 1 )
#define bug puts("here")
#define PII pair<int,int>
#define RR(x) ( x << 1 | 1 )
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define REP(i,s,t) for( int i = ( s ) ; i <= ( t ) ; ++ i )

using namespace std;

inline void RD(int &ret) {
char c;
int flag = 1 ;
do {
c = getchar();
if(c == '-')flag = -1 ;
} while(c < '0' || c > '9') ;
ret = c - '0';
while((c=getchar()) >= '0' && c <= '9')
ret = ret * 10 + ( c - '0' );
ret *= flag ;
}

inline void OT(int a) {
if(a >= 10)OT(a / 10) ;
putchar(a % 10 + '0') ;
}

/*********************************************/
#define N 1005
int n , X , Y , Z ;
struct C{
int x , y , z ;
}city
;
int num = 0 ;
int S ;
void init(){
num = S = 0 ;
}
struct kdq{
int s ,e ,l ;
}E[N * N] ;
int get_Mandis(int i ,int j){
return abs(city[i].x - city[j].x) + abs(city[i].y - city[j].y) + abs(city[i].z - city[j].z) ;
}
void add(int s ,int e ,int l){
E[num].s = s ;
E[num].e = e ;
E[num].l = l ;
num ++ ;
}
int pre
, vis
, id
, in
;
int Directed_MST(int root ,int NV ,int NE){
int ret = 0 ;
while(1){
for (int i = 0 ; i < NV ; i ++ )in[i] = inf ;
for (int i = 0 ; i < NE ; i ++ ){
int s = E[i].s ;
int e = E[i].e ;
if(in[e] > E[i].l && s != e){
in[e] = E[i].l ;
pre[e] = s ;
}
}
for (int i = 0 ; i < NV ; i ++ ){
if(i == root)continue ;
if(in[i] == inf)return -1 ;
}
int cntnode = 0 ;
mem(vis , -1) ;
mem(id , -1) ;
in[root] = 0 ;
for (int i = 0 ; i < NV ; i ++ ){
ret += 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 < NV ; i ++ )if(id[i] == -1)id[i] = cntnode ++ ;
for (int i = 0 ; i < NE ; i ++ ){
int s = E[i].s ;
int e = E[i].e ;
E[i].s = id[s] ;
E[i].e = id[e] ;
if(id[s] != id[e])E[i].l -= in[e] ;
}
NV = cntnode ;
root = id[root] ;
}
return ret ;
}
int main() {
int a , k ;
while(scanf("%d%d%d%d",&n,&X,&Y,&Z) , ( n + X + Y + Z)){
init() ;
for (int i = 1 ; i <= n ;i ++ ){
RD(city[i].x) ;RD(city[i].y) ;RD(city[i].z) ;
}
for (int i = 1 ; i <= n ; i ++ ){
RD(k) ;
while(k -- ){
RD(a) ;
if(i == a)continue ;//自环
int dis = get_Mandis(i , a) * Y ;
if(city[i].z < city[a].z)dis += Z ;
add(i , a , dis) ;
}
}
for (int i = 1 ; i <= n ; i ++ ){
add(S , i , city[i].z * X) ;
}
int ans = Directed_MST(0 , n + 1 , num) ;
if(ans == -1)puts("poor XiaoA") ;
else OT(ans) ;
puts("") ;
}
return 0 ;
}