关闭

Gym 100685G Gadget Hackwrench(LCA)

标签: GymLCA
268人阅读 评论(0) 收藏 举报
分类:
题目地址:http://codeforces.com/gym/100685/problem/G

思路:不考虑方向,则该图构成一棵树。对于每条有向边:对于正向边,权值为1;对于反向边,权值为-1。则对于一组查询(x,y),若有向图中Lca(x,y)到x距离为负且等于树中Lca(x,y)到x的距离,有向图中Lca(x,y)到y距离为正且等于树中Lca(x,y)到y的距离,则x,y连通。

#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#define debu
using namespace std;
const int maxn=1e5+50;
struct Node
{
    int p,id;
    Node(int p=0,int id=0):p(p),id(id) {}
};
int n;
queue<int> q;
int vis[maxn];
int dist[maxn];
int anc[maxn][20];
int v[maxn],d[maxn];
vector<Node> g[maxn];
void lca_dfs(int x,int dep)
{
    v[x]=1,d[x]=dep;
    for(int i=0; i<g[x].size(); i++)
    {
        int nt=g[x][i].p;
        if(!v[nt])
        {
            dist[nt]=dist[x]+g[x][i].id;
            anc[nt][0]=x;
            int k=0;
            while(anc[anc[nt][k]][k]!=0)
            {
                anc[nt][k+1]=anc[anc[nt][k]][k];
                k++;
            }
            lca_dfs(nt,dep+1);
        }
    }
}
int Lca(int x,int y)
{
    if(d[x]<d[y]) swap(x,y);
    int l=d[x]-d[y];
    int k=0;
    while(l!=0)
    {
        if(l&1) x=anc[x][k];
        l>>=1;
        k++;
    }
    k=0;
    while(x!=y)
    {
        if((anc[x][k]!=anc[y][k])||(!k))
        {
            x=anc[x][k];
            y=anc[y][k];
            k++;
        }
        else k--;
    }
    return x;
}
void init()
{
    memset(d,0,sizeof(d));
    memset(vis,0,sizeof(vis));
    memset(anc,0,sizeof(anc));
    memset(dist,0,sizeof(dist));
    for(int i=1; i<=n; i++) g[i].clear();
}
void build()
{
    memset(v,0,sizeof(v));
    for(int i=1;i<=n;i++)
        if(!v[i]) lca_dfs(i,0);
}
void query()
{
    int q;
    scanf("%d",&q);
    while(q--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        int lca=Lca(x,y);
        int tmpx=d[x]-d[lca];
        int tmpy=d[y]-d[lca];
        int distx=dist[x]-dist[lca];
        int disty=dist[y]-dist[lca];
        if(tmpx==-distx&&tmpy==disty) printf("Yes\n");
        else printf("No\n");
    }
}
int main()
{
#ifdef debug
    freopen("in.in","r",stdin);
#endif // debug
    scanf("%d",&n);
    init();
    for(int i=1; i<=n-1; i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(Node(y,1));
        g[y].push_back(Node(x,-1));
        vis[y]=1;
    }
    build();
    query();
    return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:100639次
    • 积分:4612
    • 等级:
    • 排名:第6707名
    • 原创:378篇
    • 转载:10篇
    • 译文:0篇
    • 评论:2条
    博客专栏
    ACM

    文章:232篇

    阅读:72435