조합 최적화로 혼자 해줘.
Advent Calendar 22일 기사 조합 최적화로 부등식 풀기
이게 뭐야
혼자 해줘, 파이썬으로 조합 최적화 모델을 만들어 풀어줍니다.
푸는 재미는 모델링을 고안하는 것입니다.
자신도 시도하고 싶은 사람은 아래를 참조하십시오.
문제
입력 파라미터
data
에 힌트가 있다고 가정합니다.파이썬
import numpy as np
from itertools import chain, product
from pulp import LpProblem, lpSum, value
from ortoolpy import addbinvars, unionfind
data = np.array([list(s) for s in """\
18626753
31118222
83247651
37583314
54467182
71432535
22834475
22314465""".splitlines()]).astype(int)
nn = len(data)
파이썬으로 풀기
수리 모델을 만들고 풀어 봅시다.
변수
제약
파이썬
m = LpProblem()
v = np.array(addbinvars(nn, nn)) # 0:number, 1:black (1)
for i, j in product(range(nn), range(nn)):
for x in [v[i,:][data[i,:] == j+1], v[:,i][data[:,i] == j+1]]:
m += lpSum(x) >= len(x)-1 # (2)
for e in chain((v[1:,:] + v[:-1,:]).flat, (v[:,1:] + v[:,:-1]).flat):
m += e <= 1 # (3)
while True:
m.solve()
r = np.vectorize(value)(v)
if unionfind.isconnected(1-r):
break
m += lpSum(v[r==1]) <= r.sum() - 1 # (4)
결과 표시
파이썬
data[r==1] = 0
print(data)
>>>
[[1 8 0 2 6 7 5 3]
[3 0 1 0 8 0 2 0]
[8 3 2 4 7 6 0 1]
[0 7 5 8 3 0 1 4]
[5 4 0 6 0 1 8 2]
[7 1 4 0 2 5 3 0]
[2 0 8 3 4 0 7 5]
[0 2 3 1 0 4 6 0]]
풀리고 있는지 확인할 수 있습니다.
이상
Reference
이 문제에 관하여(조합 최적화로 혼자 해줘.), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://qiita.com/SaitoTsutomu/items/727493d0437807b28c9d텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)