贝壳找房笔试 族谱java实现

问题描述

输入输出忘了。。。。

上代码

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Scanner;

/**
 * 
 */

/**
 * @author Administrator
 * @date   2018年9月3日
 */
public class Main19 {
    public static void main(String[] args)
    {
        Scanner in=new Scanner(System.in);
        int n=in.nextInt();
        int arr[]=new int[n];
        HashMap<Integer, ArrayList<Integer>> map=new HashMap<>();
        for(int i=0;i<n;i++)
        {
            ArrayList<Integer> list2=new ArrayList<>();
            int a=in.nextInt();
            int b=in.nextInt();
            if(a==-1)
            {
                //list2.add(a);
                map.put(a, list2);
            }
            else if(b==-1)
            {
                //list2.add(a);
                map.put(b, list2);
            }
            if(map.containsKey(a))
            {
                map.get(a).add(b);
            }
            else if(map.containsKey(b))
            {
                map.get(b).add(a);
            }
            else 
            {
                if(map.get(-1).contains(a))
                {
                    list2.add(b);
                    map.put(a,list2);
                }
                else
                {
                    list2.add(a);
                    map.put(b,list2);
                }
            }
        }
        int m=in.nextInt();
        for(int i=0;i<m;i++)
        {
            int a=in.nextInt();
            int b=in.nextInt();
            if(map.containsKey(a))
            {
                if(map.get(a).contains(b))
                    System.out.println(1);
                else
                    System.out.println(0);
            }
            else if(map.containsKey(b))
            {
                if(map.get(b).contains(a))
                    System.out.println(2);
                else
                    System.out.println(0);
            }
            else {
                System.out.println(0);
            }
        }
    }
}