给一个数组queries,queries[j]=[p, q, limit],你的任务是对于每个查询,判断是否存在从 p 到 q 的路径,且这条路径上的每一条边都 严格小于 limit。
2 <= n <= 105
1 <= edgeList.length, queries.length <= 105
思路:问的是两个结点带限制下的连通性;对于每个查询qs[i],可以把边集es中边长小于qs[i].limit的边全都加到并查集中,如果给定的边集一直满足就一直加,然再判端qs[i]的两个点的连通性即可
class UF:
def __init__(self, n) -> None:
self.fa=[i for i in range(n)]
def find(self, u):
if u==self.fa[u]: return u
self.fa[u]=self.find(self.fa[u])
return self.fa[u]
def union(self, u, v):
fu,fv=self.find(u),self.find(v)
self.fa[fu]=fv
class Solution:
def distanceLimitedPathsExist(self, n: int, es: List[List[int]], qs: List[List[int]]) -> List[bool]:
m,lenQ=len(es),len(qs)
qs=[[u,v,w,idx] for idx,[u,v,w] in enumerate(qs)]
uf=UF(n)
qs.sort(key=lambda q:q[2])
es.sort(key=lambda e:e[2])
j,ans=0,[0]*lenQ
for i in range(lenQ):
while j<m and es[j][2]<qs[i][2]:
uf.union(es[j][0], es[j][1])
j+=1
ans[qs[i][3]]=uf.find(qs[i][0])==uf.find(qs[i][1])
return ans
ps:刚刚还在想怎么在线算法切掉这题,所以一直拖着没发(这题也是需要换个思维去做的)