我试图模拟社交网络。喜欢这个
A是B,C,D的朋友
B是A,D的朋友
C是A
的朋友
D是A,B的朋友
我如何在Java中实现它。我需要编写带有2个输入的函数,如下所示:
AreFriends(A,D)= 1
AreFriends(C,D)= 0
答案 0 :(得分:1)
使用Adjacency list
之类的数据结构来存储所有关系。但是对于它,您需要将所有字符串转换为可以唯一标识String
的等效整数。我们可以使用map
并相应地将整数分配给Strings
。要存储连接,我们可以使用ArrayList of HashSet
。
import java.util.*;
class Main
{
static HashMap<String,Integer> map;
static ArrayList<HashSet<Integer>> list;
static boolean check(String id1,String id2) //To check if a connection exists or not
{
int index1=map.get(id1);
int index2=map.get(id2);
return list.get(index1).contains(index2);
}
public static void main(String args[])
{
Scanner sc=new Scanner(System.in);
ArrayList<String> ids=new ArrayList<String>();
ids.add("A");ids.add("B");ids.add("C");ids.add("D"); //Taking all possible IDs
map=new HashMap<String,Integer>();
int given_id=0;
for(String id:ids)
{
if(!map.containsKey(id))
{
map.put(id,given_id); //Assigning each String a unique ID
given_id++;
}
}
list=new ArrayList<HashSet<Integer>>(); //ArrayList of HashSet is used to store the connections
for(int i=0;i<given_id;i++)
{
list.add(new HashSet<Integer>());
}
//Now for example, we store the connections in HashSet
String connections="A B A C A D B A B D C A D A D B"; //You can change the following loop as per your need
String arr[]=connections.split(" ");
for(int i=0;i<arr.length;i+=2)
{
int index1=map.get(arr[i]);
int index2=map.get(arr[i+1]);
list.get(index1).add(index2); //Adding connection in both IDs
list.get(index2).add(index1);
}
if(check("A","D"))
System.out.println("A and D are friends!");
else
System.out.println("No, A and D are not friends!");
if(check("C","D"))
System.out.println("C and D are friends");
else
System.out.println("No, C and D are not friends!");
}
}
<强>输出:强>
A and D are friends!
No, C and D are not friends!