結果
問題 |
No.1420 国勢調査 (Easy)
|
ユーザー |
|
提出日時 | 2021-03-15 09:45:25 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,415 bytes |
コンパイル時間 | 229 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 272,084 KB |
最終ジャッジ日時 | 2024-11-06 22:54:53 |
合計ジャッジ時間 | 10,354 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 10 TLE * 1 -- * 19 |
ソースコード
import sys input=sys.stdin.readline def main1(n,m,aby): g=[[] for _ in range(n)] for a,b,y in aby: a,b=a-1,b-1 g[a].append([b,y]) g[b].append([a,y]) x=[0]*n for i in range(30): bit=1<<i mi=set(range(n)) while mi: v0=mi.pop() grp=set() flg=True for d in (0,bit): flg=True seen=[-1]*n seen[v0]=d todo=[[v0,-1]] while todo: v,p=todo.pop() mi.discard(v) grp.add(v) for nv,y in g[v]: if nv==p:continue elif seen[nv]==-1: seen[nv]=seen[v]^(y&bit) todo.append([nv,v]) elif seen[nv]==seen[v]^(y&bit): continue else: flg=False break if not flg:break #うまくいかなかったのでwhileループを抜け、次のforに移る。 if flg:break # うまくいくケースが見つかったのでforを出る。 for v in grp: if seen[v]>0: x[v]+=bit if not flg:return [-1] return x def chk(n,aby,x): for a,b,y in aby: a,b=a-1,b-1 if x[a]^x[b]==y: pass else: return False return True if __name__=='__main__': n,m=map(int,input().split()) aby=[] for _ in range(m): a,b=map(int,input().split()) y=int(input()) aby.append([a,b,y]) ret1=main1(n,m,aby) print(*ret1,sep='\n')