BZOJ——1602: [Usaco2008 Oct]牧场行走

Posted 2020-09-16

http://www.lydsy.com/JudgeOnline/problem.php?id=1602

题目描述

N头牛（2<=n<=1000）别人被标记为1到n，在同样被标记1到n的n块土地上吃草，第i头牛在第i块牧场吃草。 这n块土地被n-1条边连接。 奶牛可以在边上行走，第i条边连接第Ai，Bi块牧场，第i条边的长度是Li（1<=Li<=10000）。 这些边被安排成任意两头奶牛都可以通过这些边到达的情况，所以说这是一棵树。 这些奶牛是非常喜欢交际的，经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。

输入

*第一行：两个被空格隔开的整数：N和Q

 *第二行到第n行：第i+1行有两个被空格隔开的整数：AI，BI，LI

*第n+1行到n+Q行：每一行有两个空格隔开的整数：P1，P2，表示两头奶牛的编号。

输出

*第1行到第Q行：每行输出一个数，表示那两头奶牛之间的距离。

样例输入

4 2
2 1 2
4 3 2
1 4 3
1 2
3 2

样例输出

2
7

提示

 

来源

资格赛

 

很惨的一下午：倍增死活没过，然后改树剖就A了~~~QAQ

 

先贴个没过的请大佬找找错

 1 #include <algorithm>
 2 #include <cstdio>
 3  
 4 using namespace std;
 5  
 6 const int N(1000+15);
 7 int n,q,x,y,z;
 8 int dad[N<<1][23],deep[N<<1];
 9 int dis[N<<1];
10 int sumedge,head[N<<1];
11 struct Edge
12 {
13     int from,to,next,dis;
14     Edge(int from=0,int to=0,int next=0,int dis=0) :
15         from(from),to(to),next(next),dis(dis) {}
16 }edge[N<<1];
17  
18 int add(int from,int to,int dis)
19 {
20     edge[++sumedge]=Edge(from,to,head[from],dis);
21     return head[from]=sumedge;
22 }
23  
24 void DFS(int x)
25 {
26     deep[x]=deep[dad[x][0]]+1;
27     for(int i=0;dad[x][i];i++)
28         dad[x][i+1]=dad[dad[x][i]][i];
29     for(int i=head[x];i;i=edge[i].next)
30         if(!deep[edge[i].to])
31         {
32             dad[edge[i].to][0]=x;
33             dis[edge[i].to]=dis[x]+edge[i].dis;
34             DFS(edge[i].to);
35         }
36 }
37  
38 int LCA(int x,int y)
39 {
40     if(deep[x]>deep[y]) swap(x,y);
41     for(int i=20;i>=0;i--)
42         if(deep[x]<=deep[dad[y][i]]) y=dad[y][i];
43     if(x==y) return x;
44     for(int i=20;i>=0;i--)
45         if(deep[dad[x][i]]!=deep[dad[y][i]])
46             x=dad[x][i],y=dad[y][i];
47     return dad[x][0];
48 }
49  
50 int main()
51 {
52     scanf("%d%d",&n,&q);
53     for(int i=1;i<n;i++)
54     {
55         scanf("%d%d%d",&x,&y,&z);
56         add(x,y,z); add(y,x,z);
57     }
58     DFS(1);
59     for(;q;q--)
60     {
61         scanf("%d%d",&x,&y);
62         printf("%d\n",dis[x]+dis[y]-(dis[LCA(x,y)]<<1));
63     }
64     return 0;
65 }

倍增LCA

 

然后是树剖的~

 1 #include <algorithm>
 2 #include <cstdio>
 3 #include <vector>
 4 
 5 using namespace std;
 6 
 7 const int N(1000+15);
 8 vector< pair<int,int> >vec[N];
 9 int n,q,x,y,z,dis[N];
10 int dad[N],deep[N],size[N],top[N];
11 
12 void DFS(int x)
13 {
14     size[x]=1; deep[x]=deep[dad[x]]+1;
15     for(int i=0;i<vec[x].size();i++)
16         if(dad[x]!=vec[x][i].first)
17         {
18             dad[vec[x][i].first]=x;
19             dis[vec[x][i].first]=dis[x]+vec[x][i].second;
20             DFS(vec[x][i].first);
21             size[x]+=size[vec[x][i].first];
22         }
23 }
24 
25 void DFS_(int x)
26 {
27     int t=0; if(!top[x]) top[x]=x;
28     for(int i=0;i<vec[x].size();i++)
29         if(dad[x]!=vec[x][i].first&&size[t]<size[vec[x][i].first]) t=vec[x][i].first;
30     if(t) top[t]=top[x],DFS_(t);
31     for(int i=0;i<vec[x].size();i++)
32         if(dad[x]!=vec[x][i].first&&t!=vec[x][i].first) DFS_(vec[x][i].first);
33 }
34 
35 int LCA(int x,int y)
36 {
37     while(top[x]!=top[y])
38     {
39         if(deep[top[x]]<deep[top[y]]) swap(x,y);
40         x=dad[top[x]];
41     }
42     if(deep[x]>deep[y]) swap(x,y);
43     return x;
44 }
45 
46 int main()
47 {
48     scanf("%d%d",&n,&q);
49     for(int i=1;i<n;i++)
50     {
51         scanf("%d%d%d",&x,&y,&z);
52         vec[x].push_back(make_pair(y,z));
53         vec[y].push_back(make_pair(x,z));
54     }
55     DFS(1); DFS_(1);
56     for(;q;q--)
57     {
58         scanf("%d%d",&x,&y);
59         printf("%d\n",dis[x]+dis[y]-dis[LCA(x,y)]*2);
60     }
61     return 0;
62 }