https://www.luogu.org/problemnew/show/P3398

題目描述

小倉鼠的和他的基（mei）友（zi）sugar住在地下洞穴中，每個節點的編號為1~n。地下洞穴是一個樹形結構。這一天小倉鼠打算從從他的臥室（a）到餐廳（b），而他的基友同時要從他的臥室（c）到圖書館（d）。他們都會走最短路徑。現在小倉鼠希望知道，有沒有可能在某個地方，可以碰到他的基友？

小倉鼠那麽弱，還要天天被zzq大爺虐，請你快來救救他吧！

輸入輸出格式

輸入格式：

第一行兩個正整數n和q，表示這棵樹節點的個數和詢問的個數。

接下來n-1行，每行兩個正整數u和v，表示節點u到節點v之間有一條邊。

接下來q行，每行四個正整數a、b、c和d，表示節點編號，也就是一次詢問，其意義如上。

輸出格式：

對於每個詢問，如果有公共點，輸出大寫字母“Y”；否則輸出“N”。

輸入輸出樣例

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

Y
N
Y
Y
Y

說明

__本題時限1s，內存限制128M，因新評測機速度較為接近NOIP評測機速度，請註意常數問題帶來的影響。__

20%的數據 n<=200,q<=200

40%的數據 n<=2000,q<=2000

70%的數據 n<=50000,q<=50000

100%的數據 n<=100000,q<=100000

很明顯就是一道樹剖。

一開始的思路是每次給兩段路徑打上標記，用線段樹維護。

可是突然覺得代碼有點長，效率有點低，好像有更高效，代碼更短的方法>_<

於是我又畫了幾次圖發現如果有公共點的話，一定有一條路徑的lca在另一條上。

怎麽判斷一個點在一條路徑上呢？

設點為x，路徑為s到t，則

1.deep[x]>=deep[LCA(s,t)]

2.LCA(s,x)=x或LCA(t,x)=x

接下來就好辦了

 1 // luogu-judger-enable-o2
 2 #include<bits/stdc++.h>
 3 namespace ZDY{
 4 #define ll long long
 5 #define Fur(i,x,y) for(int i=x;i<=y;i++)
 6 #define Fdr(i,x,y) for(int i=x;i>=y;i--)
 7 #define in2(x,y) in(x);in(y)
 8 #define in3(x,y,z) in2(x,y);in(z)
 9 #define in4(a,b,c,d) in2(a,b);in2(c,d)
10 #define clr(x,y) memset(x,y,sizeof(x))
11 #define cpy(x,y) memcpy(x,y,sizeof(x))
12 #define fl(i,x) for(ll i=head[x],to;to=e[i].to,i;i=e[i].next)
13 #define inf 999999999
14 }
15 namespace FAST{
16 #define pob (fwrite(fob::b,sizeof(char),fob::f-fob::b,stdout),fob::f=fob::b,0)
17 #define pc(x) (*(fob::f++)=(x),(fob::f==fob::g)?pob:0)
18 #define gc ((*fib::f)?(*(fib ::f++)):(fgets(fib::b,sizeof(fib::b),stdin)?(fib::f=fib::b,*(fib::f++)):-1))
19 namespace fib{char b[300000]= {},*f=b;}
20 inline void in(ll &x){x=0;char c;bool f=0;while((c=gc)>‘9‘||c<‘0‘)if(c==‘-‘)f=!f;x=c-48;while((c=gc)<=‘9‘&&c>=‘0‘)x=x*10+c-48;if(f)x=-x;}
21 inline void in(int &x){x=0;char c;bool f=0;while((c=gc)>‘9‘||c<‘0‘)if(c==‘-‘)f=!f;x=c-48;while((c=gc)<=‘9‘&&c>=‘0‘)x=x*10+c-48;if(f)x=-x;}
22 namespace fob{char b[300000]= {},*f=b,*g=b+300000-2;}struct foce{~foce(){pob;fflush(stdout);}} _foce;namespace ib{char b[100];}
23 inline void out(ll x){if(x==0){pc(48);return;}if(x<0){pc(‘-‘);x=-x;}char *s=ib::b;while(x) *(++s)=x%10,x/=10;while(s!=ib::b) pc((*(s--))+48);}inline void outn(ll x){out(x);pc(‘\n‘);}
24 inline void out(int x){if(x==0){pc(48);return;}if(x<0){pc(‘-‘);x=-x;}char *s=ib::b;while(x) *(++s)=x%10,x/=10;while(s!=ib::b) pc((*(s--))+48);}inline void outn(int x){out(x);pc(‘\n‘);}
25 inline char sc(){char c=gc;while(!((‘A‘<=c&&c<=‘Z‘)||(‘a‘<=c&&c<=‘z‘)))c=gc;return c;}
26 }using namespace FAST;using namespace ZDY;using namespace std;
27 #define N 100010
28 int n,m,cnt=1;
29 int head[N],top[N],siz[N],d[N],f[N];
30 struct edge{int next,to;}e[N*2];
31 inline void add(int x,int y){e[cnt].to=y;e[cnt].next=head[x];head[x]=cnt++;}
32 inline void dfs(int x){
33     siz[x]=1;
34     fl(i,x)
35     if(!siz[to]){
36         d[to]=d[x]+1;
37         f[to]=x;
38         dfs(to);
39         siz[x]+=siz[to];
40     }
41 }
42 inline void bt(int x,int tp){
43     top[x]=tp;ll k=0;
44     fl(i,x)if(to!=f[x]&&siz[to]>siz[k])k=to;
45     if(!k)return;bt(k,tp);
46     fl(i,x)if(!top[to])bt(to,to);
47 }
48 inline int find(int x,int y){
49     while(top[x]!=top[y]){
50         if(d[top[x]]<d[top[y]])swap(x,y);
51         x=f[top[x]];
52     }return d[x]<d[y]?x:y;
53 }
54 int main(){
55     in2(n,m);
56     int x,y,xx,yy,k,s;
57     Fur(i,1,n-1){in2(x,y);add(x,y);add(y,x);}
58     dfs(1);bt(1,1);
59     while(m--){
60         in4(x,y,xx,yy);
61         k=find(x,y);s=find(xx,yy);if(d[k]<d[s])swap(x,xx),swap(y,yy),swap(k,s);
62         pc((find(k,xx)==k||find(k,yy)==k)?‘Y‘:‘N‘);pc(‘\n‘);
63     }
64 }

[luogu]P3398 倉鼠找sugar