1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
#!/usr/bin/env python3
from collections import deque
from functools import reduce
from operator import mul, sub
def product(a): return reduce(mul, a)
def diff(side0, side1):
return sum(abs(sub(*i)) for i in zip(reversed(side0), side1))
def sides(r, c):
# Above, right, left, below
l = [(r - 1, c), (r, c + 1), (r + 1, c), (r, c - 1)]
return [(i, (r, c))
for i, (r, c) in enumerate(l) if 0 < r <= R and 0 < c <= C]
def update_avail(image, r, c):
board[(r, c)] = image
for i in sides(r, c):
if i[1] not in board:
for j in range(1, 5):
try:
avail[j].remove(i[1])
except:
pass
avail[sum(j[1] in board for j in sides(*i[1]))].append(i[1])
class Image:
def __init__(self, filename, index):
with open(filename) as f:
self.top = deque(map(int, f.readline().split()))
self.right = deque([self.top[-1]])
self.left = deque([self.top[0]])
for line in f:
tmp = deque(map(int, line.split()))
self.right.append(tmp[-1])
self.left.appendleft(tmp[0])
tmp.reverse()
self.bottom = tmp
self.sides = deque([self.top, self.right, self.bottom, self.left])
self.index, self.used, self.rotation = index, False, 0
def rotate(self, n=1):
self.sides.rotate(n)
self.rotation += n * 90
for test in range(26):
print(test)
with open('input{}/info.txt'.format(test)) as f:
N, K, R, C = map(int, f.readline().split())
rotated = bool(int(f.readline()))
x, rx, cx = map(int, f.readline().split())
size = R * C
images = [Image('input{}/{}.txt'.format(test, j), j)
for j in range(1, N + 1)]
avail, board = {1: deque(), 2: deque(), 3: deque(), 4: deque()}, {}
update_avail(images.pop(x - 1), rx, cx)
while len(board) < size:
for x in range(4, 0, -1):
if len(avail[x]): break
coordinate = avail[x].popleft()
filled = [(i, s) for i, s in sides(*coordinate) if s in board]
best = None
for idx, img in enumerate(images):
if rotated:
for rot in range(4):
tmp = product(diff(board[s].sides[i - 2], img.sides[i])
for i, s in filled)
if best is None or tmp < best:
best, rotation, index = tmp, rot, idx
img.rotate()
img.rotate(rotation)
else:
tmp = product(diff(board[s].sides[i - 2], img.sides[i])
for i, s in filled)
if best is None or tmp < best: best, index = tmp, idx
print(images[index].index, images[index].rotation % 360, x, coordinate, filled)
update_avail(images.pop(index), *coordinate)
with open('Cxy/{}.out'.format(test), 'w') as f:
for idx, img in board.items():
print(img.index, *idx, img.rotation % 360, file=f)
|
