[Python3]leetcode.求解一个整数数组划分为两个子数组的问题

class Solution:
    #快速排序思想进行划分A，B部分
    def Partition(self,nums,low,high):
        i, j = low, high
        pivot = nums[low] # 选择基准（一般选择为第一个元素）
        while i < j: #i，j指针同时向中间走，直到i与j碰头（i>=j）
            while i<j and nums[j] >= pivot: #nums[j]大于基准时，j指针往左走
                j = j - 1
            nums[i] = nums[j] #若nums[j]小于基准，则将它移到左侧（第一次nums【j】移动的目的地为low，后面则为上一次移动后的位置）
            
            while i<j and nums[i] <= pivot:
                i = i + 1
            nums[j] = nums[i] #同上
            
        nums[i] = pivot
        return i #返回基准位置
    def solve(self,nums):
        n = len(nums)
        low,high = 0,n-1
        flag = 1
        while flag:
            i = self.Partition(nums,low,high) #获得基准
            #若基准将nums完美分开，则直接进行s1与s2的运算
            if i == n/2-1:
                flag = 0
            elif i < n/2-1: #若基准在中点左边,则去找右边的基准
                low = i + 1
            elif i > n/2 -1:
                high = i - 1
            #若基准在中点右边，则去找左边的基准，也就是这样不断的去逼近中点的位置，直到i==n/2-1后退出while循环
        s1,s2 = 0,0
        for i in range(0,n//2):#注意range取不到ｎ//2，这里的n为len（nums）
            s1 = s1 + nums[i]
        for j in range(n//2,n):
            s2 = s2 + nums[j]
        return s2-s1
nums = [1,1,2,2]
a = Solution()
b = a.solve(nums)
print(b)