比较重量

小明陪小红去看钻石，他们从一堆钻石中随机抽取两颗并比较她们的重量。这些钻石的重量各不相同。在他们们比较了一段时间后，它们看中了两颗钻石g1和g2。现在请你根据之前比较的信息判断这两颗钻石的哪颗更重。 

给定两颗钻石的编号g1,g2，编号从1开始，同时给定关系数组vector,其中元素为一些二元组，第一个元素为一次比较中较重的钻石的编号，第二个元素为较轻的钻石的编号。最后给定之前的比较次数n。请返回这两颗钻石的关系，若g1更重返回1，g2更重返回-1，无法判断返回0。输入数据保证合法，不会有矛盾情况出现。

测试样例：

2,3,[[1,2],[2,4],[1,3],[4,3]],4

返回:

1

算法思想：转化为图的遍历

代码：

import java.util.*;

public class Cmp {
    public int cmp(int g1, int g2, int[][] records, int n) {
        
        HashMap<Integer, Node> nodeMap = new HashMap<Integer, Node>();
        for(int i = 0; i < records.length; i++) {
            if(!nodeMap.containsKey(records[i][0])) {
                nodeMap.put(records[i][0], new Node());
            }
            if(!nodeMap.containsKey(records[i][1])) {
                nodeMap.put(records[i][1], new Node());
            }
            
            nodeMap.get(records[i][0]).list.add(nodeMap.get(records[i][1]));
        }
        
        Node start = nodeMap.get(g1);
        Node end = nodeMap.get(g2);
        
        if(start == null || end == null) {
            return 0;
        }
        
        if(recurse(start, end, new HashSet<Node>()))
            return 1;
        else if(recurse(end, start, new HashSet<Node>()))
            return -1;
        else
            return 0;
        
    }
    
    private boolean recurse(Node start, Node end, HashSet<Node> path) {
        
        if(path.contains(start)) 
            return false;
        else
            path.add(start);
        
        if(start == null)
            return false;
        
        if(start == end)
            return true;
        
        for(Node n : start.list) {
            if(recurse(n, end, path)) {
                return true;
            }
        }
        return false;
    }
}

class Node {
    public ArrayList<Node> list = new ArrayList<Node> ();
}