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 + ;
 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 = ;
     for (int v:g[u])
     {
         if (v != fa)
         {
             cnt += dfs(v,u);
         }
     }
     return num[u] = cnt + ;
 }
 int main()
 {
     scanf("%d%d%d",&n,&x,&y);
     for (int i = ;i < n;i++)
     {
         int u,v;
         scanf("%d%d",&u,&v);
         g[u].push_back(v);
         g[v].push_back(u);
     }
     dfs(x,-);
     long long ans = ;
     int m = y;
     while (par[m] != x)
     {
         m = par[m];
     }
     //printf("%d*\n",m);
     ans = (long long) n * (n - );
     ans -= (long long)num[y] * (1LL * (n - num[m]));
     printf("%lld\n",ans);
     return ;
 }