結果

問題 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 123
権限があれば一括ダウンロードができます

ソースコード

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