上一篇是我勉强还能看懂的题目,这里就到我的知识盲区了()
看完题目第一个思路就是找到分别以x和y为根且不为这两个子树共有的所有点,然后将这两个子树上且符合上述条件的两组点的个数相乘就是答案
很明显这个做法确实是对的
比赛的时候光想着怎么判重删点了,况且图的知识也不是很懂,勉强写一下最短路之类的,正好现在补题可以弥补这方面的一点知识
发现判重也很容易。我们进行两次bfs遍历,第一次以 x 为根,遇到y则不进行bfs,访问到的点打上 +1 的标记(这里要与已访问到的点区分开,这只是判重),这个标记在c数组里,另外已访问到的点放在vis数组里;第二次bfs以 y 为根,遇到x则停止bfs,访问到的点打上 -1 的标记。
为什么要进行+1与-1的标记呢?我们发现这个方法里,每个点在每一次bfs中最多会被访问到一次。如果每一个点在两次bfs中都被遍历到了,那么他在c数组中对应的标记就是零,因为 +1-1=0 ,也就是说这个点就属于被两个子树所占有的重根。剩下的 +1 与 -1 所标记的点就是被两个子树所独有的点了。
最后遍历所有点记录一下相乘即可。
/*I lv ya*/
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
#include <stack>
#include <sstream>
#include <map>
#include <set>
#include <queue>
#include <stdlib.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 2e5+5;
int n,x,y;
int h[3*N],cnt=0;///链式前向星
int c[3*N];
bool vis[3*N];
struct edge
{
int t,ne;
}ed[3*N];
void add(int s,int t)
{
ed[++cnt].ne=h[s];
ed[cnt].t=t;
h[s]=cnt;
}
void bfs(int now,int net,int num)
{
memset(vis,0,sizeof(vis));
queue <int> q;
q.push(now);
while(!q.empty())
{
int tmp=q.front();
q.pop();
if(vis[tmp]==1||tmp==net)
continue;
c[tmp]+=num;
vis[tmp]=1;
for(int i=h[tmp];i;i=ed[i].ne)
{
int to=ed[i].t;
q.push(to);
}
}
}
void solve()
{
cin>>n>>x>>y;
for(int i=1;i<=n-1;i++)
{
int s,t;
cin>>s>>t;
add(s,t);
add(t,s);
}
bfs(x,y,1);
bfs(y,x,-1);
int cnt1=0,cnt2=0;
for(int i=1;i<=n;i++)
{
if(c[i]==1) cnt1++;
else if(c[i]==-1) cnt2++;
}
cout<<(ll)cnt1*cnt2<<endl;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(NULL),cout.tie(NULL);
int _;
_=1;
while(_--)
{
solve();
}
return 0;
}