蒟蒻补题：I-马拉松

上一篇是我勉强还能看懂的题目，这里就到我的知识盲区了（）

看完题目第一个思路就是找到分别以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;
}