实现一个函数partition，传入int类型的数组和数组的长度size，把第一个元素成为基准，用基准将数组划分成三部分，前半部分小于等于基准，中间部分是基准，最后一部分大于基准，返回划分好数组后基准的下标，在原有数组上划分

样例

数组 5 1 2 3 7 8 9
划分后的数组为 1 2 3 5 7 8 9，返回值为3

参考格式:

Java:

class Solution {
    public int partition(int [] nums, int size)
    {
        // 函数体
    }
}

C++

class Solution 
{
public:
    int partition(int nums [], int size)
    {
       // 函数体
    }
};

C

int partition(int nums [], int size)
{
    // 函数体
}

Python:

def partition(nums, size):
    # 自己实现