結果

問題 No.1254 補強への架け橋
ユーザー marroncastle
提出日時 2020-10-10 02:03:51
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,872 bytes
コンパイル時間 237 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 296,964 KB
最終ジャッジ日時 2024-07-20 15:27:59
合計ジャッジ時間 28,098 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 121 TLE * 1 -- * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys,os,io
input = sys.stdin.readline
from collections import defaultdict
class Graph:
  def __init__(self, N):
    self.V = N
    self.edge = [[] for _ in range(self.V)]
    self.edges = defaultdict(lambda: 0)
    self.visited = [0]*N
    self.parent = [N]*N
  
  def add_edge(self, a, b, i):
    self.edge[a].append(b)
    self.edges[(a,b)] = i

  def dfs(self, start):
    stack = [start]
    self.parent[start] = -1
    self.visited[start] = 1
    #記録したい値の配列を定義
    while stack:
      v = stack[-1]
      for u,i in self.edge[v]:
        if u==self.parent[v]:
          continue
        if self.parent[u]==N: #子へ降ろす
          self.parent[u]=v
          stack.append(u)
          break
        else:
          return u
      else:
        stack.pop()

  def bfs1(self, start):
    stack = deque([start])
    self.parent[start] = -1
    while stack:
      v = stack.pop()
      for u in self.edge[v]:
        if u==self.parent[v]:
          continue
        if self.parent[u]!=N:
          return u
        self.parent[u]=v
        #頂点に対する処理
        stack.append(u)
    return
  
  
  def bfs(self, start):
    d = deque()
    d.append([start])
    cyc = []
    self.visited[start] = True
    while len(d)>0:
      vs = d.popleft()
      for w in self.edge[vs[-1]]:
        if len(vs)>1 and w==vs[-2]:
          continue
        cyc = copy(vs)
        cyc.append(w)
        if w==start:
          return cyc
        d.append(cyc)
    return cyc

from collections import deque
from copy import copy

N = int(input())
G = Graph(N)
for i in range(N):
  a,b = map(int, input().split())
  a,b = min(a-1,b-1), max(a-1,b-1)
  G.add_edge(a,b,i+1)
  G.add_edge(b,a,i+1)

ans = set()
cyc = G.bfs1(0)
vs = G.bfs(cyc)
for v0,v1 in zip(vs[:-1],vs[1:]):
  ans.add(G.edges[(v0,v1)])
ans = list(ans)
print(len(ans))
print(*ans)


0