优化这段代码for j in range(n): prob += pulp.lpSum(x[i] for i in range(n) if (i,j) in connections) >= 1
时间: 2024-05-02 19:23:30 浏览: 216
PyPI 官网下载 | prob140-0.1.6.3.tar.gz
可以将`x[i]`的求和操作放在循环外面,避免重复计算。同时,可以使用列表推导式来筛选出符合条件的`x[i]`,也能提高代码的可读性。修改后的代码如下:
```
x_sum = [x[i] for i, j in connections if j == col]
prob = pulp.lpSum(x_sum) >= 1
```
其中,`col`是循环的变量`j`,`connections`是一个包含连接信息的列表或集合。这样修改之后,每次循环只需要计算一次`x[i]`的求和。
阅读全文