题意:X 国的一个网络使用若干条线路连接若干个节点。节点间的通信是双向的。某重要数据包,为了安全起见,必须恰好被转发两次到达目的地。该包可能在任意一个节点产生,我们需要知道该网络中一共有多少种不同的转发路径。

源地址和目标地址可以相同,但中间节点必须不同。


思路:枚举每个点,对每个点dfs找能搜到长度为4的就ans++


代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 1e4+5;
vector<int> g[maxn];
ll ans;

void dfs(int u, int pre, int num)
{
    if(num == 4) { ans++; return ; }
    for(int i = 0; i < g[u].size(); i++)
    {
        int v = g[u][i];
        if(v != pre) dfs(v, u, num+1);
    }
}

int main(void)
{
    int n, m;
    while(cin >> n >> m)
    {
        ans = 0;
        for(int i = 0; i < maxn; i++)
            g[i].clear();
        while(m--)
        {
            int u, v;
            scanf("%d%d", &u, &v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        for(int i = 1; i <= n; i++)
            dfs(i, 0, 1);
        printf("%lld\n", ans);
    }
    return 0;
}


Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