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 }