CustomLinkedList/MyFriendsLinkedList.java at master · mrshubhamtyagi/CustomLinkedList · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
public class MyFriendsLinkedList {
    static int totalFriends = 0;

    private Node headNode;

    MyFriendsLinkedList() {
        headNode = null;
    }

    public void addFriend(Friend friend) {

        if (headNode == null) {
            headNode = new Node(friend);
        } else {
            Node traverse = headNode;
            while (traverse.link != null) {
                traverse = traverse.link;
            }
            Node newNode = new Node(friend);
            traverse.link = newNode;
        }
        totalFriends++;
    }

    public void addAllFriends(Friend... friend) {
        Node newNode;
        for (Friend obj : friend) {
            newNode = new Node(obj);
            if (headNode == null) {
                headNode.link = newNode;
            } else {
                Node traverse = headNode;
                newNode = new Node(obj);
                while (traverse.link != null) {
                    traverse = traverse.link;
                }
                traverse.link = newNode;
            }
            totalFriends++;
        }
        return;
    }

    public boolean removeFriend(String fullName) {
        if (fullName.trim().equals(headNode.data.getName())) {
            headNode = headNode.link;
            totalFriends--;
            return true;
        } else {
            Node traverse = headNode;
            while (traverse.link != null) {
                if (traverse.link.data.getName().equals(fullName.trim())) {
                    traverse.link = traverse.link.link;
                    totalFriends--;
                    return true;
                }
                traverse = traverse.link;
            }
            return false;
        }
    }

    public void printList() {
        Node traverse = headNode;
        while (traverse != null) {
            System.out.println(traverse.data.toString());
            traverse = traverse.link;
        }
    }

    public int totalFriends() {
        return totalFriends;
    }
}


class Node {
    Friend data;
    Node link;

    Node(Friend friend) {
        data = friend;
        link = null;
    }
}


class Friend {

    private String name;
    private double friendshipAge;

    Friend(String name, double friendshipAge) {
        this.name = name;
        this.friendshipAge = friendshipAge;
    }

    public String getName() {
        return name;
    }

    public double getFriendshipAge() {
        return friendshipAge;
    }


    public void setName(String name) {
        this.name = name;
    }

    public void setFriendshipAge(double friendshipAge) {
        this.friendshipAge = friendshipAge;
    }


    @Override
    public String toString() {
        String years = " years";
        if (this.friendshipAge == 1) years = " year";
        else if (this.friendshipAge < 1) years = " months";
        return "Name -> " + this.name + "\nFriendship Age -> " + this.friendshipAge + years + "\n";
    }
}