leetcode 2352. Equal Row and Column Pairs(相等的行和列)

leetcode 2352. Equal Row and Column Pairs(相等的行和列)_第1张图片
leetcode 2352. Equal Row and Column Pairs(相等的行和列)_第2张图片

找出相同的行和列有几对。
相同指的是一整行和一整列的元素都相同。

思路:

把一整行一整列的元素以String的形式保存进HashMap.
然后找是否存在这个String即可。

public int equalPairs(int[][] grid) {
    int n = grid.length;
    int res = 0;

    HashMap<String, Integer> map = new HashMap<>();
    for(int i = 0; i < n; i++) {
        StringBuilder sb = new StringBuilder();
        for(int j = 0; j < n; j++) {
            sb.append(grid[i][j]);
            sb.append(",");
        }
        
        map.put(sb.toString(), map.getOrDefault(sb.toString(), 0)+1);
    }

    for(int i = 0; i < n; i++) {
        StringBuilder sb = new StringBuilder();
        
        for(int j = 0; j < n; j++) {
            sb.append(grid[j][i]);
            sb.append(",");
        }
        
        //System.out.println(sb.toString());
        res += map.getOrDefault(sb.toString(), 0);
    }
    return res;
    
}

用char数组比StringBuilder更快。
但是很好奇题目中说grid[i][j] <= 105, 怎么没有超出char的范围。

    public int equalPairs(int[][] grid) {
        int res = 0;
        Map<String, Integer> mp = new HashMap<>();

        for (int col = 0; col < grid[0].length; col++) {
            char[] colArr = new char[grid.length];

            for (int row = 0; row < grid.length; row++) {
                colArr[row] = (char) grid[row][col];
            }
            String colStr = String.valueOf(colArr);
            mp.put(colStr, mp.getOrDefault(colStr, 0) + 1);
        }
        for (int row = 0; row < grid.length; row++) {
            char[] rowArr = new char[grid[0].length];

            for (int col = 0; col < grid[0].length; col++) {
                rowArr[col] = (char) grid[row][col];
            }
            String rowStr = String.valueOf(rowArr);

            if (mp.containsKey(rowStr)) {
                res += mp.get(rowStr);
            }
        }
        return res;
    }

你可能感兴趣的:(leetcode,java,算法)