调整数组顺序使奇数位于偶数前面

输入一个整数数组，实现一个函数来调整该数组中数字的顺序，使得所有奇数在数组的前半部分，所有偶数在数组的后半部分。

class Solution {
    public int[] exchange(int[] nums) {
        int a=0, b=nums.length-1;
        while (a<b) {
            while (a<b && (nums[a]&1)==1) a++;
            while (a<b && (nums[b]&1)==0) b--;

            int tmp = nums[a];
            nums[a] = nums[b];
            nums[b] = tmp;
        }
        return nums;
    }
}