ZOJ4097 Rescue the Princess(并查集+tarjan双连通分量缩点+LCA倍增算法)

题意:

给出一个无向图,每个询问给出三个点u,v,w,询问能否从v和w找到通往u的路径,且两条路径没有重合的地方。

题解:

图不一定连通,当v,w有任何一个跟u不在一个连通块上,那就直接输出No(用并查集判断)

然后用tarjan算法缩点,每个双连通分量为一个点,建立一个新的图,由于图不一定连通,可以看作是森林。

对森林里的每棵树都做一遍LCA预处理,对于之后每个询问分类讨论,有以下三种合法的情况:

ZOJ4097 Rescue the Princess(并查集+tarjan双连通分量缩点+LCA倍增算法)

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+14;
int N,M,Q;
int father[30][maxn];
int h[maxn];
vector<int> g[maxn];//原图
vector<int> sc[maxn];//tarjan以后的缩点图 
//倍增求LCA 
int visit[maxn];
void dfs (int x,int u) {
    visit[x]=1;
    h[x]=h[u]+1;
    father[0][x]=u;
    for (int i=1;(1<<i)<=h[x];i++) 
        father[i][x]=father[i-1][father[i-1][x]];
    for (int i=0;i<sc[x].size();i++) {
        int v=sc[x][i];
        if (v==father[0][x]) continue;
        dfs(v,x);
    }
}
int lca (int x,int y) {
    if (h[x]>h[y]) swap(x,y);
    for (int i=20;i>=0;i--) 
        if (h[x]<=h[y]-(1<<i)) y=father[i][y];
    if (x==y) return x;
    for (int i=20;i>=0;i--) {
        if (father[i][x]!=father[i][y]) {
            x=father[i][x];
            y=father[i][y];
        }
    }
    return father[0][x];
}
//并查集判断连通块 
int f[maxn];
int findfather (int x) {
    int a=x;
    while (x!=f[x]) x=f[x];
    while (a!=f[a]) {
        int z=a;
        a=f[a];
        f[z]=x;
    }
    return x;
}
void Union (int a,int b) {
    int faA=findfather(a);
    int faB=findfather(b);
    if (faA!=faB) f[faA]=faB;
}
//tarjan双连通分量缩点
int low[maxn];
int dfn[maxn];
int pos[maxn];
int scc=0;
int cnt;
stack<int> st;
void tarjan (int x,int pre) {
    low[x]=dfn[x]=++cnt;
    st.push(x);
    int son=0;
    for (int i=0;i<g[x].size();i++) {
        if (g[x][i]==pre&&++son<2) continue;
        if (!low[g[x][i]]) {
            tarjan(g[x][i],x);
            low[x]=min(low[x],low[g[x][i]]);
        }
        else if (!pos[g[x][i]]) 
            low[x]=min(low[x],dfn[g[x][i]]);
    }
    if (low[x]==dfn[x]) {
        scc++;
        while (1) {
            int u=st.top();
            st.pop();
            low[u]=low[x];
            pos[u]=scc;
            if (u==x) break;
        }
    }
} 

//建立缩点图
void build_sc () {
    for (int i=1;i<=N;i++) {
        for (int j=0;j<g[i].size();j++) 
            if (pos[i]!=pos[g[i][j]])
                sc[pos[i]].push_back(pos[g[i][j]]);
    }
} 
int main () {
    int T;
    scanf("%d",&T);
    while (T--) {
        scanf("%d%d%d",&N,&M,&Q);
        for (int i=1;i<=N;i++)
            g[i].clear(),f[i]=i,sc[i].clear();
        memset(father,0,sizeof(father));
        memset(h,0,sizeof(h));
        memset(visit,0,sizeof(visit));
        scc=0;
        cnt=0;
        fill(low,low+maxn,0);
        fill(dfn,dfn+maxn,0);
        fill(pos,pos+maxn,0);
        while (!st.empty()) st.pop();
        for (int i=0;i<M;i++) {
            int x,y; 
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            g[y].push_back(x);
            Union(x,y);
        }
        for (int i=1;i<=N;i++) 
            if (!dfn[i]) tarjan(i,i);
        build_sc(); 
        for (int i=1;i<=scc;i++) 
             if (!visit[i]) dfs(i,i);
        for (int i=0;i<Q;i++) {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            int faU=findfather(u);
            int faV=findfather(v);
            int faW=findfather(w);
            u=pos[u];
            v=pos[v];
            w=pos[w];
            //printf("%d %d %d\n",u,v,w);
            if (faU!=faV||faU!=faW) {
                printf("No\n");
                continue;
            }
            int flag=1;
            int uv=lca(u,v);
            int uw=lca(u,w);
            int vw=lca(v,w);
            if (uv==u) {
                if (uw==u&&vw==u) flag=1;
                else if (uw!=u) flag=1;
                else flag=0;
            }
            else if (uv==v) {
                if (uw==u) flag=1;
                else flag=0;
            }
            else {
                if (uw!=u) flag=0;
            }
            if (flag) printf("Yes\n");
            else printf("No\n");
        }
    } 
    return 0;
}

 

上一篇:【ybt金牌导航5-3-1】【luogu P3233】世界树 / 虚树例题


下一篇:面试系列一:精选大数据面试真题10道(混合型)-附答案详细解析