贝壳找房笔试族谱java实现

最新推荐文章于 2024-06-02 08:15:00 发布

原创最新推荐文章于 2024-06-02 08:15:00 发布 · 1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#算法

本文介绍了一种使用Java实现的图关系查询方法。通过构建一个包含整数节点和边的数据结构，利用HashMap存储节点间的关系，并实现了查询两个节点之间是否存在直接连接的功能。适用于需要快速查询节点间关系的应用场景。

问题描述

输入输出忘了。。。。

上代码

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);
            }
        }
    }
}