蓝桥杯历届试题 网络寻路（dfs）

题目原址：http://lx.lanqiao.cn/problem.page?gpid=T36
解法一：dfs

别用二维数组，会超时！！

别用二维数组，会超时！！

别用二维数组，会超时！！

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e5+5;
int ans=0;
vector<int>g[MAXN];
int n,m;
void dfs(int x,int t,int nob )
{
    if(t>=3)
    {
        ans++;
        return;
    }
    for(int i=0; i<g[x].size(); i++)
    {
        if(g[x][i]!=nob)
        {
            dfs(g[x][i],t+1,x);
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
    }
    for(int i=1; i<=n; i++)
    {
        dfs(i,0,0);
    }
    printf("%d\n",ans);

    return 0;
}

解法二：

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e5+5;
int ans=0;
int n,m;
int v[MAXN],u[MAXN];
int d[MAXN];
int main()
{

    cin>>n>>m;
    for(int i=0; i<m; i++)
    {
        cin>>u[i]>>v[i];
        d[u[i]]++;
        d[v[i]]++;
    }
    for(int i=0;i<m;i++)
        if(d[u[i]]>1&&d[v[i]]>1)
        ans+=(d[u[i]]-1)*(d[v[i]]-1)*2;
    cout<<ans<<endl;
    return 0;
}