Codevs 2370 小机房的树

Posted 2020-08-21 り挽歌、花开花落的流年

时间限制: 1 s　　空间限制: 256000 KB　　 题目等级 : 钻石 Diamond

题目描述 Description

小机房有棵焕狗种的树，树上有N个节点，节点标号为0到N-1，有两只虫子名叫飘狗和大吉狗，分居在两个不同的节点上。有一天，他们想爬到一个节点上去搞基，但是作为两只虫子，他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量（从父亲节点爬到此节点也相同），他们想找出一条花费精力最短的路，以使得搞基的时候精力旺盛，他们找到你要你设计一个程序来找到这条路，要求你告诉他们最少需要花费多少精力

输入描述 Input Description

第一行一个n，接下来n-1行每一行有三个整数u，v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ，v 表示两只虫子所在的节点

输出描述 Output Description

一共有m行，每一行一个整数，表示对于该次询问所得出的最短距离。

样例输入 Sample Input

3

1 0 1

2 0 1

3

1 0

2 0

1 2

样例输出 Sample Output

1

1

2

数据范围及提示 Data Size & Hint

1<=n<=50000， 1<=m<=75000， 0<=c<=1000

50分WA代码：

 1 #include<cstring>
 2 #include<cstdio>
 3 #include<algorithm>
 4 #include<iostream>
 5 using namespace std;
 6 #define maxn 60010
 7 #define S 16
 8 int n,m,a[maxn],deep[maxn],fa[maxn][S+10],ans,p1,p2,head[maxn],num,dis[maxn];
 9 struct node{
10     int u,v,pre,value;//  dis[i] 数组 从 i到根节点的最短距离 
11 }e[maxn*2];
12 void dfs(int u,int di)
13 {
14     dis[u]=di;/*统计u到根节点的距离*/
15     for(int l=head[u];l;l=e[l].pre)
16     {
17         if(!deep[e[l].v])
18         {
19             deep[e[l].v]=deep[u]+1;/*处理孩子的深度*/
20             fa[e[l].v][0]=u;/*初始化p数组*/
21             dfs(e[l].v,di+e[l].value);
22         }
23     }
24 }
25 void get_fa()
26 {
27     for(int j=1;j<=S;j++)
28         for(int i=1;i<=n;i++)
29             fa[i][j]=fa[fa[i][j-1]][j-1];
30 }
31 int LCA(int a,int b)
32 {
33     if(deep[a]<deep[b])swap(a,b); 
34     for(int j=S;j>=0;j--)
35       if((deep[a]-(1<<j))>=deep[b])
36         a=fa[a][j];
37     if(a==b)return a;
38     for(int i=S;i>=0;i--)
39       if(fa[a][i]!=fa[b][i])
40         {
41           a=fa[a][i];
42           b=fa[b][i];
43         }
44     return fa[a][0];
45 }
46 void add_egre(int from,int to,int value)
47 {
48     num++;
49     e[num].u=from;
50     e[num].v=to;
51     e[num].value=value;
52     e[num].pre=head[from];
53     head[from]=num;
54 }
55 int main()
56 {
57     scanf("%d",&n);
58     for(int i=1,x,y,z;i<n;i++)
59     {
60         scanf("%d%d%d",&x,&y,&z);
61         add_egre(x,y,z);add_egre(y,x,z);
62     }
63     memset(dis,0,sizeof(dis));
64     memset(deep,0,sizeof(deep));
65     dfs(1,0);
66     get_fa();scanf("%d",&m);
67     for(int i=1;i<=m;i++)
68     {
69         scanf("%d%d",&p1,&p2);
70         int k=LCA(p1,p2);
71         ans=dis[p1]+dis[p2]-2*dis[k];// deep shendu
72         printf("%d\n",ans);
73     }
74     return 0;
75 }

赶脚思路没毛病~~~

AC代码：

 1 #include<cstring>
 2 #include<cstdio>
 3 #include<algorithm>
 4 #include<iostream>
 5 using namespace std;
 6 #define maxn 60010
 7 #define S 16
 8 int n,m,a[maxn],deep[maxn],fa[maxn][20],ans,p1,p2,head[maxn],num;long long dis[maxn];
 9 struct node{
10     int u,v,pre,value;//  dis[i] 数组 从 i到根节点的最短距离 
11 }e[maxn*4];
12 void dfs(int u,long long di)
13 {
14     dis[u]=di;/*处理u到根节点的距离*/
15     for(int l=head[u];l;l=e[l].pre)
16     {
17         if(!deep[e[l].v])
18         {
19             deep[e[l].v]=deep[u]+1;/*处理孩子的深度*/
20             fa[e[l].v][0]=u;/*初始化fa数组*/
21             dfs(e[l].v,di+e[l].value);
22         }
23     }
24 }
25 int LCA(int a,int b)
26 {
27     int i,j;
28     if(deep[a]<deep[b]) swap(a,b); 
29     for(i=0;(1<<i)<=deep[a];++i);
30          i--;
31     for(j=i;j>=0;j--)
32       if((deep[a]-(1<<j))>=deep[b])
33         a=fa[a][j];
34     if(a==b)return a;
35     for(j=i;j>=0;j--)
36       if(fa[a][j]!=-1&&fa[a][j]!=fa[b][j])
37         {
38           a=fa[a][j];
39           b=fa[b][j];
40         }
41     return fa[a][0];
42 }
43 void init()
44 {
45      int i,j;
46      for(j=1;(1<<j)<n;j++)
47        for(int i=0;i<n;++i)
48         if(fa[i][j]=-1)
49           fa[i][j]=fa[fa[i][j-1]][j-1];/*DP处理出i的所有2^j祖先是谁*/
50 }
51 void add_egre(int from,int to,int value)
52 {
53     num++;
54     e[num].u=from;
55     e[num].v=to;
56     e[num].value=value;
57     e[num].pre=head[from];
58     head[from]=num;
59 }
60 int main()
61 {
62     scanf("%d",&n);
63     for(int i=1,x,y,z;i<n;i++)
64     {
65         scanf("%d%d%d",&x,&y,&z);
66         add_egre(x,y,z);add_egre(y,x,z);
67     }
68     memset(fa,-1,sizeof(fa));
69     deep[0]=0;
70     dfs(0,0);
71     init(); 
72     scanf("%d",&m);
73     for(int i=1;i<=m;i++)
74     {
75         scanf("%d%d",&p1,&p2);
76         int k=LCA(p1,p2);
77         ans=dis[p1]+dis[p2]-2*dis[k];
78         printf("%d\n",ans);
79     }
80     return 0;
81 }