a_lc_检查边长度限制的路径是否存在(并查集离线处理)

给一个数组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:刚刚还在想怎么在线算法切掉这题,所以一直拖着没发(这题也是需要换个思维去做的)

posted @ 2020-12-20 16:20  童年の波鞋  阅读(84)  评论(0编辑  收藏  举报