CodeForces - 1454E - Number of Simple Paths 思维

CodeForces - 1454E - Number of Simple Paths 思维

题意：给出一个n个顶点n条无向边的连通图，求出不重复的所有路径数量。

分析：

• 对于一棵树而言(含有n-1条无向边)，所有路径的数量就等于所有顶点两两配对的数量的二分之一(因为在无向图中x到y和y到x其实是同一条)。即$ans=$$\frac{n(n-1)}{2}$。
• 对于一个环而言(含有n条无向边)，所有路径的数量等于所有顶点两两配对的数量(因为环的特殊性，所以x到y有两种不同的走法或方向，不需要除2)，即$ans=n(n-1)$。
• 构成连通图首先需要消耗n-1条边构成一棵树，然后剩余的那条边必然后构成唯一一个环。

---

做法一：对于环上的每一个点，都可以视为一棵独立的树的根，其顶点数记为$sz[i]$，那么对于该小树上的任意一点$v$，到本树上的点的所有路径数量就是$ans1=$$\frac{sz[i](sz[i]-1)}{2}$，到其余的点的路径数量是$ans2=sz[i](n-sz[i])$，这是因为它必然要经过环，所以路径选择有两条，不需要除2。

想不到，看的官方题解
代码：

const int maxn=2e5+7;
const int INF=0x3f3f3f3f;
const ll INFF=1e18;
int t,n,sz[maxn],u,v;
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        ll ans=0;
        scanf("%d",&n);
        vector<set<int>> edge(n);
        rep(i,0,n-1)
        {
            scanf("%d%d",&u,&v);
            u--;v--;
            edge[u].insert(v);
            edge[v].insert(u);
            sz[i]=1;
        }
        queue<int>q;
        rep(i,0,n-1)if (edge[i].size()==1)q.push(i);
        while(q.size())
        {
            int top=q.front();q.pop();
            int fa=*edge[top].begin();
            sz[fa]+=sz[top];
            sz[top]=0;
            edge[top].clear();
            edge[fa].erase(top);
            if (edge[fa].size()==1)q.push(fa);
        }
        rep(i,0,n-1)
        {
            ans+=1ll*sz[i]*(sz[i]-1)/2;
            ans+=1ll*sz[i]*(n-sz[i]);
        }
        WW(ans);
    }
    return 0;
}

---

---

---

做法二：对于图上任意两点$x$,$y$，如果他们不属于同一个小树，则必然有2条路径数量。换句话说，如果他们属于同一个小树，就只有一条唯一路径。所以可以先用总的数量$\frac{n(n-1)}{2}$，遍历所有的小树，减去$\frac{sz[i](sz[i]-1)}{2}$，就是最终答案

代码：

const int maxn=2e6+7;
const int INF=0x3f3f3f3f;
const ll INFF=1e18;
int n,u,v,t,sum,pre[maxn];
vector<int> edge[maxn];
bool vis1[maxn],vis2[maxn],in[maxn];
void dfs1(int x,int fa)//找环
{
    vis1[x]=true;pre[x]=fa;
    repp(i,0,edge[x].size())
    {
        int v=edge[x][i];
        if (v==fa)continue;
        else if (vis1[v])
        {
            if (in[v])continue;
            in[v]=true;
            for (int p=x;p!=v;p=pre[p])in[p]=true;
        }
        else dfs1(v,x);
    }
}
void dfs2(int x)//找树
{
    vis2[x]=true;sum++;
    repp(i,0,edge[x].size())
    {
        int v=edge[x][i];
        if (!in[v]&&!vis2[v])dfs2(v);
    }
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        rep(i,1,n)
        {
            vis1[i]=vis2[i]=in[i]=false;
            edge[i].clear();
        }
        rep(i,1,n)
        {
            scanf("%d%d",&u,&v);
            edge[u].pb(v);
            edge[v].pb(u);
        }
        dfs1(1,0);
        ll ans=1ll*n*(n-1);
        rep(i,1,n)
        { 
            if (in[i])
            {
                sum=0;
                dfs2(i);
                ans-=1ll*sum*(sum-1)/2;
            }
        }
        WW(ans);
    }
    return 0;
}