【Lintcode】737. Find Elements in Matrix

题目地址：

https://www.lintcode.com/problem/find-elements-in-matrix/description

给定一个矩阵，求出在每一行都出现过的元素。题目保证答案存在且唯一。

可以先把第一行存在一个哈希表里，并且用其记录其每个元素的出现次数为$1$，接着开始遍历矩阵的每一行，当遍历矩阵的第$i$行时（$i$从$0$开始计），对每个元素$e$，我们先看一下其是否在哈希表中，如果不在那这个元素肯定不会是最终答案；如果在，查看一下其出现在多少行中，如果恰好出现了$i$次，那说明前面每一行都有这个元素，则将其出现的行数记录加$1$。最后找出那个出现的行的个数恰好等于矩阵行数的那个元素即可。代码如下：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    /**
     * @param Matrix: the input
     * @return: the element which appears every row
     */
    public int FindElements(int[][] Matrix) {
        // write your code here
        // key代表元素，value代表这个元素在多少行出现过
        Map<Integer, Integer> map = new HashMap<>();
        for (int e : Matrix[0]) {
            map.put(e, 1);
        }
    
        for (int i = 1; i < Matrix.length; i++) {
            for (int j = 0; j < Matrix[0].length; j++) {
                int e = Matrix[i][j];
                // 如果map含e并且e出现的行数恰好是i，说明e出现在当前行之前的所有行中，
                // 则记录其出现的行数加1
                if (map.containsKey(e) && map.get(e) == i) {
                    map.put(e, i + 1);
                }
            }
        }
    
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue() == Matrix.length) {
                return entry.getKey();
            }
        }
        
        return -1;
    }
}

时间复杂度$O(mn)$，空间$O(n)$，$m$为矩阵的行数，$n$为列数。