Get Nth Element of Linked List

package org.wesome.dsalgo;

import lombok.Data;

import java.util.Objects;

public class LinkedList {
    static Node head;
    static int size = 0;

    public static int getNth(int index) {
        System.out.println("LinkedList.getNth");
        Node current = head;
        int count = 1;
        while (Objects.nonNull(current)) {
            if (count == index) {
                System.out.println(count + " element is at = " + current.data);
                return index;
            }
            count++;
            current = current.next;
        }
        return 0;
    }

    //this method print the list using loop
    static void printIteratively() {
        System.out.println("LinkedList.printIteratively");
        if (Objects.isNull(head)) {
            System.out.println("Linked List is null = " + head);
            return;
        }
        Node tempNode = head;
        System.out.println("Linked List Size is " + size);
        while (Objects.nonNull(tempNode)) {
            System.out.println("Node = " + tempNode.data);
            tempNode = tempNode.next;
        }
    }

    static void addInLast(int element) {
        System.out.println("\nLinkedList.addInLast " + element);
        Node newNode = new Node(element);
        if (Objects.isNull(head)) {
            head = newNode;
            return;
        }
        Node tempNode = head;
        while (Objects.nonNull(tempNode.next)) {
            tempNode = tempNode.next;
        }
        tempNode.next = newNode;
        size++;
    }
}

@Data
class Node {
    int data;
    Node next;

    public Node(int data) {
        this.data = data;
    }
}
package org.wesome.dsalgo;

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import static org.wesome.dsalgo.LinkedList.getNth;
import static org.wesome.dsalgo.LinkedList.printIteratively;

public class LinkedListTest {
    @Test
    void getNthTest() {
        addInLast();
        printIteratively();
        Assertions.assertEquals(3, getNth(3));
    }

    void addInLast() {
        LinkedList.addInLast(1);
        LinkedList.addInLast(2);
        LinkedList.addInLast(3);
        LinkedList.addInLast(4);
        LinkedList.addInLast(5);
    }
}
plugins {
    id 'java'
    id "io.freefair.lombok" version "6.2.0"
}

group = 'org.wesome'
version = '0.0.1-SNAPSHOT'
sourceCompatibility = JavaVersion.VERSION_1_8

configurations {
    compileOnly {
        extendsFrom annotationProcessor
    }
}

repositories {
    mavenCentral()
}

dependencies {
    testImplementation 'org.junit.jupiter:junit-jupiter:5.6.2'
}

test {
    useJUnitPlatform()
}

 

follow us on