Codeforces Round #482 (Div. 2) ：C

题目连接：http://codeforces.com/contest/979/problem/C

解题心得：

• 题意就是给你n个点，在点集中间有n-1条边（无重边），在行走的时候不能从x点走到y点，问你任意两点一共有多少种走法，u到v和v到u为不同的走法。
• 首先要明白的是n个点n-1条边就是一棵树。而在树中任意两点有且仅有一条路径。这样就简单了，先从x点dfs到y点，将唯一的一条路径标记上就将树x点和y点形成的子树分开了。然后不走标记的点就可以再次dfs的得到x点子树上的点数sumx，以及y点子树上的点数sumy。答案就是n*(n-1)-sumx*sumy

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 3e5+100;

vector <int> ve[maxn];
int n,x,y;
int vis[maxn];
void init(){
    memset(vis,-1, sizeof(vis));
    scanf("%d%d%d",&n,&x,&y);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        ve[a].push_back(b);
        ve[b].push_back(a);
    }
}

int find(int pos){
    vis[pos] = -2;
    if(pos == y) {
        vis[y] = 0;
        return 0;
    }
    for(int i=0;i<ve[pos].size();i++) {
        int v = ve[pos][i];
        if(vis[v] == -1) {
            if(find(v) == 0) {
                vis[pos] = 0;
                return 0;
            }
        }
    }
    return vis[pos];
}

ll cnt_chx;
void find_chx(int pos){
    vis[pos] = 1;
    cnt_chx++;
    for(int i=0;i<ve[pos].size();i++) {
        int v = ve[pos][i];
        if(vis[v] < 0)
            find_chx(v);
    }
}

ll cnt_chy;
void find_chy(int pos) {
    vis[pos] = 2;
    cnt_chy++;
    for(int i=0;i<ve[pos].size();i++) {
        int v = ve[pos][i];
        if(vis[v] < 0)
            find_chy(v);
    }
}

int main(){
    init();
    ll ans = (ll)n*((ll)n-(ll)1);
    find(x);
    find_chx(x);
    find_chy(y);
    ans -= cnt_chx*cnt_chy;
    printf("%lld
",ans);
    return 0;
}