P3884 [JLOI2009]二叉树问题(洛谷)

题目描述

如下图所示的一棵二叉树的深度、宽度及结点间距离分别为:

深度:4 宽度:4(同一层最多结点个数)

结点间距离: ⑧→⑥为8 (3×2+2=8)

⑥→⑦为3 (1×2+1=3)

注:结点间距离的定义:由结点向根方向(上行方向)时的边数×2,

与由根向叶结点方向(下行方向)时的边数之和。

输入格式

输入文件第一行为一个整数n(1≤n≤100),表示二叉树结点个数。接下来的n-1行,表示从结点x到结点y(约定根结点为1),最后一行两个整数u、v,表示求从结点u到结点v的距离。

输出格式

三个数,每个数占一行,依次表示给定二叉树的深度、宽度及结点u到结点v间距离。

输入输出样例

输入 #1

10                               

1 2                           

1 3                           

2 4

2 5

3 6

3 7

5 8

5 9

6 10

8 6

输出 #1

4

4

8

题解思路:

原本的打算是先建立一棵树,再进行dfs。但对于此题,已经知道根节点始终是1且各节点连接信息表现出来的情况下,可以直接利用邻接矩阵+dfs从根节点搜,保存各节点的深度信息即可。

       具体步骤如下:
       1.先使用邻接矩阵(即二维数组)储存各节点长度关系,如a[i][j]储存第i和第j个结点的距离,若两节点不连接,则设置为0。

       2.首先,从根节点(1)开始往下搜,用dfs遍历所有的结点,在遍历的同时用depth数组储存各节点的深度,如根节点1的深度为1,每访问到一个节点就保存深度信息。

       3.有了结点信息后,进行排序求出最大深度。另外,通过统计相同的深度的节点数量即可确定树的最大宽度。

       4.接下来,开始对起始要搜索的点dfs,当碰见一个节点,进行深度比较,若深度大于该节点则路程加2,若小于该节点则路程加1,知道遇到终止点。、

代码

 1 #include<iostream>
 2 #include<algorithm>
 3 #include<cstring>
 4 using namespace std;
 5 
 6 //原本的打算是先建立一棵树,再进行dfs 
 7 //但对于此题,已经知道根节点始终是1且各节点连接信息表现出来的情况下,可以直接利用邻接矩阵+dfs从根节点搜,保存各节点的深度信息即可 
 8 #define INF 999999999
 9 int depth[101];    //记录各节点的深度信息 
10 int broad[100];    //记录宽度(dfs后记录) 
11 int map[101][101]={0};        //邻接矩阵,0代表无连接,1代表连接 
12 int n=0;
13 int flag[101]={0};        //存储各节点是否被访问的信息 
14 int u=0,v=0;        //起始和目标 
15 int minsum=INF;
16 
17 bool cmp(int a,int b){
18     return a>b;
19 }
20 void dfs(int cur,int dep){
21     flag[cur]=1;
22     depth[cur]=dep;
23 //    cout<<"cur:"<<cur<<" depth:"<<depth[cur]<<endl;
24     
25     for(int i=1;i<=n;i++){
26         if(flag[i]==1)    continue;        //防止来回搜死循环 
27         if(map[cur][i]!=0){
28             dfs(i,dep+1);
29         }
30         flag[i]=0;    //回溯 
31     }
32     return ; 
33 }
34 void dfs1(int cur,int sum){        //cur是结点编号,sum为当前距离 
35 //    cout<<"cur:"<<cur<<" sum:"<<sum<<endl;
36     flag[cur]=1;
37     if(sum>=minsum){
38         return ;        //剪枝 
39     }
40     if(cur==v){
41         if(sum<minsum){
42             minsum=sum;
43         } 
44         return;        //找到并返回,更新minsum 
45     }
46     
47     for(int i=1;i<=n;i++){
48         if(flag[i]==1)    continue;
49         if(map[cur][i]!=0){
50             if(depth[i]>depth[cur]){
51                 dfs1(i,sum+1);        //向下搜 
52             }else if(depth[i]<depth[cur]){
53                 dfs1(i,sum+2);        //向上搜 
54             }
55         }
56         flag[i]=0;    //回溯 
57     }
58     
59     return;     
60 }
61 int main(){
62     cin>>n;
63     for(int i=1;i<=n-1;i++){
64         int m,k;
65         cin>>m>>k;
66         map[m][k]=map[k][m]=1;
67     }
68     cin>>u>>v;
69     
70     dfs(1,1);
71     int maxdepth=0;        //最大深度 
72     int maxbroad=0;        //最大宽度 
73     for(int i=1;i<=n;i++){
74         if(depth[i]>maxdepth){
75             maxdepth=depth[i];
76         }
77         broad[depth[i]]++;        //同一深度的结点数量累加 
78     }
79     sort(broad,broad+n,cmp);
80     maxbroad=broad[0];
81     
82     //上述进行完树的深度和宽度查询后,下面开始距离搜索
83     memset(flag,0,sizeof(flag));
84     dfs1(u,0); 
85     cout<<maxdepth<<endl<<maxbroad<<endl<<minsum;
86     return 0;
87 }

猜你喜欢

转载自www.cnblogs.com/xwh-blogs/p/12579957.html