【Leetcode】997. Find the Town Judge

题目地址:

https://leetcode.com/problems/find-the-town-judge/

给定一个二维数组,里面的数字范围是 1 ∼ N 1\sim N 1N,每一行有两个数字 ( i , j ) (i,j) (i,j)表示从 i i i j j j有一条有向边。题目保证无自环边,也没有平行边。问是否存在唯一的一个顶点,它入度等于 N − 1 N-1 N1,出度为 0 0 0

我们可以用一个数组保存每个顶点的出度减入度,看看最后数组里是否存在 N − 1 N-1 N1即可。一旦找到 N − 1 N-1 N1就可以直接返回顶点编号(这是因为,对于一个 N N N阶有向图,无自环边无平行边,如果其中某个顶点入度为 N − 1 N-1 N1,那么说明别的顶点的出度至少是 1 1 1,而每个顶点的入度最多就是 N − 1 N-1 N1,所以别的顶点入度减出度最多是 N − 2 N-2 N2。即,如果其中某个顶点入度减出度等于 N − 1 N-1 N1,那么它就是唯一的一个满足此条件的点)。代码如下:

public class Solution {
    public int findJudge(int N, int[][] trust) {
        if (trust.length < N - 1) {
            return -1;
        }
        
        int[] count = new int[N];
        for (int[] pair : trust) {
            count[pair[0] - 1]--;
            count[pair[1] - 1]++;
        }
        
        for (int i = 0; i < count.length; i++) {
            if (count[i] == N - 1) {
                return i + 1;
            }
        }
        
        return -1;
    }
}

时间复杂度 O ( n ) O(n) O(n),空间复杂度 O ( N ) O(N) O(N)

你可能感兴趣的:(LC,DFS,BFS与图论,leetcode,数据结构,算法,java)