如何将整数数组划分为2个子数组并使它们的平均值相等？

def partitionArray(nums):
    n = len(nums)
    if n == 0:
        return []
    
    total_sum = sum(nums)
    if total_sum % 2 != 0:
        return []
    
    target_sum = total_sum // 2
    dp = [[False] * (target_sum + 1) for _ in range(n)]
    
    if nums[0] <= target_sum:
        dp[0][nums[0]] = True
    
    for i in range(1, n):
        for j in range(target_sum + 1):
            if dp[i-1][j] or (j >= nums[i] and dp[i-1][j-nums[i]]):
                dp[i][j] = True
    
    if not dp[n-1][target_sum]:
        return []
    
    partition_point = target_sum
    while not dp[n-1][partition_point]:
        partition_point -= 1
    
    partition1 = []
    partition2 = []
    i = n - 1
    j = partition_point
    while i >= 0 and j >= 0:
        if i == 0 and dp[i][j]:
            partition1.append(nums[i])
            break
        if dp[i-1][j]:
            i -= 1
        else:
            partition1.append(nums[i])
            j -= nums[i]
            i -= 1
    
    for num in nums:
        if num not in partition1:
            partition2.append(num)
    
    return [partition1, partition2]