結果

問題 No.1078 I love Matrix Construction
ユーザー tktk_snsntktk_snsn
提出日時 2020-11-28 18:04:15
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 3,706 bytes
コンパイル時間 249 ms
コンパイル使用メモリ 11,228 KB
実行使用メモリ 134,744 KB
最終ジャッジ日時 2023-10-10 00:50:01
合計ジャッジ時間 16,881 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 16 ms
8,352 KB
testcase_01 AC 17 ms
8,680 KB
testcase_02 AC 215 ms
26,740 KB
testcase_03 AC 587 ms
56,504 KB
testcase_04 AC 817 ms
75,260 KB
testcase_05 WA -
testcase_06 AC 206 ms
26,400 KB
testcase_07 AC 82 ms
15,392 KB
testcase_08 AC 671 ms
63,884 KB
testcase_09 WA -
testcase_10 AC 1,574 ms
134,744 KB
testcase_11 AC 884 ms
78,440 KB
testcase_12 AC 1,276 ms
113,148 KB
testcase_13 AC 1,469 ms
126,832 KB
testcase_14 AC 1,003 ms
89,628 KB
testcase_15 AC 1,418 ms
119,880 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 155 ms
21,824 KB
testcase_19 AC 370 ms
39,252 KB
testcase_20 AC 358 ms
38,724 KB
testcase_21 AC 27 ms
9,436 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)


class SCC_graph(object):
    def __init__(self, n):
        """n:ノード数"""
        self.n = n
        self.edges = []

    def add_edge(self, frm, to):
        """frm -> toへ有効辺を張る"""
        self.edges.append((frm, to))

    def __csr(self):
        self.start = [0] * (self.n + 1)
        self.elist = [0] * len(self.edges)
        for frm, to in self.edges:
            self.start[frm + 1] += 1
        for i in range(1, self.n + 1):
            self.start[i] += self.start[i - 1]
        cnt = self.start[:]
        for frm, to in self.edges:
            self.elist[cnt[frm]] = to
            cnt[frm] += 1

    def __dfs(self, v):
        self.low[v] = self.now_ord
        self.order[v] = self.now_ord
        self.now_ord += 1
        self.visited.append(v)
        for i in range(self.start[v], self.start[v + 1]):
            to = self.elist[i]
            if self.order[to] == -1:
                self.__dfs(to)
                self.low[v] = min(self.low[v], self.low[to])
            else:
                self.low[v] = min(self.low[v], self.order[to])
        if self.low[v] == self.order[v]:
            while self.visited:
                u = self.visited.pop()
                self.order[u] = self.n
                self.ids[u] = self.group_num
                if u == v:
                    break
            self.group_num += 1

    def _make_scc_ids(self):
        self.__csr()
        self.now_ord = 0
        self.group_num = 0
        self.visited = []
        self.low = [0] * self.n
        self.ids = [0] * self.n
        self.order = [-1] * self.n
        for i in range(self.n):
            if self.order[i] == -1:
                self.__dfs(i)
        for i in range(self.n):
            self.ids[i] = self.group_num - 1 - self.ids[i]

    def scc(self):
        self._make_scc_ids()
        groups = [[] for _ in range(self.group_num)]
        for i in range(self.n):
            groups[self.ids[i]].append(i)
        return groups


class TwoSAT(SCC_graph):
    def __init__(self, n):
        """ n: ノード数"""
        self._n = n
        super().__init__(2 * n)

    def add_clause(self, i, f, j, g):
        """ (xi == f)∨(xj == g)というクローズを追加 """
        x = 2 * i + (0 if f else 1)
        y = 2 * j + (1 if g else 0)
        self.add_edge(x, y)
        x = 2 * j + (0 if g else 1)
        y = 2 * i + (1 if f else 0)
        self.add_edge(x, y)

    def satisfiable(self):
        """ 条件を満たす割り当てが存在するか判定する """
        self._make_scc_ids()
        self._answer = [False] * self._n
        for i in range(self._n):
            if self.ids[2 * i] == self.ids[2 * i + 1]:
                return False
            self._answer[i] = (self.ids[2 * i] < self.ids[2 * i + 1])
        return True

    def answer(self):
        """ 最後に読んだsatisfiableのクローズを満たす割り当てを返す """
        return self._answer


n = int(input())
S = list(map(int, input().split()))
T = list(map(int, input().split()))
U = list(map(int, input().split()))

ts = TwoSAT(n * n)
for s, t, u in zip(S, T, U):
    for j in range(n):
        x = (s - 1) * n + j
        y = j * n + t - 1
        if u == 0:
            ts.add_clause(x, 0, y, 0)
        elif u == 1:
            ts.add_clause(x, 1, y, 0)
        elif u == 2:
            ts.add_clause(x, 0, y, 1)
        else:
            ts.add_clause(x, 1, y, 1)

if not ts.satisfiable():
    print(-1)
    exit()

ans = [[-1] * n for _ in range(n)]
for i, a in enumerate(ts.answer()):
    x, y = divmod(i, n)
    ans[x][y] = int(a)

for a in ans:
    print(*a)
0