小编use*_*901的帖子

给定一个整数数组,我需要返回一个包含原始数组中间元素的新数组.具体来说,如果原始数组的长度是奇数,则结果将具有一个元素,如果是偶数,则结果将具有两个元素.

这是我现在的代码,适用于偶数长度的数组.如何使它适用于奇数长度的数组？

public int[] makeMiddle(int[] nums) {
    int[] a = new int[2];
    if(nums.length>1) {
        a[1]=nums[nums.length/2];
        a[0]=nums[nums.length/2-1];
        return a;
    } else {
        a[2]=nums[((nums.length+1)/2) -1];
    }
    return a;
}