1. 程式人生 > >20190110-用笨辦法找到二維矩陣的鞍點

20190110-用笨辦法找到二維矩陣的鞍點

元素 else 數組 turn code pre 長度 poi app

1:找出一個多維數組的鞍點,即該元素在該行上最大,在該列上最小,也可能沒有鞍點

a = [
[1,2,3,4],
[4,5,6,2],
[7,0,5,2],
[11,10,7,9]]

解題思路如下:

step1:先找出每行的最大值,及其index,輸出最大值,即所在行和列作為要給list存如result列表中,函數寫法如下:

def find_saddle_max_point(s):
    result =[]
    #用result存儲每行最大值以及其index
    for i in range(len(s)):
        #遍歷次數,即一共有多少個行,如例子中一共有4行,該值等於a的元素個數
max =s[i][0] #設定每行的最大值為第一個 location =[] #新建一個location列表來存儲該值在外層的index和內層的index index=0 #設定最大值所在的列的index值為0 for j in range(len(s[i])): #遍歷每行的每個值 if s[i][j]>max: max = s[i][j] index =j #找出最大值以及最大值所在列,即j的值
location.append(i) location.append(index) #存入橫向列表坐標和縱向坐標值 result.append((max,location)) #將最大值及所在的location(包括橫向坐標和縱向坐標)存如result中 return result print(find_saddle_max_point(a))

step2:找出二維矩陣每列最小值,函數寫法同step1

def find_saddle_min_point(s):
    result =[]
    for i in range(len(s[0])):
        
#遍歷列的次數,如a有4列,列的遍歷次數等於a的嵌套列表的長度 min = s[0][i] #設定最小值為每列第一個 location=[] index=0 for j in range(len(s)): #遍歷每列的值 if s[j][i]<min: min =s[j][i] index = j location.append(index) location.append(i) #將每列的最小值所在的index存如location列表中 result.append((min,location)) #將每列的最小值以及所在的location存如result中 return result print(find_saddle_min_point(a))

step3:遍歷step1和step2生成的列表,如果有最大值和最小值重復,並且index一樣,即2個列表中有重復元素則為二維數組的鞍點

#step3:如果最小值與最大值重復,並且其index一樣,則為鞍點
def find_saddle(a):
    list1 = find_saddle_max_point(a)
    list2 = find_saddle_min_point(a)
    for i in list1:
        if i in list2:
            return i
    else:
        return False
print(find_saddle(a))

20190110-用笨辦法找到二維矩陣的鞍點