hdu 2489 MST+dfs
http://acm.hdu.edu.cn/showproblem.php?pid=2489
题目大意是有n个顶点的无向完全图，要你选m个点，使得m个点构成的图的m-1条边的权重之和比上m个点的值的和最小，即Sum(edge weight)/Sum(point weight)最小；
由于题目的数据很小，因此我就直接暴力了。。。
就是从n个点中先选m个点，对应一个最小值，然后对这m个点进行搜索就可以了。。。
如果一开始就从第一个点就开始dfs，那么就可以保证最后的结果就是题目要求的。。。
View Code
1 #include<iostream> 2 const int N=20; 3 const int inf=10000000; 4 using namespace std; 5 6 int node[N];//点的权值 7 int recode[N];//保存点 8 int edge[N][N]; 9 int lowcost[N]; 10 int map[N][N];//用来保存边 11 int path[N]; //保存路径 12 int n,m; 13 double MIN; 14 15 double prim(int v0){ 16 double sum=0; 17 for(int i=1;i<=m;i++){ 18 lowcost[i]=map[v0][i]; 19 } 20 lowcost[v0]=-1; 21 for(int i=1;i<m;i++){ 22 int min=inf,v=-1; 23 for(int j=1;j<=m;j++){ 24 if(lowcost[j]!=-1&&lowcost[j]<min){ 25 v=j,min=lowcost[j]; 26 } 27 } 28 if(v!=-1){ 29 sum+=lowcost[v]; 30 lowcost[v]=-1; 31 for(int k=1;k<=m;k++){ 32 if(lowcost[k]!=-1&&map[v][k]<lowcost[k]){ 33 lowcost[k]=map[v][k]; 34 } 35 } 36 } 37 } 38 return sum; 39 } 40 void dfs(int v0,int count){ 41 recode[count]=v0; 42 if(count==m){ 43 for(int i=1;i<=m;i++){ 44 map[i][i]=0; 45 for(int j=i+1;j<=m;j++){ 46 map[i][j]=map[j][i]=inf; 47 } 48 } 49 for(int i=1;i<=m;i++){ 50 for(int j=1;j<=m;j++){ 51 map[i][j]=edge[recode[i]][recode[j]]; 52 } 53 } 54 double s1=prim(1),s2=0; 55 for(int i=1;i<=m;i++){ 56 s2+=node[recode[i]]; 57 } 58 double s=s1/s2; 59 if(s<MIN){ 60 MIN=s; 61 //保存路径 62 for(int i=1;i<=m;i++) 63 path[i]=recode[i]; 64 } 65 } 66 for(int i=v0+1;i<=n;i++) 67 dfs(i,count+1); 68 } 69 70 int main(){ 71 while(scanf("%d%d",&n,&m)!=EOF){ 72 if(n==0&&m==0)break; 73 for(int i=1;i<=n;i++){ 74 scanf("%d",&node[i]); 75 } 76 for(int i=1;i<=n;i++){ 77 for(int j=1;j<=n;j++){ 78 scanf("%d",&edge[i][j]); 79 } 80 } 81 MIN=1e18; 82 for(int i=1;i<=n;i++) 83 dfs(i,1); 84 printf("%d",path[1]); 85 for(int i=2;i<=m;i++){ 86 printf(" %d",path[i]); 87 } 88 printf("\n"); 89 } 90 return 0; 91 }
