#565335 (PyPy3) No.1254 補強への架け橋

提出ソース

結果

問題	No.1254 補強への架け橋
ユーザー	uni_python
提出日時	2020-10-10 00:04:24
言語	PyPy3 (7.3.15)
結果	AC
実行時間	715 ms / 2,000 ms
コード長	5,039 bytes
コンパイル時間	210 ms
コンパイル使用メモリ	82,048 KB
実行使用メモリ	151,608 KB
最終ジャッジ日時	2024-07-20 14:51:24
合計ジャッジ時間	40,740 ms
ジャッジサーバーID （参考情報）	judge2 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	80 ms 67,584 KB
testcase_01	AC	74 ms 67,456 KB
testcase_02	AC	74 ms 67,456 KB
testcase_03	AC	75 ms 67,584 KB
testcase_04	AC	73 ms 67,712 KB
testcase_05	AC	74 ms 67,456 KB
testcase_06	AC	74 ms 67,584 KB
testcase_07	AC	70 ms 67,584 KB
testcase_08	AC	74 ms 67,584 KB
testcase_09	AC	76 ms 67,328 KB
testcase_10	AC	75 ms 67,712 KB
testcase_11	AC	72 ms 67,712 KB
testcase_12	AC	76 ms 67,712 KB
testcase_13	AC	75 ms 68,224 KB
testcase_14	AC	75 ms 68,224 KB
testcase_15	AC	75 ms 67,712 KB
testcase_16	AC	73 ms 67,968 KB
testcase_17	AC	74 ms 67,584 KB
testcase_18	AC	75 ms 67,328 KB
testcase_19	AC	74 ms 67,840 KB
testcase_20	AC	76 ms 67,968 KB
testcase_21	AC	79 ms 67,840 KB
testcase_22	AC	73 ms 67,968 KB
testcase_23	AC	75 ms 68,096 KB
testcase_24	AC	76 ms 68,224 KB
testcase_25	AC	75 ms 67,968 KB
testcase_26	AC	77 ms 68,352 KB
testcase_27	AC	75 ms 68,096 KB
testcase_28	AC	75 ms 68,352 KB
testcase_29	AC	75 ms 67,968 KB
testcase_30	AC	74 ms 67,968 KB
testcase_31	AC	76 ms 68,352 KB
testcase_32	AC	74 ms 67,584 KB
testcase_33	AC	78 ms 68,736 KB
testcase_34	AC	75 ms 68,352 KB
testcase_35	AC	75 ms 68,480 KB
testcase_36	AC	77 ms 68,480 KB
testcase_37	AC	77 ms 68,352 KB
testcase_38	AC	76 ms 68,480 KB
testcase_39	AC	75 ms 68,096 KB
testcase_40	AC	75 ms 68,352 KB
testcase_41	AC	75 ms 68,096 KB
testcase_42	AC	74 ms 67,712 KB
testcase_43	AC	123 ms 78,976 KB
testcase_44	AC	87 ms 72,448 KB
testcase_45	AC	85 ms 72,064 KB
testcase_46	AC	124 ms 78,592 KB
testcase_47	AC	101 ms 77,440 KB
testcase_48	AC	121 ms 78,592 KB
testcase_49	AC	76 ms 68,480 KB
testcase_50	AC	139 ms 79,488 KB
testcase_51	AC	130 ms 79,104 KB
testcase_52	AC	134 ms 79,616 KB
testcase_53	AC	122 ms 79,104 KB
testcase_54	AC	131 ms 78,848 KB
testcase_55	AC	124 ms 78,976 KB
testcase_56	AC	81 ms 69,504 KB
testcase_57	AC	118 ms 78,976 KB
testcase_58	AC	119 ms 78,720 KB
testcase_59	AC	78 ms 68,864 KB
testcase_60	AC	85 ms 71,936 KB
testcase_61	AC	126 ms 78,848 KB
testcase_62	AC	80 ms 69,376 KB
testcase_63	AC	247 ms 86,016 KB
testcase_64	AC	191 ms 80,768 KB
testcase_65	AC	220 ms 83,200 KB
testcase_66	AC	220 ms 83,188 KB
testcase_67	AC	180 ms 80,640 KB
testcase_68	AC	226 ms 82,964 KB
testcase_69	AC	241 ms 84,244 KB
testcase_70	AC	192 ms 81,644 KB
testcase_71	AC	185 ms 80,620 KB
testcase_72	AC	232 ms 83,712 KB
testcase_73	AC	190 ms 81,592 KB
testcase_74	AC	235 ms 84,340 KB
testcase_75	AC	220 ms 83,072 KB
testcase_76	AC	151 ms 79,744 KB
testcase_77	AC	211 ms 82,460 KB
testcase_78	AC	246 ms 85,228 KB
testcase_79	AC	242 ms 85,376 KB
testcase_80	AC	236 ms 84,224 KB
testcase_81	AC	242 ms 84,864 KB
testcase_82	AC	238 ms 84,736 KB
testcase_83	AC	691 ms 135,876 KB
testcase_84	AC	697 ms 134,624 KB
testcase_85	AC	499 ms 114,180 KB
testcase_86	AC	620 ms 124,816 KB
testcase_87	AC	665 ms 131,984 KB
testcase_88	AC	272 ms 88,192 KB
testcase_89	AC	715 ms 136,088 KB
testcase_90	AC	537 ms 115,284 KB
testcase_91	AC	456 ms 108,668 KB
testcase_92	AC	326 ms 94,212 KB
testcase_93	AC	599 ms 123,044 KB
testcase_94	AC	547 ms 118,816 KB
testcase_95	AC	528 ms 118,740 KB
testcase_96	AC	651 ms 133,652 KB
testcase_97	AC	398 ms 103,548 KB
testcase_98	AC	662 ms 133,568 KB
testcase_99	AC	473 ms 111,784 KB
testcase_100	AC	694 ms 137,140 KB
testcase_101	AC	301 ms 92,736 KB
testcase_102	AC	243 ms 87,296 KB
testcase_103	AC	313 ms 93,808 KB
testcase_104	AC	360 ms 98,548 KB
testcase_105	AC	591 ms 122,412 KB
testcase_106	AC	407 ms 106,552 KB
testcase_107	AC	698 ms 135,972 KB
testcase_108	AC	684 ms 135,036 KB
testcase_109	AC	573 ms 122,216 KB
testcase_110	AC	546 ms 118,644 KB
testcase_111	AC	567 ms 120,788 KB
testcase_112	AC	349 ms 98,252 KB
testcase_113	AC	530 ms 117,692 KB
testcase_114	AC	406 ms 104,948 KB
testcase_115	AC	264 ms 88,404 KB
testcase_116	AC	462 ms 109,388 KB
testcase_117	AC	350 ms 98,208 KB
testcase_118	AC	662 ms 133,356 KB
testcase_119	AC	480 ms 112,484 KB
testcase_120	AC	661 ms 131,508 KB
testcase_121	AC	339 ms 95,936 KB
testcase_122	AC	443 ms 108,932 KB
testcase_123	AC	70 ms 67,456 KB
testcase_124	AC	659 ms 151,352 KB
testcase_125	AC	641 ms 151,608 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

