Как нужно сделать данную задачу, я просто запутался в написании программы( map collection)

 Scanner in = new Scanner(System.in);
       TreeMap<String,List< String>>  map = new TreeMap<>();
 
List<String> list = new ArrayList();
while(in.hasNext()){
String s = in.nextLine();
String[] m = s.split(" ");
    list.add(m[1]);
map.put(m[0], list);
     List< String > friends = map.get(m[0]); 
    if (friends == null) {
friends = new ArrayList< String >(); 
friends.add(m[1]);
}
}
for (Map.Entry< String, List<String> > entry : map.entrySet()) {
System.out.println(entry.getKey()+" дружит с : "+entry.getValue());
}

List<String> strings = Arrays.asList("Petr Oleg", "Oleg Petr", "Mike Petr", "Ivan Oleg", "Petr Mike");
        Map<String, List<String>> friendship = new HashMap<>();
        strings.stream().map(in -> in.split("\\s+")).filter(ar -> ar.length == 2).forEach(ar -> {
            friendship.computeIfAbsent(ar[0], k -> new ArrayList<>());
            friendship.get(ar[0]).add(ar[1]);
        });
        friendship.entrySet().forEach(entry ->
                entry.setValue(entry.getValue().stream().filter(name -> friendship.get(name).contains(entry.getKey())).collect(toList())));
 
        friendship.entrySet().stream().filter(entry -> !entry.getValue().isEmpty()).forEach(entry -> {
            entry.getValue().forEach(name -> System.out.println(entry.getKey() + " дружит с " + name));
        });

private static Map<String, List<String>> friends = new TreeMap<>();
 
    public static void main(String[] args) {
        String[] pairs= {"Mike Nike","Oleg Petr", "Mike Petr", "Ivan Oleg"};
        for(String pair:pairs){
            addFriends(pair);
        }
        printAllFriends();
    }
 
    public static void addFriends(String pair){
        String first = pair.split(" ")[0];
        String second = pair.split(" ")[1];
        put(first,second);
        put (second,first);
    }
 
    private static void put(String one, String two){
        if (!friends.containsKey(one)){
            friends.put(one, new ArrayList<>());
        }
        friends.get(one).add(two);
    }
 
    public static void printAllFriends(){
        for (Map.Entry<String,List<String>> pair:friends.entrySet()){
            pair.getValue().sort(Comparator.naturalOrder());
            System.out.println(pair.getKey()+" дружит с "+ String.join(", ",pair.getValue()));
        }
    }

import java.util.*;
 
public class FroindData {
    TreeMap<String, ArrayList<String>> map = new TreeMap<>();
 
 
    public void add(String stroka) {
        String[] froinds = stroka.split(" ");
        if (froinds.length == 2) {
            if (this.map.get(froinds[0]) == null) {
                this.map.put(froinds[0], new ArrayList<String>(Arrays.asList(froinds[1])));
            } else {
                ArrayList<String> temp = this.map.get(froinds[0]);
                temp.add(froinds[1]);
            }
        } else {
            System.out.println("Вводите текст правильно йопта согласно заданию");
        }
    }
 
    public String get(String stroka) {
        String rsl = "";
        int k = 0;
        LinkedHashSet<String> res = new LinkedHashSet<>();
        res.addAll(Arrays.asList(stroka, " дружит с : "));
        if (this.map.containsKey(stroka)) {
            res.addAll(this.map.get(stroka));
        }
        Iterator<String> iterator = this.map.keySet().iterator();
        while (iterator.hasNext()) {
            String temp = iterator.next();
            for (int i = 0; i < this.map.get(temp).size(); i++) {
                if (stroka.contains(this.map.get(temp).get(i))) {
                    for (int j = 0; j < this.map.get(temp).size(); j++) {
                        if (res.contains(this.map.get(temp).get(i))) {
                            res.add(temp);
                        }
                        break;
                    }
 
                }
 
            }
        }
 
        Iterator<String> iter = res.iterator();
        while (iter.hasNext()) {
            rsl += iter.next();
            if (k > 1 && iter.hasNext()) {
                rsl += ", ";
            }
            k++;
        }
        return rsl;
    }
}

import org.junit.Test;
 
import static org.hamcrest.Matchers.is;
import static org.junit.Assert.*;
 
public class FroindDataTest {
 
    @Test
    public void add() {
        FroindData data = new FroindData();
        String one = "Mike Nike";
        String two = "Oleg Petr";
        String fri = "Mike Petr";
        String five = "Ivan Oleg";
        data.add(one);
        data.add(two);
        data.add(fri);
        data.add(five);
        System.out.println(data.get("Ivan"));
        System.out.println(data.get("Mike"));
        System.out.println(data.get("Nike"));
        System.out.println(data.get("Oleg"));
        System.out.println(data.get("Petr"));
 
        assertThat(data.get("Ivan"), is("Ivan дружит с : Oleg"));
        assertThat(data.get("Mike"), is("Mike дружит с : Nike, Petr"));
        assertThat(data.get("Nike"), is("Nike дружит с : Mike"));
        assertThat(data.get("Oleg"), is("Oleg дружит с : Petr, Ivan"));
        assertThat(data.get("Petr"), is("Petr дружит с : Mike, Oleg"));
    }
}