結果

問題 No.2277 Honest or Dishonest ?
ユーザー prin_kemkemprin_kemkem
提出日時 2023-04-22 14:49:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 758 ms / 2,000 ms
コード長 3,791 bytes
コンパイル時間 408 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 121,664 KB
最終ジャッジ日時 2024-11-08 06:52:39
合計ジャッジ時間 26,441 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 58 ms
61,312 KB
testcase_01 AC 59 ms
61,824 KB
testcase_02 AC 59 ms
61,568 KB
testcase_03 AC 646 ms
110,988 KB
testcase_04 AC 657 ms
113,944 KB
testcase_05 AC 223 ms
93,356 KB
testcase_06 AC 522 ms
101,904 KB
testcase_07 AC 470 ms
100,028 KB
testcase_08 AC 173 ms
94,980 KB
testcase_09 AC 496 ms
101,856 KB
testcase_10 AC 387 ms
93,516 KB
testcase_11 AC 278 ms
84,924 KB
testcase_12 AC 235 ms
85,296 KB
testcase_13 AC 646 ms
108,224 KB
testcase_14 AC 286 ms
87,840 KB
testcase_15 AC 210 ms
91,108 KB
testcase_16 AC 477 ms
100,372 KB
testcase_17 AC 317 ms
88,188 KB
testcase_18 AC 612 ms
112,096 KB
testcase_19 AC 635 ms
114,408 KB
testcase_20 AC 525 ms
105,348 KB
testcase_21 AC 443 ms
95,752 KB
testcase_22 AC 463 ms
100,648 KB
testcase_23 AC 470 ms
100,972 KB
testcase_24 AC 467 ms
99,196 KB
testcase_25 AC 557 ms
106,236 KB
testcase_26 AC 144 ms
92,416 KB
testcase_27 AC 351 ms
89,708 KB
testcase_28 AC 251 ms
91,244 KB
testcase_29 AC 374 ms
98,336 KB
testcase_30 AC 363 ms
90,372 KB
testcase_31 AC 527 ms
101,908 KB
testcase_32 AC 349 ms
96,184 KB
testcase_33 AC 641 ms
112,060 KB
testcase_34 AC 652 ms
113,428 KB
testcase_35 AC 123 ms
85,376 KB
testcase_36 AC 508 ms
100,012 KB
testcase_37 AC 529 ms
104,828 KB
testcase_38 AC 218 ms
82,896 KB
testcase_39 AC 292 ms
86,212 KB
testcase_40 AC 132 ms
83,572 KB
testcase_41 AC 663 ms
115,832 KB
testcase_42 AC 532 ms
103,540 KB
testcase_43 AC 681 ms
119,496 KB
testcase_44 AC 721 ms
121,044 KB
testcase_45 AC 741 ms
121,664 KB
testcase_46 AC 718 ms
120,076 KB
testcase_47 AC 758 ms
120,428 KB
testcase_48 AC 725 ms
121,204 KB
testcase_49 AC 641 ms
120,424 KB
testcase_50 AC 587 ms
119,060 KB
testcase_51 AC 719 ms
120,272 KB
testcase_52 AC 718 ms
119,772 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict, deque, Counter
import copy
from itertools import combinations, permutations, product, accumulate
from heapq import heapify, heappop, heappush
import math
import bisect
import sys
# sys.setrecursionlimit(700000)
input = lambda: sys.stdin.readline().rstrip('\n')
inf = float('inf')
mod1 = 10**9+7
mod2 = 998244353
def ceil_div(x, y): return -(-x//y)

#################################################

class Weight:
    def __init__(self, x) -> None:
        self.x = x
    def __add__(self, other):
        return Weight(self.x^other.x)
    def __sub__(self, other):
        return Weight(self.x^other.x)
    def __neg__(self):
        return Weight(self.x)
    def __eq__(self, other):
        return self.x == other.x
    def e():
        return Weight(0)

class WeightedUnionFind:
    #コンストラクタ
    def __init__(self, n, Weight=None):
        self.n = n
        self.parents = [-1]*n
        self.C = Weight
        if self.C is None:
            self.weight = [0]*n
        else:
            self.weight = [self.C.e() for _ in range(N)]

    #点xの根と根までの距離を調べる+親が根になるよう移動し辺の重みを更新
    def find(self, x):
        if self.parents[x] < 0:
            return x, 0 if self.C is None else Weight.e()
        else:
            r, w = self.find(self.parents[x])
            self.parents[x] = r
            self.weight[x] += w
            return self.parents[x], self.weight[x]

    #点x,yの属する集合同士を連結(要素数が少ない方を多い方に連結)
    #x->yの距離がwとなるように辺を張る
    #既に連結済みである場合、x->yの距離がwでなければ何もせずにFalseをreturn
    #それ以外の場合はTrueをreturn
    def union(self, x, y, w):
        rx, wx = self.find(x)
        ry, wy = self.find(y)
        if rx == ry:
            return wx-wy == w
        if self.parents[rx] > self.parents[ry]:
            self.weight[ry] = -w+wx-wy
            self.parents[rx] += self.parents[ry]
            self.parents[ry] = rx
        else:
            self.weight[rx] = w-wx+wy
            self.parents[ry] += self.parents[rx]
            self.parents[rx] = ry
        return True

    #点xが属する集合の要素数を取得
    def size(self, x):
        return -self.parents[self.find(x)[0]]

    #点x,yが同じ集合に属しているか判定
    def same(self, x, y):
        return self.find(x)[0] == self.find(y)[0]

    #点xの属する集合の全要素を取得
    def members(self, x):
        root = self.find(x)
        return [i for i in range(self.n) if self.find(i)[0] == root]

    #根になっている全要素を取得
    def roots(self):
        return [i for i, x in enumerate(self.parents) if x < 0]

    #集合の数を取得
    def group_count(self):
        return len(self.roots())

    #全集合の「根と全要素」を取得
    def get_groups(self):
        groups = defaultdict(list)
        for x in range(self.n):
            groups[self.find(x)[0]].append(x)
        return groups

    # x->yの距離を取得
    # 非連結の場合Noneをreturn
    def dist(self, x, y):
        rx, wx = self.find(x)
        ry, wy = self.find(y)
        if rx != ry:
            return None
        else:
            return wx-wy

    #print(インスタンス)で、全集合の「根と全要素」を出力
    def __str__(self):
        return '\n'.join('{}:{}'.format(r, self.menbers(r)) for r in self.roots())

N, Q = map(int, input().split())
wuf = WeightedUnionFind(N, Weight)
qs = [tuple(map(int, input().split())) for _ in range(Q)]
for a, b, c in qs:
    a -= 1; b -= 1
    if not wuf.union(a, b, Weight(c)):
        print(0)
        exit()
print(pow(2, wuf.group_count(), mod2))
0