import sys
input=sys.stdin.readline
def I(): return int(input())
def MI(): return map(int, input().split())
def LI(): return list(map(int, input().split()))

"""
木+1本で，閉路が一つあるのでその周状がダメ
"""
def main():

    ##########################################
    import heapq
    class Dijkstra():
        """
        ・有向 / 無向は問わない(無向の場合は，逆向きの辺もたす)
        ・負のコストがない場合のみ
        ・計算量はO(E log|V|)　
        ・heapを使うことで頂点を走査する必要がなくなる（代わりに，距離更新したものは確定でなくともqueに入れておく）
        """
        
        #最短のpathをたす
        
        
        class Edge():
            #重み付き有向辺
            
            def __init__(self, _to, _cost):
                self.to =_to
                self.cost = _cost
        
        def __init__(self, V):
            #引数Vは頂点数
            self.G = [[] for _ in range(V)] #隣接リストG[u][i]が頂点uのi番目の辺
            self. _E = 0 #辺の数
            self._V = V #頂点数
        
        #proparty - 辺の数
        def E(self):
            return self._E
        
        #proparty - 頂点数
        def V(self):
            return self._V
        
        def add(self, _from, _to, _cost):
            #2頂点と辺のコストを追加
            self.G[_from].append(self.Edge(_to,_cost))
            self._E +=1
            
        def add2(self, _from, _to, _cost):
            #2頂点と辺のコスト（無向）を追加
            self.G[_from].append(self.Edge(_to, _cost))
            self.G[_to].append(self.Edge(_from, _cost))
            self._E +=2
            
        def shortest_path(self,s,g):
            #始点sから頂点iまでの最短経路長のリストを返す
            
            que = [] #priority queue
            d = [float("inf")] * self.V()
            prev = [None]*self.V() #prev[j]は，sからjへ最短経路で行くときのjの一つ前の場所
            d[s] = 0
            heapq.heappush(que,(0,s)) #始点の距離と頂点番号をヒープに追加
            
            while len(que)!=0:
                #キューに格納されてある中で一番コストが小さい頂点を取り出す
                cost,v = heapq.heappop(que)
                
                #キューに格納された最短経路長候補がdの距離よりも大きい場合に処理をスキップ
                if d[v] < cost:
                    continue
                    
                #頂点vに隣接する各頂点iに対して，vを経由した場合の距離を計算して，これがd[i]よりも小さい場合に更新
                for i in range(len(self.G[v])):
                    e = self.G[v][i] #vのi個目の隣接辺
                    if d[e.to] > d[v] + e.cost:
                        d[e.to] = d[v] + e.cost #更新
                        prev[e.to] = v
                        heapq.heappush(que,(d[e.to],e.to)) #queに新たな最短経路長候補を追加
            
            #sからgまでの最短経路
            path = []
            pos = g #今いる場所，ゴールで初期化
            for _ in range(self.V()+1):
                path.append(pos)
                if pos == s:
                    break
                #print("pos:",format(pos))
                pos = prev[pos]
            path.reverse()
            #print(path)
                        
            return d,path
    ########################

    mod=10**9+7
    N=I()
    adj=[[]for _ in range(N)]
    A=[]
    B=[]
    from collections import defaultdict
    dd = defaultdict(int)
    for i in range(N):
        a,b=MI()
        a-=1
        b-=1
        A.append(a)
        B.append(b)
        adj[a].append(b)
        adj[b].append(a)
        
        if a>b:
            a,b=b,a
            
        dd[(a,b)]=i+1
        
    import queue
    q=queue.Queue()
    q.put((0,-1))
    used=[0]*N
    edge=[-1,-1]
    
    while not q.empty():
        v,p=q.get()
        for nv in adj[v]:
            if nv!=p:
                if used[nv]:
                    edge=[v,nv]
                else:
                    used[nv]=1
                    q.put((nv,v))
    
    #これを抜けば木になって，lca閉路出せば終了，ダイクストラでいいや
    st=edge[0]
    en=edge[1]
    
    if st>en:
        st,en=en,st
    
    adj[st].remove(en)
    adj[en].remove(st)
    
    djk=Dijkstra(N)
    for v in range(N):
        for nv in adj[v]:
            djk.add(v,nv,1)
            
    d,path=djk.shortest_path(st,en)
    
    ans=[dd[(st,en)]]
    
    for i in range(len(path)-1):
        now=path[i]
        nxt=path[i+1]
        if now>nxt:
            now,nxt=nxt,now
        ans.append(dd[(now,nxt)])
    
    ans.sort()
    print(len(ans))
    print(' '.join(map(str, ans)))
    # print(st+1,en+1)
    

    
    
    


main()

yukicoder

結果

テストケース

ソースコード