Algorithm
2차원 동전 뒤집기 (Python)
RetepMil
2023. 9. 19. 21:01
접근 방법
- target 보드 위에 beginning을 덮는다고 생각해보면 (XOR 연산), 어느 돌을 뒤집어야할지 표시하는 것과 비슷한 효과를 보인다.
- 같은 행 및 열을 두번 뒤집으면 안뒤집은 것과 마찬가지다. 열과 행은 한번씩만 뒤집으면 된다.
- 각 행을 기준으로 순회를 하며, 그 행과 똑같은 행이면 넘어가고 뒤집은 것과 비슷한 행은 뒤집어주고 뒤집은 갯수를 1을 추가한다. 두 조건 모두 만족하지 못한다면 정답을 찾을 수 없는 경우이기 때문에 바로 -1을 반환해준다.
- 여기서 해당 행에 1이 있다면 다시 뒤집어야 한다는 것이기 때문에 행을 뒤집은 것 + 행에 있는 1의 갯수를 기준으로 최소 값을 반환하면 정답이다.
코드
def solution(beginning, target):
xor = [[0] * len(beginning[0]) for _ in range(len(beginning))]
for i in range(len(beginning)):
for j in range(len(beginning[0])):
if beginning[i][j] != target[i][j]:
xor[i][j] = 1
else:
xor[i][j] = 0
minimum = 1_000_000
base_row = xor[0]
for row in xor:
diff = 0
for other_row in xor:
if row == other_row: continue
temp_row = [1 - i for i in other_row]
if row == temp_row:
diff += 1
continue
return -1
diff += sum(row)
if diff < minimum:
base_row = row
minimum = diff
return minimum
복기