結果

問題 No.2780 The Bottle Imp
ユーザー timi
提出日時 2024-06-07 22:35:37
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 2,275 bytes
コンパイル時間 409 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 111,672 KB
最終ジャッジ日時 2024-12-27 14:11:28
合計ジャッジ時間 7,355 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 4
other WA * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import Counter, defaultdict, deque
import sys
input=sys.stdin.readline
N=int(input())
n=N
#n,m=map(int,input().split())
class SCC:
    def __init__(self,n):
        self.n = n
        self.edges = []
 
    def add_edge(self,a,b):
        self.edges.append((a,b))
 
    def scc(self):
        n = self.n
        counter = [0]*(n+1)
        elist = [0]*len(self.edges)
        for x,y in self.edges:
            counter[x+1] += 1
        for x in range(n):
            counter[x+1] += counter[x]
        start = counter[:]
        for x,y in self.edges:
            elist[counter[x]] = y
            counter[x] += 1
 
        k = 0
        low = [0]*n
        num = [-1]*n
        pre = [None]*n
        visited = []
        q = []
        sccs = []
        for x in range(n):
            if num[x] < 0:
                q.append(x)
                q.append(x)
                while q:
                    x = q.pop()
                    if num[x] < 0:
                        low[x] = num[x] = k
                        k += 1
                        visited.append(x)
                        for i in range(start[x],start[x+1]):
                            y = elist[i]
                            if num[y] < 0:
                                q.append(y)
                                q.append(y)
                                pre[y] = x
                            else:
                                low[x] = min(low[x],num[y])
                    else:
                        if low[x] == num[x]:
                            scc = []
                            while 1:
                                y = visited.pop()
                                num[y] = n
                                scc.append(y)
                                if x == y:
                                    break
                            sccs.append(scc)
                        y = pre[x]
                        if y is not None:
                            low[y] = min(low[y],low[x])
        return sccs[::-1]
scc = SCC(n)
C=[]
for i in range(N):
  A=list(map(int, input().split()))
  B=A[1:]
  for b in B:
    C.append((i,b-1))
    scc.add_edge(i,b-1)
    
ans = scc.scc()

d=len(ans)
D=[-1]*N 
for i in range(len(ans)):
  B=ans[i]
  for b in B:
    D[b]=i
0