CF979C Kuro and Walking Route

CF979C Kuro and Walking Route 题解

最近做了好多图题，感觉还是图有意思，争取做个图论选手hh

---

两步dfs遍历图（其实是树）即可求解
翻译题目里面已经有了在这就不多加赘述了 总共n个节点n-1条边，说明这个无向图图是个树

我们可以先以x为根节点遍历图，遇到y即停止， 计算x不经过y所能到达的节点数numx，n-numx即为y及y一侧的节点数（不包括x与y之间的节点）

再以y为根节点，重复上述过程

所有路径的总数为组合数Cn2即n*（n-1）

x到y的路径数为总数减去x，y及各自两侧节点的积

因为数据较大，应用long long计算

#include<bits/stdc++.h>
using namespace std;

vector<int>ve[300005];
bool vis[300005];
bool pos[300005];
long long n,x,y,numx=0,numy=0;
void build(int a,int b)
{
    ve[a].push_back(b);
    ve[b].push_back(a);
}

void dfs1(int node)
{
    if(node==y)
    {
        return;
    }
    vis[node]=1;
    int big=ve[node].size();
    for(int i=0;i<big;i++)
    {
        if(!vis[ve[node][i]])
        {
            vis[ve[node][i]]=1;
            numx++;
            dfs1(ve[node][i]);
        }
    }
}

void dfs2(int node)
{
    if(node==x)
    {
        return;
    }
    vis[node]=1;
    int big=ve[node].size();
    for(int i=0;i<big;i++)
    {
        if(!vis[ve[node][i]])
        {
            vis[ve[node][i]]=1;
            numy++;
            dfs2(ve[node][i]);
        }
    }
}

int main()
{

    scanf("%lld%lld%lld",&n,&x,&y);
    for(int i=1;i<=n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        build(a,b);
    }
    dfs1(x);
    memset(vis,0,sizeof(vis));
    dfs2(y);
    printf("%lld\n",n*(n-1)-(n-numx)*(n-numy));
    return 0;
}