COGS 2098. Asm.Def的病毒

★☆   输入文件：asm_virus.in   输出文件：asm_virus.out   简单对比
时间限制：1 s   内存限制：256 MB

【题目描述】

“这就是我们最新研制的，世界上第一种可持久化动态计算机病毒，‘创世纪’。”方教授介绍道。

“哦。”主席面无表情地点点头。

“‘创世纪’无法真正杀死透明计算网络，但是可以把它变成傻子。可惜透明计算网络能轻松地辨认出病毒，所以我建议……”

“为什么不伪装呢？”Asm.Def说。

“当然不行，它比我们更懂伪装。”

“我是说，把我们的病毒伪装成杀毒软件。”

方教授震惊地盯着Asm.Def看了一会。“你是个天才。”

Asm.Def想把病毒伪装成杀毒软件，入侵透明计算网络。透明计算网络的文件结构是一棵N个节点的树，每个病毒可以入侵一条路径上的所有节点。但如果两个病毒入侵了同一个节点，由于它们伪装成了杀毒软件，就会自相残杀。Asm.Def不希望这样的情况发生，所以他需要仔细制定入侵计划。为此，他需要频繁地询问，两条路径是否经过同一个节点（即是否相交）。

【输入格式】

第一行两个整数N,Q。

接下来N-1行，每行两个整数a,b，表示(a,b)是树上的一条边。

接下来Q行，每行四个整数s1,t1,s2,t2，表示询问s1~t1的路径是否与s2~t2的路径相交。

【输出格式】

对每个询问，若相交则输出一行”YES”，否则输出一行”NO”。

【样例输入】

6 5
1 2
1 3
2 4
4 5
4 6
1 1 5 6
1 2 6 3
2 3 5 6
6 4 3 1
4 3 1 2

【样例输出】

NO
YES
NO
NO
YES

【提示】

N,Q<=1000.

1<=s1,t1,s2,t2<=N。

【来源】

在此键入。

倍增求LCA

屠龙宝刀点击就送

#include <ctype.h>
#include <cstdio>
#define M 1005

void read(int &x)
{
    x=;
    bool f=;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=;ch=getchar();}
    while(isdigit(ch)) {x=x*+ch-'';ch=getchar();}
    x=f?(~x)+:x;
}
struct node
{
    int next,to;
    node(int next=,int to=) :next(next),to(to){}
}edge[M<<];
int head[M],cnt,dad[M][],dep[M],N,Q;
void add(int u,int v)
{
    edge[++cnt]=node(head[u],v);
    head[u]=cnt;
}
void dfs(int x)
{
    dep[x]=dep[dad[x][]]+;
    for(int i=;dad[x][i];i++)
    dad[x][i+]=dad[dad[x][i]][i];
    for(int i=head[x];i;i=edge[i].next)
    {
        if(dad[x][]!=edge[i].to)
        {
            dad[edge[i].to][]=x;
            dfs(edge[i].to);
        }
    }
}
int max(int a,int b) {return a>b?a:b;}
void swap(int &x,int &y)
{
    int tmp=y;
    y=x;
    x=tmp;
}
int lca(int x,int y)
{
    if(dep[x]>dep[y]) swap(x,y);
    for(int i=;i>=;i--)
    if(dep[dad[y][i]]>=dep[x]) y=dad[y][i];
    if(x==y) return x;
    for(int i=;i>=;i--)
    if(dad[x][i]!=dad[y][i]) x=dad[x][i],y=dad[y][i];
    return dad[x][];
}
int main()
{
    freopen("asm_virus.in","r",stdin);
    freopen("asm_virus.out","w",stdout);
    read(N);
    read(Q);
    for(int x,y,i=;i<N;i++)
    {
        read(x);
        read(y);
        add(x,y);
        add(y,x);
    }
    dfs();
    for(int a,b,c,d;Q--;)
    {
        read(a);read(b);
        read(c);read(d);
        int maxn=;
        maxn=max(lca(a,b),lca(c,d));
        if(lca(a,c)>=maxn||lca(a,d)>=maxn||lca(b,c)>=maxn||lca(b,d)>=maxn) printf("YES\n");
        else printf("NO\n");
    }
    return ;
}