网络寻路---蓝桥杯练习

又没有注意时间复杂度，哎

题目描述

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

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

如下图所示的网络。

1 -> 2 -> 3 -> 1 是允许的; 1 -> 2 -> 1-> 2 或者 1->2->3->2 都是非法的。

输入描述
输入数据的第一行为两个整数 N,M，分别表示节点个数和连接线路的条数 (1≤N≤10^4 , 0≤M≤10^5)。

接下去有 M 行，每行为两个整数 u 和 v，表示节点 u 和 v 联通(1≤u,v≤N,u!=v)。

输入数据保证任意两点最多只有一条边连接，并且没有自己连自己的边，即不存在重边和自环。

输出描述
输出一个整数，表示满足要求的路径条数。

输入输出样例
示例
输入

	3 3
	1 2
	2 3
	1 3

输出

	6

思路

1、错误思路

直接使用回溯法，从第一个点深度搜索到最后一个点，通过第二个数字到最后一个数字中没有重复进行剪枝

这样的话理解起来也不难，实现起来也很省事，可就是时间复杂度是O(n**4)，如果数据多了就直接超时了，呜呜呜

暴力法代码 O(n ** 4)

import os
import sys

# 请在此输入您的代码

m=list(map(int,input().split()))

N=m[0]
M=m[1]

numbers=dict()

for i in range(M):
  m=list(map(int,input().split()))
  
  if m[0] not in numbers:
    numbers[m[0]]=[]
  numbers[m[0]].append(m[1])
  
  if m[1] not in numbers:
    numbers[m[1]]=[]
  numbers[m[1]].append(m[0])
for j in range(N):
    if j not in numbers:
        numbers[j]=[]

steps=[]

route=[]

def con(num,nums):
    global numbers,route,steps,N
    if nums==4:
        return
    for i in range(len(numbers[num])):
        if numbers[num][i]==route[0]:
            route.append(numbers[num][i])
            if nums+1==4:
                temp=[]
                if route[0]>route[-1]:
                    for j in range(len(route)-1,-1,-1):
                        temp.append(route[j])
                elif route[0]==route[-1]:
                    for j in range(len(route)):
                        temp.append(route[j])                
                if temp not in steps:
                    steps.append([])
                    for j in range(len(temp)):
                        steps[-1].append(temp[j])
            del route[-1]
        elif numbers[num][i] in route[1:]:
            continue
        else:
            route.append(numbers[num][i])
            if nums+1==4:
                temp=[]
                if route[0]>route[-1]:
                    for j in range(len(route)-1,-1,-1):
                        temp.append(route[j])
                elif route[0]==route[-1]:
                    for j in range(len(route)):
                        temp.append(route[j]) 
                if temp not in steps:
                    steps.append([])
                    for j in range(len(temp)):
                        steps[-1].append(temp[j])
            con(numbers[num][i],nums+1)
            del route[-1]
        
            

for i in range(1,N+1):
    route.append(i)
    con(i,1)
    del route[-1]
print(len(steps))

2、正确思路
题中提到恰好两次转发，那么可以确定一共出现四个点，那么我们设置一个字典，记录每个数字的度。

	numbers={1: [2, 3], 2: [1, 3], 3: [2, 1]}

然后找出中间两个点。

比如 1 和 2

那么 1 还可以连的数字有 len(numbers[1])-1

同理 2 还可以连的数字有 len(numbers[2])-1

那么 1 和 2 在中间的种类就有 2*(len(numbers[2])-1) *(len(numbers[1])-1)

这样就可以做到时间复杂度为O(n**2)

正确代码

import os
import sys

# 请在此输入您的代码

m=list(map(int,input().split()))

N=m[0]
M=m[1]

numbers=dict()

for i in range(M):
  m=list(map(int,input().split()))
  
  if m[0] not in numbers:
    numbers[m[0]]=[]
  numbers[m[0]].append(m[1])
  
  if m[1] not in numbers:
    numbers[m[1]]=[]
  numbers[m[1]].append(m[0])
for j in range(N):
    if j not in numbers:
        numbers[j]=[]

nums=0

for i in range(N+1):
    for j in range(len(numbers[i])):
        nums+=(len(numbers[i])-1)*(len(numbers[numbers[i][j]])-1)
            
print(nums)

编码不易，有帮助的话点个赞支持一下吧~~~