[閒聊] 每日leetcode 75 - Day8 - 7

[yam276 (史萊哲林的優等生)]

2352. Equal Row and Column Pairs


題目：

計算一個方形陣列有多少直排數字與橫排數字相等的情況

可以多 row / column 比一


思路：

先記錄 row 的次數 HashMap

然後開始組裝 column (因為比較難組)

每次組完去找 row 的同 Vec 組成的 freq

直接把次數加上去即可


Code:

use std::collections::HashMap;
impl Solution {
    pub fn equal_pairs(grid: Vec<Vec<i32>>) -> i32 {
        let n: usize = grid.len();
        let mut freq = HashMap::new();

        for i in 0..n {
            let row = grid[i].clone();
            *freq.entry(row).or_insert(0) += 1;
        }

        let mut count = 0;
        for j in 0..n {
            let mut column = Vec::new();
            for i in 0..n {
                column.push(grid[i][j]);
            }

            if let Some(&c) = freq.get(&column) {
                count += c;
            }
        }

        count
    }
}

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.32.48.170 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1749549220.A.D50.html