Submission #667483

---

Source Code Expand

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import itertools
import numpy

DS = [(1, 0), (-1, 0), (0, 1), (0, -1)]
H = 30
W = 30


class Solver(object):

    def __init__(self, stage):
        self.stage = stage
        self.hands = None
        self.init_hands()

    def init_hands(self):
        self.hands = []

    def naive_solution(self):
        for r, c in itertools.product(range(H), range(W)):
            cell = self.stage[r, c]
            if cell <= 0:
                continue
            for _ in range(cell):
                self.hands.append((r + 1, c + 1))
                self.stage[r, c] -= 1

    def use_one_adj_solution(self):
        for r0, c0 in itertools.product(range(H), range(W)):
            cell = self.stage[r0, c0]
            if cell <= 0:
                continue
            self.hands.append((r0 + 1, c0 + 1))
            self.stage[r0, c0] -= 1
            cell = self.stage[r0, c0]
            for dr, dc in DS:
                (r, c) = (r0 + dr, c0 + dc)
                if r < 0 or r >= H:
                    continue
                elif c < 0 or c >= W:
                    continue
                elif 0 < self.stage[r, c] == cell:
                    self.hands.append((r + 1, c + 1))
                    self.stage[r, c] -= 1
        self.naive_solution()


def main():
    stage = numpy.array([list(map(int, input().split())) for _ in range(H)])
    solver = Solver(stage)
    solver.use_one_adj_solution()
    for r, c in solver.hands:
        print("{} {}".format(r, c))


if __name__ == '__main__':
    main()

Submission Info

Judge Result