算法93---修改矩陣

1、題目：修改矩陣（美團2019春招實習筆試）

思路：採用兩個矩陣存儲這兩個不一樣的數字

代碼：

from collections import Counter

n, m = input().split()
n, m = int(n), int(m)
arr = []
for i in range(n):
    arr.append(list(map(int, input().split())))


def bwMatrix(arr, n, m):
    res = 0
    one, two = [], []
    for i in range(n):
        for j in range(m):
            if (i + j) & 1:
                one.append(arr[i][j])
            else:
                two.append(arr[i][j])
    dicone = Counter(one)
    dictwo = Counter(two)
    sort_one = sorted(dicone.items(), key=lambda x: x[1], reverse=True)
    sort_two = sorted(dictwo.items(), key=lambda x: x[1], reverse=True)
    
    sort_one.append([float('inf'),0])
    sort_two.append([-float('inf'),0])
    ##若是one中最多的數和two中最多的數不等
    if sort_one[0][0] != sort_two[0][0]:
        return (len(one) - sort_one[0][1]) + (len(two) - sort_two[0][1])
    ##若是相等，分三種狀況，最多數的個數不等（one多、two多），最多數的個數相等（比較第二多數的個數大小）。
    else:
        if sort_one[0][1] > sort_two[0][1]:
            return (len(two) - sort_two[1][1]) + (len(one) - sort_one[0][1])
        elif sort_one[0][1] < sort_two[0][1]:
            return (len(one) - sort_one[1][1]) + (len(two) - sort_two[0][1])
        else:
            if sort_one[1][1] >= sort_two[1][1]:
                return (len(two) - sort_two[0][1]) + (len(one) - sort_one[1][1])
            else:
                return (len(two) - sort_two[1][1]) + (len(one) - sort_one[0][1])


print(bwMatrix(arr, n, m))
# 3 3
# 1 1 1
# 1 5 1
# 1 1 1