华为机试---社交网站好友推荐



描述：  现有一个社交网站，其好友推荐策略为：用户A和用户B不是好友，当二人的共同好友数量超过好友推荐阈值m时，就向A和B分别推荐为彼此好友。
本题任务为：对设定的m值，给定一组用户及各自好友列表，对这一组用户，反复自动应用上述好友推荐策略后（假设每次推荐都被采纳），求指定用户的最终好友列表。
注：好友关系是双向的，即：如果用户A是用户B的好友，那么用户B一定也是用户A的好友。
 
写一个程序，在社交网络中实现：
1）初始化社交网络
2）创建用户
3）增加指定两个用户之间的好友关系
4）反复自动应用好友推荐策略后，获取某个用户的好友数量
5）反复自动应用好友推荐策略后，判断某两个用户间是否存在好友关系
   
说明：
1、一个用户有且只有一个名字，且不存在重名
2、自己和自己不存在好友关系
3、用户名字大小写敏感
4、用户名字字符串长度范围为[1..20]
5、用户总数小于100个
输入： 五个整数，好友推荐阈值P，创建用户数量m，增加指定两个用户之间的好友关系数量M，查询某个用户的好友数量n，查询指定两个用户是否是好友N字符串，每个数据一行，按到上面的顺序依次输入数据，共m+M+n+N行字符串
字符串，每个一行，共m+M+n+N行字符串

输出： 输出用户的好友数量，共n个，每个一行；如果用户不存在，输出-1，否则输出好友数量。样例中的用户Jack、Peter、Tom的好友数量都是2个。
输出指定两个用户是否是好友，共N个，每个一行，如果是输出0，否则输出-1。样例中的用户Jack与Peter、Peter与Tom、Jack与Tom都是好友关系，所有输出0。
样例一输入： 
2 3 3 3 3
Jack
Peter
Tom
Jack Peter
Peter Tom
Jack Tom
Jack
Peter
Tom
Jack Peter
Peter Tom
Jack Tom

样例输出： 
2
2
2
0
0
0
样例二输入：
2 4 5 3 3
Jack
Peter
Tom
Lisi
Jack Peter
Peter Tom
Jack Tom
Lisi Peter
Lisi Tom
Jack
Peter
Tom
Jack Peter
Peter Tom
Jack Tom
样例二输出：
3
3
3
0
0
0

import java.util.*;
public class Main{
 public static void main(String[] args){
  Scanner scan = new Scanner(System.in);
  while(scan.hasNext()){
   int p = scan.nextInt();//好友推荐阈值;
   int m = scan.nextInt();//用户数量
   int M = scan.nextInt();//增加指定两个用户之间的好友关系数量
   int n = scan.nextInt();//查询某个用户的好友数量n
   int N = scan.nextInt();//查询指定两个用户是否是好友N字符串
   scan.nextLine();
   //用户姓名 m个
   String[] user_name = new String[m];
   for(int i = 0 ; i < m ; i++){
    user_name[i] = scan.nextLine();
   }
   //用户关系 M条
   Map<String , HashSet<String>> map = new HashMap<String , HashSet<String>>();
   for(int i = 0 ; i < m ; i++){
    HashSet<String> set = new HashSet<String>();
    map.put(user_name[i], set);
   }
   //添加关系
   for(int i = 0 ; i < M ; i++){
    String[] relation = scan.nextLine().split(" ");
    map.get(relation[0]).add(relation[1]);
    map.get(relation[1]).add(relation[0]);
   } 
   //如果用户A和用户B不是好友，当两人共同好友数量超过阈值P时，推荐为彼此好友
   recommendedFriend(p , m , user_name ,map);
   //查询某个用户的好友数量
   String[] query = new String[n];
   for(int i = 0 ; i < n ; i++){
    query[i] = scan.nextLine();
   }
   //打印指定用户好友数量
   printNumberOfFriend(query , n , map);   
   //查询指定两个用户是否是好友
   String[] isFriend = new String[N];
   for(int i = 0 ; i < N ; i++){    
    isFriend[i] = scan.nextLine();
   }   
   queryIsFriend(isFriend , N , map);
  }
  scan.close();
 }
 private static void queryIsFriend(String[] isFriend , int N , Map<String , HashSet<String>> map){
  String[] judge_relation = new String[2];
  for(int i = 0 ; i < N ; i++){
   judge_relation = isFriend[i].split(" ");
   if(map.get(judge_relation[0]).contains(judge_relation[1])){
    System.out.println("0");
   }else{
    System.out.println("-1");
   }
  }
 }
 private static void printNumberOfFriend(String[] query , int n ,Map<String , HashSet<String>> map){
  for(int i = 0 ; i < n ; i++){
   //如果用户存在
   if(map.containsKey(query[i])){
    System.out.println(map.get(query[i]).size());
   }else{
    System.out.println("-1");
   }   
  }
 }
 private static void recommendedFriend(int p , int m , String[] user_name ,Map<String , HashSet<String>> map){
  //尝试给m个用户推荐好友
  //1.本身不是好友 2.好友数量达到阈值
  int count = 0;//统计共同好友的个数
  for(int i = 0 ; i < m - 1 ; i++){
   for(int j = i + 1 ; j < m ; j++){
    //彼此不是对方的好友
    if(map.get(user_name[i]).contains(user_name[j])){
     continue;
    }else{
     for(int k = 0 ; k < m ; k++){
      if(map.get(user_name[i]).contains(user_name[k]) && map.get(user_name[j]).contains(user_name[k])){
       count++;
      }
     }
     //如果共同好友数量大于阈值，推荐为彼此好友
     if(count >= p){
      map.get(user_name[i]).add(user_name[j]);
      map.get(user_name[j]).add(user_name[i]);
     }
    }
   }
  }
 }
}