如何使用or-tools在布尔数组中生成K个不同的元素？

from ortools.sat.python import cp_model

model = cp_model.CpModel()
solver = cp_model.CpSolver()

n = len(bool_array)  # 布尔数组的长度

selected = [model.NewBoolVar(f'selected_{i}') for i in range(n)]  # 定义一个布尔变量列表，表示每个元素是否被选择

model.Add(sum(selected) == k)  # 约束条件：选择的元素个数为K
model.Add(sum(selected[i] for i in range(n) if bool_array[i]) == sum(selected))  # 约束条件：选择的元素必须在原始数组中存在

model.Maximize(sum(selected[i] for i in range(n) if bool_array[i]))  # 最大化选择的元素个数

solver.Solve(model)

result = [i for i in range(n) if bool_array[i] and solver.Value(selected[i]) == 1]  # 获取选择的元素的索引列表