提出 #27932262

---

ソースコード 拡げる

$ln = $<.map{|ln| ln.split.map(&:to_i) }
N,M,K = $ln.shift
UV = $ln.shift M
S = 0,*$ln.shift
Q, = $ln.shift
AB = $ln.shift Q

ST2UV = Hash.new{|h,st| h[st] = [] }
G = [-1]*(N+1)
F = lambda{|g,u|
	g[u]<0 ? u : g[u] = F[g,g[u]]
}
U = lambda{|g,u,v|
	u,v = F[g,u],F[g,v]
	next if u==v
	u,v = v,u if g[v]<g[u]
	g[u] += g[v]
	g[v] = u
}
UV.each{|u,v|
	s,t = S[u],S[v]
	if s==t
		U[G,u,v]
	else
		ST2UV[s+t+[s,t].max*K]<<[u,v]
	end
}

Ans = [nil]*Q
Q.times.group_by{|q|
	AB[q].map{|a| S[a] }.then{|st| st.sum+st.max*K }
}.each{|st,qs|
	g2 = Hash.new(-1)
	ST2UV[st].each{|u,v|
		U[g2,F[G,u],F[G,v]]
	}
	qs.each{|q|
		a,b = AB[q].map{|a| F[g2,F[G,a]] }
		Ans[q] = a==b ? 1 : 0
	}
}

puts Ans

提出情報

ジャッジ結果