The challenge
Can you find the needle in the haystack?
Write a function findNeedle()
that takes an array
full of junk but containing one "needle"
After your function finds the needle it should return a message (as a string) that says:
"found the needle at position "
plus the index
it found the needle, so:
findNeedle(new Object[] {"hay", "junk", "hay", "hay", "moreJunk", "needle", "randomJunk"})
should return "found the needle at position 5"
The solution in Java code
public class Solution { public static String findNeedle(Object[] haystack) { for (int i=0; i<haystack.length; i++) { if (haystack[i] == "needle") return "found the needle at position "+i; } return null; } }
A more succinct solution, without using a for loop:
import java.util.*; public class Solution { public static String findNeedle(Object[] haystack) { return "found the needle at position " + Arrays.asList(haystack).indexOf("needle"); } }
Test cases to validate our solution
import org.junit.Test; import static org.junit.Assert.assertEquals; public class NeedleExampleTests { @Test public void tests() { Object[] haystack1 = {"3", "123124234", null, "needle", "world", "hay", 2, "3", true, false}; Object[] haystack2 = {"283497238987234", "a dog", "a cat", "some random junk", "a piece of hay", "needle", "something somebody lost a while ago"}; Object[] haystack3 = {1,2,3,4,5,6,7,8,8,7,5,4,3,4,5,6,67,5,5,3,3,4,2,34,234,23,4,234,324,324,"needle",1,2,3,4,5,5,6,5,4,32,3,45,54}; assertEquals("found the needle at position 3", Solution.findNeedle(haystack1)); assertEquals("found the needle at position 5", Solution.findNeedle(haystack2)); assertEquals("found the needle at position 30", Solution.findNeedle(haystack3)); } }