#861332 (PyPy3) No.2277 Honest or Dishonest ?

• 問題: No.2277 Honest or Dishonest ?
• ユーザー: shotoyoo
• 提出日時: 2023-04-21 21:48:53
• 言語: PyPy3 (7.3.15)
• 結果: AC
• 実行時間: 361 ms / 2,000 ms
• コード長: 2,571 bytes
• コンパイル時間: 278 ms
• コンパイル使用メモリ: 82,048 KB
• 実行使用メモリ: 110,848 KB
• 最終ジャッジ日時: 2024-11-08 06:29:01
• 合計ジャッジ時間: 14,721 ms
• ジャッジサーバーID （参考情報）: judge1 / judge4

ファイルパターン	結果
sample	AC * 3
other	AC * 50

ソースコード

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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import sys, random
input = lambda : sys.stdin.readline().rstrip()
write = lambda x: sys.stdout.write(x+"\n"); writef = lambda x: print("{:.12f}".format(x))
debug = lambda x: sys.stderr.write(x+"\n")
YES="Yes"; NO="No"; pans = lambda v: print(YES if v else NO); INF=10**18
LI = lambda : list(map(int, input().split())); II=lambda : int(input()); SI=lambda : [ord(c)-ord("a") for c in input()]
def debug(_l_):
    for s in _l_.split():
        print(f"{s}={eval(s)}", end=" ")
    print()
def dlist(*l, fill=0):
    if len(l)==1:
        return [fill]*l[0]
    ll = l[1:]
    return [dlist(*ll, fill=fill) for _ in range(l[0])]
class UF:
    # unionfind
    def __init__(self, n):
        self.n = n
        self.parent = list(range(n))
        self.size = [1] * n
#         self.es = [0] * n
    def check(self):
        return [self.root(i) for i in range(self.n)]
    def root(self, i):
        inter = set()
        while self.parent[i]!=i:
            inter.add(i)
            i = self.parent[i]
        r = i
        for i in inter:
            self.parent[i] = r
        return r
    def merge(self, i, j):
        # 繋いだかどうかを返す
        ri = self.root(i)
        rj = self.root(j)
        if ri==rj:
#             self.es[ri] += 1
            return False
        if self.size[ri]>self.size[rj]:
            ri,rj = rj,ri
        self.parent[ri] = rj
        self.size[rj] += self.size[ri]
#         self.es[rj] += self.es[ri] + 1
        return True
    def rs(self):
        return sorted(set([self.root(i) for i in range(self.n)]))
    
n,q = list(map(int, input().split()))
uf = UF(n)
ab = []
for i in range(q):
    a,b,c = LI()
    a -= 1
    b -= 1
    if c==0:
        uf.merge(a,b)
    else:
        ab.append((a,b))
rs = uf.rs()
ns = {r: [] for r in rs}
M = 998244353
for a,b in ab:
    ra = uf.root(a)
    rb = uf.root(b)
    if ra==rb:
        ans = 0
        break
    else:
        ns[ra].append(rb)
        ns[rb].append(ra)
else:
    num = 0
    done = {}
    for r in rs:
        if r in done:
            continue
        q = [r]
        done[r] = 0
        num += 1
        ng = 0
        while q:
            u = q.pop()
            for v in ns[u]:
                if v not in done:
                    done[v] = done[u]^1
                    q.append(v)
                else:
                    if done[v]==done[u]:
                        ng = 1
                        break
            if ng:
                break
        if ng:
            ans = 0
            break
    else:
        ans = pow(2, num, M)
print(ans%M)
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX