[leetcode] 2352. Equal Row and Column Pairs

문제

 

Map에서 containsKey로 List도 확인가능하다.

22ms

class Solution {
    public int equalPairs(int[][] grid) {
        int result = 0;
        int n = grid.length;
        Map<List<Integer>, Integer> map = new HashMap<>();
        List<List<Integer>> list = new ArrayList<>();

        for(int i = 0; i < n; i++) {
            List<Integer> temp = new ArrayList<>();
            List<Integer> temp1 = new ArrayList<>();
            for(int j = 0; j < n; j++) {
                temp.add(grid[i][j]);
                temp1.add(grid[j][i]);
            }
            map.put(temp, map.getOrDefault(temp, 0) + 1);
            list.add(temp1);
        }

        for(List<Integer> l: list) {
            if (map.containsKey(l)) {
                result += map.get(l);
            }
        }

        return result;
    }
}