用遗传算法分割等和子集

01背包
416. 分割等和子集

给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。

示例 1：

输入：nums = [1,5,11,5]
输出：true
解释：数组可以分割成 [1, 5, 5] 和 [11] 。

提示：

1 <= nums.length <= 200
1 <= nums[i] <= 100

import random
nums = [1,5,12,5,1,0]

population=[[0,1,1,1],[0,0,1,1],[1,0,1,1],[1,1,1,1]] # 二进制编码


population_pool=[]

def calTarget(_list:list):
    ans0=0
    ans1=0
    for i in range(len(_list)):
        if _list[i]==1:
            ans0+=nums[i]
        else:
            ans1+=nums[i]
    return abs(ans0-ans1)

def calFitness(_list):
    return sum(nums)-calTarget(_list)

def getPerson():
    index=random.randint(0,len(population_pool)-1)
    return population_pool[index]

STOP_EPOCH=100
ans=0
while True:
    ## 选择
    ## 计算适应度
    syd_list=[]
    for i in range(len(population)):
        syd_list.append((i,calFitness(population[i])))
    syd_list = sorted(syd_list, key=lambda tup: tup[1])
    population_pool=[]
    for i in syd_list[-2:]:
        population_pool.append(population[i[0]])
    ## 交配
    population=[]

    while len(population)!=4:
        gt0=getPerson()
        gt1=getPerson()
        new=[]
        start=random.randint(0,3)
        xzpd=gt0[start:start+3]
        new.extend(xzpd)
        start=random.randint(0,3)
        xzpd=gt1[start:start+3]
        new.extend(xzpd)
        population.append(new)

    for i in population:
        if calTarget(i)==0:
            print(i)
            ans=STOP_EPOCH
            break
        else:
            print(i,calTarget(i))
    if ans==STOP_EPOCH:
        break
    ans+=1