[Java][Answer] CodingBat Array-1 > frontPiece()

Given an int array of any length, return a new array of its first 2 elements. If the array is smaller than length 2, use whatever elements are present.

frontPiece([1, 2, 3]) → [1, 2]
frontPiece([1, 2]) → [1, 2]
frontPiece([1]) → [1]

Answer 1:

public int[] frontPiece(int[] nums) {
  if(nums.length<2) return nums;
  return new int[]{nums[0],nums[1]};
}

Answer 2:

public int[] frontPiece(int[] nums) {
  if (nums.length == 1)
  return new int[] {nums[0]};
  else if (nums.length == 0)
  return new int[] {};
  else
  return new int[] {nums[0],nums[1]};
}

Reference
http://www.javaproblems.com/2012/12/coding-bat-java-array-1-frontpiece.html