字典應用: 計算相同的column row pairs_Leetcode 精選75題解析

Chris Brignola on Unsplash

題目敘述

題目會給定我們一個二維陣列，要求我們計算內部元素相同的column row pairs總共有多少條?

註: pair的定義就是row i 和 column j 彼此內部元素值都相同，這樣就算一條pair。

題目的原文敘述

測試範例

Example 1:

Input: grid = [[3,2,1],[1,7,6],[2,7,7]]
Output: 1
Explanation: There is 1 equal row and column pair:
- (Row 2, Column 1): [2,7,7]

Example 2:

Input: grid = [[3,1,2,2],[1,4,4,5],[2,4,2,2],[2,4,2,2]]
Output: 3
Explanation: There are 3 equal row and column pairs:
- (Row 0, Column 0): [3,1,2,2]
- (Row 2, Column 2): [2,4,2,2]
- (Row 3, Column 2): [2,4,2,2]

約束條件

Constraints:

• n == grid.length == grid[i].length

輸入陣列一定是方陣。

• 1 <= n <= 200

方陣的邊長介於1 ~ 200之間。

• 1 <= grid[i][j] <= 10^5

元素值都介於1 ~ 10^5 之間。

演算法

這題的關鍵在於把每條row和每條column都轉換為可hash的對象，並且儲存在字典裏面，統計出現次數。

又因為記憶體排列是row-major的關係，在讀取某個column j的時候，會先把矩陣grid作轉秩grid T，再讀取grid T的row j，會比較方便。

在python，有一個很實用的語法，可以很簡潔的實現矩陣轉秩的功能。

transpose = list( zip( *grid ) )​