Problem p02762 (python)

Fixed submission:
class UnionFind():
    def __init__(self, n):
        self.n = n
        self.parents = [-1] * n

    def find(self, x):
        if self.parents[x] < 0:
            return x
        else:
            self.parents[x] = self.find(self.parents[x])
            return self.parents[x]

    def union(self, x, y):
        x = self.find(x)
        y = self.find(y)

        if x == y:
            return

        if self.parents[x] > self.parents[y]:
            x, y = y, x

        self.parents[x] += self.parents[y]
        self.parents[y] = x

    def size(self, x):
        return -self.parents[self.find(x)]

    def same(self, x, y):
        return self.find(x) == self.find(y)

    def members(self, x):
        root = self.find(x)
        return [i for i in range(self.n) if self.find(i) == 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 all_group_members(self):
        return {r: self.members(r) for r in self.roots()}

    def __str__(self):
        return '\n'.join('{}: {}'.format(r, self.members(r)) for r in self.roots())


n,m,k=map(int,input().split())
uf=UnionFind(n)
block={}
for i in range(n):
    block[i]=[i]
for i in range(m):
    a,b=map(int,input().split())
    uf.union(a-1,b-1)
    block[a-1].append(b-1)
    block[b-1].append(a-1)
    
for i in range(k):
    a,b=map(int,input().split())
    block[a-1].append(b-1)
    block[b-1].append(a-1)
ans=[]
#print(block)
for i in range(n):
    res=uf.size(i)
 #   print(res)
    for value in block[i]:
        if uf.same(value,i):
            res-=1
    ans.append(res)
print(*ans)

Buggy submission (verdict: Runtime Error):
import queue
import numpy
def main():
    n,m,k=map(int,input().split())
    root=numpy.array([[0]*n for i in range(n)])
    for i in range(n):
        root[i][i]=2
    for i in range(m):
        a,b=map(int,input().split())
        root[a-1][b-1]=1
        root[b-1][a-1]=1
    for i in range(k):
        a,b=map(int,input().split())
        root[a-1][b-1]=-100
        root[b-1][a-1]=-100
    toVis=queue.Queue()
    Visted=[]
    checkOK=numpy.zeros(n)#その点に関しての友達候補つくり済みかどうか
    OK={}#友達候補リスト
    checkVis=numpy.zeros(n)#訪れたか確認するためのリスト
    checkVinit=numpy.zeros(n)#初期化のためのりすと
    for i in range(n):
        if checkOK[i]==5:
            continue
        checkVis=checkVinit
        Visted=[i]
        checkVis[i]=5
        toVis=queue.Queue()
        
        for j in range(len(root[i])):
            nex=root[i][j]
            if checkVis[j]!=5 and nex==1:
                toVis.put(j)

        while toVis.empty()==False:
        
            now=toVis.get()

            if checkVis[now]==5:
                continue

            Visted.append(now)
            checkVis[now]=5
            for j in range(n):#len(rooot[now]
                nex=root[now][j]
                if nex==1 and checkVis[j]!=5:
                    toVis.put(j)
    
        for j in Visted:
            OK[j]=Visted
            checkOK[j]=5
    ans=[]
    for i in range(n):
        num=0
      """  for value in OK[i]:
            if root[i][value]==0:
                num+=1"""
        ans.append(num)
    print(" ".join(map(str,ans)))
if __name__=="__main__":
    main()

Buggy submission (verdict: Runtime Error):
import queue
import numpy
def main():
    n,m,k=map(int,input().split())
    root=numpy.array([[0]*n for i in range(n)])
    for i in range(n):
        root[i][i]=2
    for i in range(m):
        a,b=map(int,input().split())
        root[a-1][b-1]=1
        root[b-1][a-1]=1
    for i in range(k):
        a,b=map(int,input().split())
        root[a-1][b-1]=-100
        root[b-1][a-1]=-100
    toVis=queue.Queue()
    Visted=[]
    checkOK=numpy.zeros(n)#その点に関しての友達候補つくり済みかどうか
    OK={}#友達候補リスト
    checkVis=numpy.zeros(n)#訪れたか確認するためのリスト
    checkVinit=numpy.zeros(n)#初期化のためのりすと
    for i in range(n):
        if checkOK[i]==5:
            continue
        checkVis=checkVinit
        Visted=[i]
        checkVis[i]=5
        toVis=queue.Queue()
        
        for j in range(len(root[i])):
            nex=root[i][j]
            if checkVis[j]!=5 and nex==1:
                toVis.put(j)

        while toVis.empty()==False:
        
            now=toVis.get()

            if checkVis[now]==5:
                continue

            Visted.append(now)
            checkVis[now]=5
            for j in range(n):#len(rooot[now]
                nex=root[now][j]
                if nex==1 and checkVis[j]!=5:
                    toVis.put(j)
    
        for j in Visted:
            OK[j]=Visted
            checkOK[j]=5
    ans=[]
    for i in range(n):
        num=0
      """  for value in OK[i]:
            if root[i][value]==0:
                num+=1
        ans.append(num)"""
    print(" ".join(map(str,ans)))
if __name__=="__main__":
    main()

Buggy submission (verdict: Runtime Error):
import queue

import numpy
def main():
    n,m,k=map(int,input().split())
    root=numpy.array([numpy.zeros(n) for i in range(n)])
    for i in range(n):
        root[i][i]=2
    for i in range(m):
        a,b=map(int,input().split())
        root[a-1][b-1]=1
        root[b-1][a-1]=1
    for i in range(k):
        a,b=map(int,input().split())
        root[a-1][b-1]=-100
        root[b-1][a-1]=-100
    toVis=queue.Queue()
    Visted=[]
    checkOK=numpy.zeros(n)#その点に関しての友達候補つくり済みかどうか
    OK={}#友達候補リスト
    checkVis=numpy.zeros(n)#訪れたか確認するためのリスト
    checkVinit=numpy.zeros(n)#初期化のためのりすと
    for i in range(n):
        if checkOK[i]==5:#既にもうお友達候補探索済みならしない
            continue
       # print(i)
        checkVis=checkVinit#訪れたかリストの初期化
        Visted=[i]#iは現在地だ
        checkVis[i]=5#当然訪れている
        toVis=queue.Queue()#訪れるべき点のキュー
        
        for j in range(len(root[i])):#iの列を見て、
            nex=root[i][j]          #i->jの値
            if checkVis[j]!=5 and nex==1:#5は既に訪れている 1は不通
                toVis.put(j)#でなけれな訪れるべき場所なので追加

        while toVis.empty()==False:
        
            now=toVis.get()#キューからとって訪れる

            if checkVis[now]==5:##そこが訪れ済みならスルー（ほんらいこれいらないと思う）
                continue

            Visted.append(now)#訪れたリストに追加
            checkVis[now]=5#チェック用にも追加
            for j in range(n):#len(root[now]
                nex=root[now][j]
                if nex==1 and checkVis[j]!=5:#次に訪れることのできる点をキューに入れる
                    toVis.put(j)
    
        for j in Visted:#訪れたリストにある点は互いに行き来ができるはず
            OK[j]=Visted#各点の訪れられる点リストはvistedで共通
            checkOK[j]=5#ｊはもうやらなくていいのでチェック
    
    ans=[]
    for i in range(n):
        num=0
        for value in OK[i]:
            if root[i][value]==0:#各行ける点からお友達じゃないかつブロックじゃないヒトをピックアップ
                num+=1
        ans.append(num)
    print(" ".join(map(str,ans)))
if __name__=="__main__":
    main()

Buggy submission (verdict: Runtime Error):
import queue
import sys
import numpy
input = sys.stdin.readline

def main():
    n,m,k=map(int,input().split())
    root=numpy.array([numpy.zeros(n) for i in range(n)])
    for i in range(n):
        root[i][i]=2
    for i in range(m):
        a,b=map(int,input().split())
        root[a-1][b-1]=1
        root[b-1][a-1]=1
    for i in range(k):
        a,b=map(int,input().split())
        root[a-1][b-1]=-100
        root[b-1][a-1]=-100
    toVis=queue.Queue()
    Visted=[]
    checkOK=numpy.zeros(n)#その点に関しての友達候補つくり済みかどうか
    OK={}#友達候補リスト
    checkVis=numpy.zeros(n)#訪れたか確認するためのリスト
    checkVinit=numpy.zeros(n)#初期化のためのりすと
    for i in range(n):
        if checkOK[i]==5:#既にもうお友達候補探索済みならしない
            continue
       # print(i)
        checkVis=checkVinit#訪れたかリストの初期化
        Visted=[i]#iは現在地だ
        checkVis[i]=5#当然訪れている
        toVis=queue.Queue()#訪れるべき点のキュー
        
        for j in range(len(root[i])):#iの列を見て、
            nex=root[i][j]          #i->jの値
            if checkVis[j]!=5 and nex==1:#5は既に訪れている 1は不通
                toVis.put(j)#でなけれな訪れるべき場所なので追加

        while toVis.empty()==False:
        
            now=toVis.get()#キューからとって訪れる

            if checkVis[now]==5:##そこが訪れ済みならスルー（ほんらいこれいらないと思う）
                continue

            Visted.append(now)#訪れたリストに追加
            checkVis[now]=5#チェック用にも追加
            for j in range(n):#len(root[now]
                nex=root[now][j]
                if nex==1 and checkVis[j]!=5:#次に訪れることのできる点をキューに入れる
                    toVis.put(j)
    
        for j in Visted:#訪れたリストにある点は互いに行き来ができるはず
            OK[j]=Visted#各点の訪れられる点リストはvistedで共通
            checkOK[j]=5#ｊはもうやらなくていいのでチェック
    
    ans=[]
    for i in range(n):
        num=0
        for value in OK[i]:
            if root[i][value]==0:#各行ける点からお友達じゃないかつブロックじゃないヒトをピックアップ
                num+=1
        ans.append(num)
    print(" ".join(map(str,ans)))
if __name__=="__main__":
    main()

Buggy submission (verdict: Runtime Error):
import queue
import sys
import numpy
input = sys.stdin.readline

def main():
    n,m,k=map(int,input().split())
    root=numpy.array([numpy.zeros(n) for i in range(n)])
    for i in range(n):
        root[i][i]=2
    for i in range(m):
        a,b=map(int,input().split())
        root[a-1][b-1]=1
        root[b-1][a-1]=1
    for i in range(k):
        a,b=map(int,input().split())
        root[a-1][b-1]=-100
        root[b-1][a-1]=-100
    toVis=queue.Queue()
    Visted=[]
    checkOK=numpy.zeros(n)#その点に関しての友達候補つくり済みかどうか
    OK={}#友達候補リスト
    checkVis=numpy.zeros(n)#訪れたか確認するためのリスト
    checkVinit=numpy.zeros(n)#初期化のためのりすと
    for i in range(n):
        if checkOK[i]==5:#既にもうお友達候補探索済みならしない
            continue
        print(i)
        checkVis=checkVinit#訪れたかリストの初期化
        Visted=[i]#iは現在地だ
        checkVis[i]=5#当然訪れている
        toVis=queue.Queue()#訪れるべき点のキュー
        
        for j in range(len(root[i])):#iの列を見て、
            nex=root[i][j]          #i->jの値
            if checkVis[j]!=5 and nex==1:#5は既に訪れている 1は不通
                toVis.put(j)#でなけれな訪れるべき場所なので追加

        while toVis.empty()==False:
        
            now=toVis.get()#キューからとって訪れる

            if checkVis[now]==5:##そこが訪れ済みならスルー（ほんらいこれいらないと思う）
                continue

            Visted.append(now)#訪れたリストに追加
            checkVis[now]=5#チェック用にも追加
            for j in range(n):#len(root[now]
                nex=root[now][j]
                if nex==1 and checkVis[j]!=5:#次に訪れることのできる点をキューに入れる
                    toVis.put(j)
    
        for j in Visted:#訪れたリストにある点は互いに行き来ができるはず
            OK[j]=Visted#各点の訪れられる点リストはvistedで共通
            checkOK[j]=5#ｊはもうやらなくていいのでチェック
    
    ans=[]
    for i in range(n):
        num=0
        for value in OK[i]:
            if root[i][value]==0:#各行ける点からお友達じゃないかつブロックじゃないヒトをピックアップ
                num+=1
        ans.append(num)
    print(" ".join(map(str,ans)))
if __name__=="__main__":
    main()

Buggy submission (verdict: Runtime Error):
import queue
import numpy
def main():
    n,m,k=map(int,input().split())
    root=numpy.array([numpy.zeros(n) for i in range(n)])
    for i in range(n):
        root[i][i]=2
    for i in range(m):
        a,b=map(int,input().split())
        root[a-1][b-1]=1
        root[b-1][a-1]=1
    for i in range(k):
        a,b=map(int,input().split())
        root[a-1][b-1]=-100
        root[b-1][a-1]=-100
    toVis=queue.Queue()
    Visted=[]
    checkOK=numpy.zeros(n)#その点に関しての友達候補つくり済みかどうか
    OK={}#友達候補リスト
    checkVis=numpy.zeros(n)#訪れたか確認するためのリスト
    checkVinit=numpy.zeros(n)#初期化のためのりすと
    for i in range(n):
        if checkOK[i]==5:#既にもうお友達候補探索済みならしない
            continue
        print(i)
        checkVis=checkVinit#訪れたかリストの初期化
        Visted=[i]#iは現在地だ
        checkVis[i]=5#当然訪れている
        toVis=queue.Queue()#訪れるべき点のキュー
        
        for j in range(len(root[i])):#iの列を見て、
            nex=root[i][j]          #i->jの値
            if checkVis[j]!=5 and nex==1:#5は既に訪れている 1は不通
                toVis.put(j)#でなけれな訪れるべき場所なので追加

        while toVis.empty()==False:
        
            now=toVis.get()#キューからとって訪れる

            if checkVis[now]==5:##そこが訪れ済みならスルー（ほんらいこれいらないと思う）
                continue

            Visted.append(now)#訪れたリストに追加
            checkVis[now]=5#チェック用にも追加
            for j in range(n):#len(root[now]
                nex=root[now][j]
                if nex==1 and checkVis[j]!=5:#次に訪れることのできる点をキューに入れる
                    toVis.put(j)
    
        for j in Visted:#訪れたリストにある点は互いに行き来ができるはず
            OK[j]=Visted#各点の訪れられる点リストはvistedで共通
            checkOK[j]=5#ｊはもうやらなくていいのでチェック
    print(1," ",1," ",1)
    """
    ans=[]
    for i in range(n):
        num=0
        for value in OK[i]:
            if root[i][value]==0:#各行ける点からお友達じゃないかつブロックじゃないヒトをピックアップ
                num+=1
        ans.append(num)
    print(" ".join(map(str,ans)))"""
if __name__=="__main__":
    main()

Buggy submission (verdict: Runtime Error):
import queue
import numpy
def main():
    n,m,k=map(int,input().split())
    root=numpy.array([numpy.zeros(n) for i in range(n)])
    for i in range(n):
        root[i][i]=2
    for i in range(m):
        a,b=map(int,input().split())
        root[a-1][b-1]=1
        root[b-1][a-1]=1
    for i in range(k):
        a,b=map(int,input().split())
        root[a-1][b-1]=-100
        root[b-1][a-1]=-100
    toVis=queue.Queue()
    Visted=[]
    checkOK=numpy.zeros(n)#その点に関しての友達候補つくり済みかどうか
    OK={}#友達候補リスト
    checkVis=numpy.zeros(n)#訪れたか確認するためのリスト
    checkVinit=numpy.zeros(n)#初期化のためのりすと
    for i in range(n):
        if checkOK[i]==5:#既にもうお友達候補探索済みならしない
            continue
        print(i)
        checkVis=checkVinit#訪れたかリストの初期化
        Visted=[i]#iは現在地だ
        checkVis[i]=5#当然訪れている
        toVis=queue.Queue()#訪れるべき点のキュー
        
        for j in range(len(root[i])):#iの列を見て、
            nex=root[i][j]          #i->jの値
            if checkVis[j]!=5 and nex==1:#5は既に訪れている 1は不通
                toVis.put(j)#でなけれな訪れるべき場所なので追加

        while toVis.empty()==False:
        
            now=toVis.get()#キューからとって訪れる

            if checkVis[now]==5:##そこが訪れ済みならスルー（ほんらいこれいらないと思う）
                continue

            Visted.append(now)#訪れたリストに追加
            checkVis[now]=5#チェック用にも追加
            for j in range(n):#len(root[now]
                nex=root[now][j]
                if nex==1 and checkVis[j]!=5:#次に訪れることのできる点をキューに入れる
                    toVis.put(j)
    
        for j in Visted:#訪れたリストにある点は互いに行き来ができるはず
            OK[j]=Visted#各点の訪れられる点リストはvistedで共通
            checkOK[j]=5#ｊはもうやらなくていいのでチェック
    print(1 1 1 1 1)
    """
    ans=[]
    for i in range(n):
        num=0
        for value in OK[i]:
            if root[i][value]==0:#各行ける点からお友達じゃないかつブロックじゃないヒトをピックアップ
                num+=1
        ans.append(num)
    print(" ".join(map(str,ans)))"""
if __name__=="__main__":
    main()