1.输入
import matplotlib.pyplot as plt
import numpy as np
import math
from pylab import *
print("请输入方格纸的大小a b")
ab=np.array((input().split()),int)
print("请输入点的个数n:")
n=int(input())
print("请输入各个点的横坐标")
x=np.array((input().split()),int)
print("请输入各个点的纵坐标")
y=np.array((input().split()),int)
2.计算长度的函数
首先我们先写出计算两个点之间的距离,再把他们按照不同的顺序相加
def distance(x1,y1,x2,y2): #计算两点之间的距离
d=pow((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2),0.5)
return d
#------------------------------以下是计算长度总和
def length(a):
i=0
l=0
while i<n-1:
m1=int(a[i])
m2=int(a[i+1])
l=l+distance(x[m1],y[m1],x[m2],y[m2])
i=i+1
return l
3.找出不同的连线方式
在此我们的思路是先将n个点编号为0~n-1,同时我们将数组一个包含n个零的一维数组从第n-1位加一,每逢n-1则进位,如n=3时,则会产生[0,0,0],[0,0,1],[0,0,2],[0,1,0],[0,1,1],[0,1,2],[0,2,0]一直到[2,2,2]这便代表了不同的连线方式,显然其中会有不少重复的连线方式,且同一个点不能被连接两次,如[0 1 2]与[2 1 0]是完全一样的,[0,0,0] [0,0,1]这种连线方式也不应该存在,因此我们还需要删去他们
进位函数:
def carry(b,i):
b[i]=0
b[i-1]=b[i-1]+1
if(b[i-1]>=n):
carry(b,i-1)
检查出0123 3210这种相同的连线方式:
def check(a,b): #这个函数用来检查连接方式是否是一样的,如ABCD=DCBA如果一样。将二者变得一样
ch=0
i=0
while i<a.size:
if(a[i]==b[a.size-1-i]):
ch=ch+1
i=i+1
if(ch==a.size):
i=0
while i<a.size:
b[i]=a[i]
i=i+1
生成不同连接方式并删去相同的连线方式:
i=0
b=np.zeros(n)
b[n-1]=-1
c=np.zeros(n)
c[n-1]=1
cc=1
ccc=0
while cc==1:
m=0
b=b+c #这里便是上述中一直加一的部分
if(b[n-1]>=n):
carry(b,n-1)
i1=0
while i1<b.size:
j1=0
for j1 in range(i1+1,b.size):
if(b[i1]==b[j1]):
m=m+1
i1=i1+1
if(m==0):
l=length(b) #这里length(b)是用来计算数组b连线方式的总长度,并把它储存在length1[k]中
length1[k]=round(l,4)
number[k,...]=b
k=k+1
#print(float(l))
ccc=ccc+1
if(ccc==pow(n,n)):
break
#print(number)
while i<math.factorial(n)-1:
j=0
while j<math.factorial(n):
nbi=number[i,...]
nbj=number[j,...]
check(nbi,nbj) #这里的check函数为去除0123与3210这种相同连线方式
j=j+1
i=i+1
#print(number)
newnum=np.ones([int(0.5*math.factorial(n)),n])
i=0
n1=0
while i<math.factorial(n)-1:
#j=i+1
for j in range(i+1,math.factorial(n)):
k=0
m=0
while k<n:
if(number[i,k]==number[j,k]):
m=m+1
k=k+1
if(m==n):
newnum[n1,...]=number[i,...]
n1=n1+1
j=j+1
i=i+1
print(newnum) #newnum为二维数组 它储存了n!种不同的连线方式
4.筛选出最小连线方式
比大小找出最小的,并且记录他们在newnum中位置,即记录他们所对应的连线方式
length2=np.empty(int(0.5*math.factorial(n)))#length2是为了记录不同连线方式在newnum中的位置
i=0
while i<int(0.5*math.factorial(n)):
length2[i]=length(newnum[i,...])
i=i+1
print(length2)
#--------------------------以下筛选出最小的连接方式
i=0
minvalue=length2[0]
k=0
minlength=np.ones(int(0.5*math.factorial(n)))
while i<int(0.5*math.factorial(n)):
if(length2[i]<=minvalue):
minvalue=length2[i]
i=i+1
print(minvalue)
i=0
while i<int(0.5*math.factorial(n)):
if(length2[i]==minvalue):
minlength[k]=i
minlength[k+1]=-1 #储存截止标志,即遇见-1便是截止位置
k=k+1
i=i+1
print(minlength)
i=0
while i<minlength.size:
if(minlength[i]==-1):
break
nnm=newnum[int(minlength[i])]
connect(nnm)
i=i+1
5.连接最小连线
我们现在已经找出最小连线方式在newnum中的位置,并且把它储存在minlength接下来只需要连线即可
def connect(a):
i=0
#plt.figure(figsize=(4,4))
while i<a.size-1:
xs=int(a[i])
ys=int(a[i+1])
xpoints = np.array([x[xs], x[ys]])
ypoints = np.array([y[xs], y[ys]])
plt.subplot(1,1,1)
plt.plot(xpoints, ypoints,'b')
xticks(np.linspace(0,ab[0],ab[0]+1,endpoint=True))#设置xy的区间,生成axb的方格纸
yticks(np.linspace(0,ab[1],ab[0]+1,endpoint=True))
i=i+1
plt.grid()
plt.axis([0,ab[0],0,ab[1]])
plt.show()
下面展示一下运行结果:
另外附上总代码
import matplotlib.pyplot as plt
import numpy as np
import math
from pylab import *
print("请输入方格纸的大小a b")
ab=np.array((input().split()),int)
print("请输入点的个数n:")
n=int(input())
print("请输入各个点的横坐标")
x=np.array((input().split()),int)
print("请输入各个点的纵坐标")
y=np.array((input().split()),int)
l=0
k=0
#print(x,y)
length1=np.empty(math.factorial(n))
number=np.empty([math.factorial(n),n])
#d=np.empty(0.5*n*(n-1))
#------------------------------------------- 这个函数用来绘制连线图
def connect(a):
i=0
#plt.figure(figsize=(4,4))
while i<a.size-1:
xs=int(a[i])
ys=int(a[i+1])
xpoints = np.array([x[xs], x[ys]])
ypoints = np.array([y[xs], y[ys]])
#plt.xlim([0,4])
#plt.ylim([0,4])
plt.subplot(1,1,1)
plt.plot(xpoints, ypoints,'b')
xticks(np.linspace(0,ab[0],ab[0]+1,endpoint=True))
yticks(np.linspace(0,ab[1],ab[0]+1,endpoint=True))
i=i+1
plt.grid()
plt.axis([0,ab[0],0,ab[1]])
plt.show()
def check(a,b): #这个函数用来检查连接方式是否是一样的,如ABCD=DCBA 如果一样。则另两个数组变成一样的
ch=0
i=0
while i<a.size:
if(a[i]==b[a.size-1-i]):
ch=ch+1
i=i+1
if(ch==a.size):
i=0
while i<a.size:
b[i]=a[i]
i=i+1
def distance(x1,y1,x2,y2): #计算两点之间的距离
d=pow((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2),0.5)
return d
#------------------------------以下是计算长度总和
def length(a):
i=0
l=0
while i<n-1:
m1=int(a[i])
m2=int(a[i+1])
l=l+distance(x[m1],y[m1],x[m2],y[m2])
i=i+1
return l
#-----------------------------以下是不同的连线方式的顺序(其中包含了一样的连接顺序,如:ABCD=DCBA)
i=0
b=np.zeros(n)
b[n-1]=-1
c=np.zeros(n)
c[n-1]=1
cc=1
ccc=0
def carry(b,i):
b[i]=0
b[i-1]=b[i-1]+1
if(b[i-1]>=n):
carry(b,i-1)
while cc==1:
m=0
b=b+c
if(b[n-1]>=n):
carry(b,n-1)
i1=0
while i1<b.size:
j1=0
for j1 in range(i1+1,b.size):
if(b[i1]==b[j1]):
m=m+1
i1=i1+1
if(m==0):
l=length(b)
length1[k]=round(l,4)
number[k,...]=b
k=k+1
#print(float(l))
ccc=ccc+1
if(ccc==pow(n,n)):
break
#print(number)
while i<math.factorial(n)-1:
j=0
while j<math.factorial(n):
nbi=number[i,...]
nbj=number[j,...]
check(nbi,nbj)
j=j+1
i=i+1
#print(number)
newnum=np.ones([int(0.5*math.factorial(n)),n])
i=0
n1=0
while i<math.factorial(n)-1:
#j=i+1
for j in range(i+1,math.factorial(n)):
k=0
m=0
while k<n:
if(number[i,k]==number[j,k]):
m=m+1
k=k+1
if(m==n):
newnum[n1,...]=number[i,...]
n1=n1+1
j=j+1
i=i+1
print(newnum)
length2=np.empty(int(0.5*math.factorial(n)))
i=0
while i<int(0.5*math.factorial(n)):
length2[i]=length(newnum[i,...])
i=i+1
print(length2)
#--------------------------以下筛选出最小的连接方式
i=0
minvalue=length2[0]
k=0
minlength=np.ones(int(0.5*math.factorial(n)))
while i<int(0.5*math.factorial(n)):
if(length2[i]<=minvalue):
minvalue=length2[i]
i=i+1
print(minvalue)
i=0
while i<int(0.5*math.factorial(n)):
if(length2[i]==minvalue):
minlength[k]=i
minlength[k+1]=-1 #储存截止标志
k=k+1
i=i+1
print(minlength)
i=0
while i<minlength.size:
if(minlength[i]==-1):
break
nnm=newnum[int(minlength[i])]
connect(nnm)
i=i+1
#--------------------------
感谢观看!!!