Codeforces Round #482 (Div. 2) C. Kuro and Walking Route(思维)

题目链接
题意:不能经过x之后再经过y,问有多少条路径满足题意
解析:一点到其他一点都只有一条道路,所以我们把从x到y的路径找出来之后,统计y之后的点数,记为cnt,把x之前的点数记为cnt3,之后令cnt++,cnt3++,也就是把x,y算上
令总点数减去不满足的路径数即可

#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define inf 0x3f3f3f3f
#define pll pair<ll,ll>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep1(i,a,b) for(int i=a;i>=b;i--)
#define rson rt<<1|1,m+1,r
#define lson rt<<1,l,m
using namespace std;
const int N=3e5+100;
vector<ll>G[N];
ll vis[N];
ll cnt,have,n,x,y,cnt1,cnt3,pp;
void fi(ll a,ll pre)
{
    vis[a]=1;
    //cout<<a<<' '<<pre<<endl;
    for(int i=0;i<G[a].size();i++)
    {
        ll b=G[a][i];
        if(vis[b])
            continue;
        if(pre)
            cnt++;
        if(b==y)
            fi(b,1);
        else
            fi(b,pre);
    }
}
void fi1(ll a)
{
    if(pp) return ;
    vis[a]=1;
    //cout<<a<<endl;
    for(int i=0;i<G[a].size();i++)
    {
        ll b=G[a][i];
        if(!vis[b])
        {
            if(b==x)
            {
                pp=1;
                return ;
            }
            fi1(b);
        }
    }
}
void fi2(ll a)
{
    vis[a]=1;
    for(int i=0;i<G[a].size();i++)
    {
        ll b=G[a][i];
        if(!vis[b])
        {
            cnt3++;
            fi2(b);
        }
    }
}
int main()
{
    #ifdef LOCAL_DEFINE
        freopen("D:\\rush.txt","r",stdin);
    #endif
    ios::sync_with_stdio(false),cin.tie(0);
    cin>>n>>x>>y;
    pp=0;
    rep(i,0,n-2)
    {
        ll a,b;
        cin>>a>>b;
        G[a].pb(b);
        G[b].pb(a);
    }
    cnt=0,have=0,cnt1=0,cnt3=0;
    fi(x,0);
    memset(vis,0,sizeof vis);
    fi1(y);
    fi2(x);
    //cout<<cnt+1<<' '<<cnt3+1<<endl;
    cnt3++;
    cnt++;
    cout<<n*(n-1)-cnt3*(cnt)<<endl;
    return 0;
}