P2052 [NOI2011]道路修建

题目描述:

  传送门

题解思路:

 

 

代码:

 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 } 

 

上一篇:题解【洛谷P1407】 [国家集训队]稳定婚姻


下一篇:【题解】[NOI2011]阿狸的打字机