結果

問題 No.1254 補強への架け橋
ユーザー uni_pythonuni_python
提出日時 2020-10-10 00:04:24
言語 PyPy3
(7.3.2)
結果
AC  
実行時間 860 ms / 2,000 ms
コード長 5,039 Byte
コンパイル時間 204 ms
使用メモリ 154,072 KB
最終ジャッジ日時 2020-10-10 00:05:16
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 134 ms
82,008 KB
testcase_01 AC 132 ms
82,204 KB
testcase_02 AC 132 ms
82,124 KB
testcase_03 AC 133 ms
82,016 KB
testcase_04 AC 141 ms
82,116 KB
testcase_05 AC 135 ms
82,216 KB
testcase_06 AC 135 ms
82,200 KB
testcase_07 AC 136 ms
82,136 KB
testcase_08 AC 136 ms
82,292 KB
testcase_09 AC 134 ms
81,884 KB
testcase_10 AC 129 ms
82,136 KB
testcase_11 AC 131 ms
81,896 KB
testcase_12 AC 131 ms
82,252 KB
testcase_13 AC 149 ms
82,240 KB
testcase_14 AC 139 ms
81,972 KB
testcase_15 AC 133 ms
82,112 KB
testcase_16 AC 134 ms
82,060 KB
testcase_17 AC 140 ms
82,248 KB
testcase_18 AC 140 ms
82,140 KB
testcase_19 AC 141 ms
82,112 KB
testcase_20 AC 144 ms
82,080 KB
testcase_21 AC 139 ms
82,168 KB
testcase_22 AC 156 ms
81,992 KB
testcase_23 AC 133 ms
82,144 KB
testcase_24 AC 132 ms
82,008 KB
testcase_25 AC 131 ms
82,236 KB
testcase_26 AC 132 ms
82,476 KB
testcase_27 AC 130 ms
82,092 KB
testcase_28 AC 132 ms
82,344 KB
testcase_29 AC 131 ms
82,092 KB
testcase_30 AC 129 ms
82,208 KB
testcase_31 AC 131 ms
82,248 KB
testcase_32 AC 142 ms
81,992 KB
testcase_33 AC 134 ms
82,136 KB
testcase_34 AC 132 ms
82,408 KB
testcase_35 AC 133 ms
82,396 KB
testcase_36 AC 132 ms
82,072 KB
testcase_37 AC 132 ms
81,952 KB
testcase_38 AC 133 ms
82,136 KB
testcase_39 AC 131 ms
82,044 KB
testcase_40 AC 130 ms
81,848 KB
testcase_41 AC 142 ms
81,988 KB
testcase_42 AC 141 ms
82,168 KB
testcase_43 AC 179 ms
84,092 KB
testcase_44 AC 142 ms
82,104 KB
testcase_45 AC 141 ms
82,244 KB
testcase_46 AC 182 ms
84,152 KB
testcase_47 AC 151 ms
83,036 KB
testcase_48 AC 168 ms
83,956 KB
testcase_49 AC 133 ms
82,280 KB
testcase_50 AC 200 ms
85,220 KB
testcase_51 AC 184 ms
85,256 KB
testcase_52 AC 193 ms
85,604 KB
testcase_53 AC 183 ms
84,136 KB
testcase_54 AC 185 ms
85,304 KB
testcase_55 AC 176 ms
84,692 KB
testcase_56 AC 134 ms
82,176 KB
testcase_57 AC 181 ms
84,632 KB
testcase_58 AC 186 ms
83,872 KB
testcase_59 AC 135 ms
82,404 KB
testcase_60 AC 140 ms
82,184 KB
testcase_61 AC 196 ms
84,272 KB
testcase_62 AC 137 ms
82,188 KB
testcase_63 AC 316 ms
90,216 KB
testcase_64 AC 258 ms
85,172 KB
testcase_65 AC 293 ms
88,308 KB
testcase_66 AC 286 ms
90,052 KB
testcase_67 AC 233 ms
85,196 KB
testcase_68 AC 281 ms
88,976 KB
testcase_69 AC 316 ms
89,016 KB
testcase_70 AC 286 ms
88,040 KB
testcase_71 AC 260 ms
85,204 KB
testcase_72 AC 312 ms
89,108 KB
testcase_73 AC 250 ms
86,564 KB
testcase_74 AC 306 ms
89,472 KB
testcase_75 AC 291 ms
88,116 KB
testcase_76 AC 213 ms
85,520 KB
testcase_77 AC 277 ms
89,436 KB
testcase_78 AC 314 ms
90,824 KB
testcase_79 AC 347 ms
89,632 KB
testcase_80 AC 332 ms
89,372 KB
testcase_81 AC 313 ms
89,420 KB
testcase_82 AC 308 ms
90,868 KB
testcase_83 AC 860 ms
142,068 KB
testcase_84 AC 811 ms
138,724 KB
testcase_85 AC 633 ms
118,848 KB
testcase_86 AC 745 ms
131,748 KB
testcase_87 AC 804 ms
135,852 KB
testcase_88 AC 348 ms
93,284 KB
testcase_89 AC 809 ms
142,240 KB
testcase_90 AC 652 ms
120,112 KB
testcase_91 AC 540 ms
113,460 KB
testcase_92 AC 415 ms
100,060 KB
testcase_93 AC 735 ms
129,252 KB
testcase_94 AC 656 ms
123,316 KB
testcase_95 AC 697 ms
123,380 KB
testcase_96 AC 804 ms
137,824 KB
testcase_97 AC 523 ms
109,388 KB
testcase_98 AC 794 ms
138,768 KB
testcase_99 AC 589 ms
117,284 KB
testcase_100 AC 847 ms
143,104 KB
testcase_101 AC 388 ms
96,876 KB
testcase_102 AC 335 ms
94,088 KB
testcase_103 AC 419 ms
99,508 KB
testcase_104 AC 466 ms
105,112 KB
testcase_105 AC 711 ms
130,092 KB
testcase_106 AC 524 ms
112,600 KB
testcase_107 AC 831 ms
140,676 KB
testcase_108 AC 850 ms
139,112 KB
testcase_109 AC 711 ms
128,584 KB
testcase_110 AC 676 ms
124,384 KB
testcase_111 AC 675 ms
126,076 KB
testcase_112 AC 437 ms
105,340 KB
testcase_113 AC 670 ms
124,568 KB
testcase_114 AC 503 ms
110,456 KB
testcase_115 AC 339 ms
93,796 KB
testcase_116 AC 578 ms
113,600 KB
testcase_117 AC 458 ms
104,832 KB
testcase_118 AC 818 ms
138,328 KB
testcase_119 AC 598 ms
116,052 KB
testcase_120 AC 802 ms
133,460 KB
testcase_121 AC 438 ms
101,476 KB
testcase_122 AC 553 ms
113,884 KB
testcase_123 AC 133 ms
82,016 KB
testcase_124 AC 831 ms
153,736 KB
testcase_125 AC 808 ms
154,072 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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()
0