結果

問題 No.1382 Travel in Mitaru city
コンテスト
ユーザー marroncastle
提出日時 2021-02-07 23:04:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 399 ms / 2,000 ms
コード長 1,606 bytes
コンパイル時間 260 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 101,944 KB
最終ジャッジ日時 2024-07-04 17:41:57
合計ジャッジ時間 18,311 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 68
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
from heapq import *
from collections import defaultdict
class Graph:
  def __init__(self, N, M=-1):
    self.V = N
    if M>=0: self.E = M
    self.edge = [[] for _ in range(self.V)]
    self.edge_rev = [[] for _ in range(self.V)]
    self.order = []
    self.to = [0]*self.V
    self.visited = [False]*self.V
    self.dp = [0]*self.V

  def add_edges(self, ind=1, bi=False, rev=False):
    for i in range(self.E):
      a,b = map(int, input().split())
      a -= ind; b -= ind
      self.edge[a].append(b)
      if rev: self.edge_rev[b].append(a)
      if not bi: self.to[b] += 1
      if bi: self.edge[b].append(a)

  def add_edge(self, a, b, dist=-1, bi=False, rev=False):
    if dist>=0:
      self.edge[a].append((dist, b))
      if rev: self.edge_rev[b].append((dist, a))
      if bi: self.edge[b].append((dist, a))
    else:
      self.edge[a].append(b)
      if rev: self.edge_rev[b].append(a)
      if bi: self.edge[b].append(a)
    if not bi: self.to[b] += 1

  def bfs(self, start):
    self.visited[start] = True
    d = [(-P[start], start)]
    heapify(d)
    dic = defaultdict(lambda: False)
    ans = 0
    while len(d)>0:
      p,v = heappop(d)
      for w in self.edge[v]:
        if not self.visited[w]:
          if P[w]<-p:
            if dic[P[w]]==False:
              ans += 1
              dic[P[w]] = True
          else: P[w] = -p
          self.visited[w] = True
          heappush(d, (-P[w],w))
    return ans

N, M, S, T = map(int, input().split())
P = list(map(int, input().split()))
G = Graph(N,M)
G.add_edges(bi=True)
print(G.bfs(S-1))
0