P2052 [NOI2011]道路修建

题目描述:

  传送门

题解思路:

   这是一道深搜的题目,把题目意思翻译解读后理解起来就更容易一些了。题目翻译:首先我们明确这个图是一棵树,边的权值=(边两端结点数量之差)*边的长度。

  进一步翻译:我们只需要知道每条边两端节点数量分别是多少即可求解问题。那么如何计算边两端结点数量——首先选当前边上的其中一点,以该节点为根节点计算整棵树的结点数量即可。而通过深搜的方式来遍历到这些端点。

  图解:以下图中i和j分别是当前边的两个端点,假设我们以点j作为根节点,那么我们需要求的结点数量就是j+Y的部分的结点个数。而另一边的结点个数则为n-(j+Y的结点个数)。但是注意,在计算j+Y 部分时,为了防止搜回到i处,加入一个状态判断要搜索的点是否为上一个刚刚搜过的点,如果搜过则直接跳过这个点。比如dfs方向从i->j,搜完i进入j搜索,在j处搜索下一个结点时不能把i列入考虑范围!

  

 

代码:

 1 #include<iostream>
 2 #include<cmath>
 3 #include<string.h>
 4 using namespace std;
 5 #define re register
 6 
 7 //由于题目里所给的数据太大,不适合用邻接矩阵存图
 8 //而比较好用的存图方式有前向星(输入形式为"边起点 边终点 边权值"便可以考虑前向星) 
 9 struct edge{
10     int to;        //边的终点
11     int next;        //下一条边的编号
12     int w;            //边的权值 
13 }; 
14 edge e[2000001];
15 int head[1000005]={0};
16 int cnt=0;
17 int n;
18 int child[1000005];         //此处的child[i]代表本身加上孩子结点的数量 
19 long long ans=0;            //记录所有边的权值之和 
20 void add(int a,int b,int c){
21     e[cnt].to=b;
22     e[cnt].w=c;
23     e[cnt].next=head[a];
24     head[a]=cnt;                //第一条边编号更新 
25     cnt++;    
26 } 
27 void dfs(int cur,int last){//cur代表当前结点编号,last代表cur的上一个搜索过的结点编号,dfs方向从last->cur 
28     child[cur]=1;
29     int u=cur; 
30     for(re int k=head[u];~k;k=e[k].next){        //点u的邻接边的遍历,k是编号 
31         if(e[k].to!=last){                //因为是从last搜索过来的,不重复搜索last结点 
32             dfs(e[k].to,cur);            //递归边的终点
33             ans+=(long long)abs(child[e[k].to]-(n-child[e[k].to]))*e[k].w;        //在搜索点的时候权值累加 
34             child[cur]+=child[e[k].to];        //累加子节点个数
35         }
36     }
37 }
38 int main(){
39     cin>>n;
40     memset(head,-1,sizeof(head));
41     for(re int i=1;i<=n-1;i++){
42         int a,b,c;
43         cin>>a>>b>>c;
44         add(a,b,c);            //注意此处用到深搜,所以起始终止点是没有区别的 
45         add(b,a,c);
46     }
47     dfs(1,0);    //从点的编号1开始搜索,由于点的编号从1开始,所以假设点1的上个点搜索过的是0 
48     cout<<ans; 
49     return 0;
50 } 

 

posted @ 2020-05-02 11:43  neverstopcoding  阅读(166)  评论(0编辑  收藏  举报