讲一下建图过程,首先建立一个超级源点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[N] ; 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[N] , vis[N] , id[N] , in[N] ; 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 ; }