首页 > 代码库 > hdoj 4009 Transfer water 【无源点最小树形图】【好题】
hdoj 4009 Transfer water 【无源点最小树形图】【好题】
题目:hdoj 4009 Transfer water
题意:题目有点长,说是有个村子,有 n 户人家要用水,他们加的位置用三维坐标来表示(x,y,z),他们有两种选择:
1:自己挖一口井,花费为 z * cost_x
2:从别人家接个水管引过来,化为为距离 * cost_y,如果要引的地方比当前地方低的话,还要买一个水泵,花费cost_z。
距离算法|x2‐x1|+|y2‐y1|+|z2‐z1|.
然后求让所有的人都有用的水的最小花费、
分析:发现它是一个求最小生成树的题目,但是关键点有二
1:源点有很多,自建水井的地方都是源点。
2:图是有向图,这样很明显想到的是最小树形图
那么处理第一点就很简单了,我们加一个源点S,让s和每个点连一条边,距离为挖井的花费,然后以这个点就一次最小树形图就ok了。
AC代码:
#include <iostream> #include <algorithm> #include <string> #include <math.h> #include <vector> #include <cstring> #include <cstdio> using namespace std; const int N = 1100; const int inf = 0x3f3f3f3f; int n,m; struct Node { int from,to,dis; }; vector<Node> e; int ha[N],vis[N],father[N],in[N]; int Minroot; int zhuliu(int root) { int ans = 0; while(true) { for(int i=0;i<n;i++) in[i] = inf; memset(father,-1,sizeof(father)); for(int i=0; i<e.size(); i++) //找最小入边 { int to = e[i].to; if(e[i].dis<in[to] && e[i].from!=e[i].to) { in[to] = e[i].dis; father[to] = e[i].from; // if(e[i].from == root) //找最小 // Minroot = i; } } for(int i=0;i<n;i++) {//printf("%d ",in[i]); if(i!=root && in[i]==inf) return -1; } int cnt = 0; in[root] = 0; memset(ha,-1,sizeof(ha)); memset(vis,-1,sizeof(vis)); for(int i=0;i<n;i++) //找自环 { ans += in[i]; int v = i; while(v!=root && ha[v]==-1 && vis[v]!=i) { vis[v] = i; v = father[v]; } if(v!=root && ha[v]==-1) { for(int j = father[v];j != v;j=father[j]) { ha[j] = cnt; } ha[v] = cnt++; } } if(cnt == 0) //跳出条件 break; for(int i=0;i<n;i++) if(ha[i]==-1) ha[i]=cnt++; for(int i = 0; i< e.size();i++) { int tmp = e[i].to; e[i].from = ha[e[i].from]; e[i].to = ha[e[i].to]; if(e[i].from != e[i].to) e[i].dis -= in[tmp]; } n = cnt; root = ha[root]; } return ans; } struct Tree { int x,y,z; }; Tree p[N]; int dis(Tree a,Tree b) { return abs(a.x-b.x)+abs(a.y-b.y)+abs(a.z-b.z); } int main() { //freopen("Input.txt","r",stdin); int x,y,z; while(~scanf("%d%d%d%d",&n,&x,&y,&z) && n+x+y+z) { for(int i=0;i<n;i++) { scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z); } for(int i=0;i<n;i++) { int k,x; scanf("%d",&k); while(k--) { scanf("%d",&x); x--; if(x==i) continue; int tmp = dis(p[i],p[x])*y; if(p[i].z<p[x].z) tmp+=z; e.push_back((Node){i,x,tmp}); } } n++; for(int i=0;i<n-1;i++) { e.push_back((Node){n-1,i,p[i].z*x}); } // for(int i=0;i<e.size();i++) // { // printf("%d %d %d\n",e[i].from,e[i].to,e[i].dis); // } int ans = zhuliu(n-1); printf("%d\n",ans); e.clear(); } return 0; }
hdoj 4009 Transfer water 【无源点最小树形图】【好题】
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。