Печать уровня соединений друзей по уровню

Question

Недавно я столкнулся с этим вопросом в интервью и сделал очень плохо.

Setup: 
Assume primitive Facebook. FB has Members. 

class Member { 
String name; 
String email; 
List<Member> friends; 
} 

Вопрос: Код printSocialGraph (член м). Прямыми друзьями m являются друзья Уровня 1. Друзья друзей - друзья второго уровня ..... и т. Д. Уровень печати 1 друга. Тогда уровень печати 2 друзей .... и так далее

void printSocialGraph (Member m){ 
//Your code here 
} 

Я пытался сохранить очереди для хранения своих друзей на каждом уровне, но я не получил его очень далеко. Любая идея, как мы могли бы решить ее при всех условиях проверки ошибок?

Answer 1

Просто добавьте новых детей в конце очереди поиска, пока их больше нет. Так как граф может содержать циклы, вы также должны сломать их с помощью «посещенного» набора.

void printSocialGraph (Member m) { 
    Queue<Member> queue = new LinkedList<>(); 
    Set<Member> visited = new HashSet<>(); 
    queue.add(m); 
    while (!queue.isEmpty()) { 
    Member member = queue.getFirst(); 
    if (!visited.contains(member)) { 
     visited.add(member); 
     System.out.println(member.name + ' ' + member.email); 
     if (member.friends != null) { 
     queue.addAll(member.friends); 
     } 
    } 
    } 
} 

Если вам также нужно распечатать, какой уровень член включен, то самый простой, вероятно, использовать две очереди: одна для текущего уровня, а другой для следующего, обменивая очереди между ними.