codeforces 979C Kuro and Walking Route

题意；

给出一棵树，其中有两个点，x和y，限制走了x之后的路径上不能有y，问可以走的路径(u,v)有多少条，(u,v)和(v,u)考虑为两条不同的路径。

思路：

简单树形dp，dfs统计在x到y路径（不包括x和y）之外的所有点，在x这边的有a个，y这边的有b个，那么答案就是n*(n-1) - a * b。

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 3e5 + 10;
vector<int> g[N];
int num[N];
int par[N];
int n,x,y;
int dfs(int u,int fa)
{
    par[u] = fa;
    int cnt = 0;
    for (int v:g[u])
    {
        if (v != fa)
        {
            cnt += dfs(v,u);
        }
    }
    return num[u] = cnt + 1;
}
int main()
{
    scanf("%d%d%d",&n,&x,&y);
    for (int i = 1;i < n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(x,-1);
    long long ans = 0;
    int m = y;
    while (par[m] != x)
    {
        m = par[m];
    }
    //printf("%d*
",m);
    ans = (long long) n * (n - 1);
    ans -= (long long)num[y] * (1LL * (n - num[m]));
    printf("%lld
",ans);
    return 0;
}