結果
問題 | No.2205 Lights Out on Christmas Tree |
ユーザー |
|
提出日時 | 2023-02-04 11:08:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 269 ms / 2,000 ms |
コード長 | 962 bytes |
コンパイル時間 | 153 ms |
コンパイル使用メモリ | 82,444 KB |
実行使用メモリ | 159,028 KB |
最終ジャッジ日時 | 2024-07-03 09:41:20 |
合計ジャッジ時間 | 9,564 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 37 |
ソースコード
import sysinput = lambda :sys.stdin.readline()[:-1]ni = lambda :int(input())na = lambda :list(map(int,input().split()))yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")#######################################################################n = ni()g = [[] for i in range(n)]deg = [0] * nfor _ in range(n-1):a,b = na()a-=1b-=1g[a].append(b)g[b].append(a)deg[a] += 1deg[b] += 1c = [i^1 for i in na()]p = [i for i in range(n) if deg[i] == 1]ans = 0while p:x = p.pop()#print(p, x)deg[x] -= 1for y in g[x]:if deg[y] >= 1:deg[y] -= 1if deg[y] == 1:p.append(y)if deg[y] >= 1 and c[x] == 1:c[x] ^= 1c[y] ^= 1ans += 1if sum(c) == 2:print(ans+1)elif sum(c) == 1:print(-1)else:print(ans)