数据结构总结之lca

1.tarjan：poj1330

# include <iostream>
# include <cstdio>
# include <cstring>
# include <string>
# include <vector>
# define MAXN 500001
using namespace std;
int x,y,t,n,s,ans;
int vis[MAXN],fa[MAXN];
vector <int> vec[MAXN];

int find(int x)
{
    return (x == fa[x]) ? x : fa[x] = find(fa[x]);
}

void dfs(int u)
{
    int i, v;
    for(i = 0; i < vec[u].size(); i++)
    {
        v = vec[u][i];
        dfs(v);
        vis[v]=1;
        fa[v] = u;
    }
    if(u==x && vis[y])
       ans=find(y);
    if(u==y && vis[x])
        ans=find(x);

}

int main()
{
    cin>>t;
    while(t--)
    {
        cin>>n;
        memset(vis,0,sizeof vis);
        for(int i = 1; i < n; i++)
        {
            cin>>x>>y;
            vis[y]=1;
            vec[x].push_back(y);
        }
        cin>>x>>y;
        for(int i=1; i<=n; i++)
        if(vis[i]==0)
        {
            s=i;
            fa[s]=s;
            break;
        }
        memset(vis,0,sizeof vis);
        for(int i=1;i<=n;i++) fa[i]=i;
        dfs(s);
        cout<<ans<<endl;
        for(int i=1;i<=n;i++) vec[i].clear();
    }

    return 0;
}
/*
1//测试样例数
5//边数+1
2 3
3 4
3 1
1 5
3 5输入2个询问点
3//输出

2.倍增法：zoj2838

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <string.h>
using namespace std;
const int n=50002;
vector<int> g[n];//表示图（树）
int nn;
int p[n][20];//表示向上翻2^i代的祖先节点
int vis[n],s[n];//s[i]表示2^i
int deep[n];//表示点i的深度
int dfs(int u,int fa,int dep)//作用：得到了第1代的祖先节点，得出树的深度
{
    int tree_deep=dep;
    deep[u]=dep;
    p[u][0]=fa;//u的第2^0代祖先是u的父亲
    vis[u]=1;
    int len=g[u].size();
    for(int i=0;i<len;i++)
    {
        int v=g[u][i];
        if(vis[v]==0)
        {
            tree_deep=max(tree_deep,dfs(v,u,dep+1));//tree_deep是树的深度
        }
    }
    return tree_deep;
}
void init()
{
    memset(vis,0,sizeof vis);
    memset(p,-1,sizeof p);
    int tree_deep=dfs(0,-1,1);//tree_deep是树的深度
    for(int i=1;s[i]<=tree_deep;i++)//依次得出2^i代的祖先节点
    {
        for(int j=1;j<n;j++)//依次得出节点j的2^i代的祖先节点
            if(p[j][i-1]!=-1)
        {
            int tmp=p[j][i-1];//tmp是j点的第2^(i-1)代祖先
            p[j][i]=p[tmp][i-1];//p[j][i]是j的第2^i代祖先，也就是tmp的第2^(i-1)代祖先
        }
    }
}
int lca(int x,int y)
{
    if(deep[x]<deep[y]) x^=y,y^=x,x^=y;//确保节点x比y深
    int log;
    for(log=0;s[log]<=deep[x];log++);//2^log是x向上移动代数的上限
    log--;
    for(int i=log;i>=0;i--)
    {
        if(deep[x]-s[i]>=deep[y])//如果向上移动2^i代没有超过y那代
            x=p[x][i];//x向上移动2^i代
    }
    if(x==y) return x;//上面那个for循环使得x和y在同一代上，如果刚好x就是y，返回x
    for(int i=log;i>=0;i--)//从上限开始for循环遍历
    {
        if(p[x][i]!=-1 && p[x][i]!=p[y][i])//已经在同一代的x，y同时向上移动2^i,确保没有”移过头“（就是p[x][i]=-1)
            x=p[x][i],y=p[y][i];                                                //还要确保p[x][i]=p[y][i]才能向上移动
    }
    return p[x][0];
}
int main()
{
    s[0]=1;
    for(int i=0;i<19;i++)
        s[i+1]=s[i]*2;
        int t=1;
    while(~scanf("%d",&nn))
    {
         int x,y,z;
         if(t!=1)printf("\n");
         printf("Case %d:\n",t++);
         for(int i=0;i<=n;i++) g[i].clear();
        for(int i=1;i<nn;i++)
        {
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            g[y].push_back(x);
        }
        init();
        int mm;scanf("%d",&mm);
        for(int i=1;i<=mm;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            int a=lca(x,y);
            if(deep[a]<=deep[z])
            {
                if(lca(x,z)==z || lca(y,z)==z)
                {
                    printf("Yes\n");
                    continue;
                }
            }
            printf("No\n");
        }
    }
    return 0;
}

3.暴力找：zoj3664

//暴力lca：这个题，从头找，最后一个符合条件的点就是最近祖先点
//每次画一条线，一个矩形变成2个，这不就是二叉树嘛？
#include <stdio.h>
#include <iostream>
#include <string.h>

using namespace std;
const int N=3000;
struct pot
{
    int x,y;
    pot(int a,int b)
    {
        x=a,y=b;
    }
};
struct re
{
    int x,y,x2,y2;
    void read(int xl,int yl,int xr,int yr)
    {
        x=xl,y=yl,x2=xr,y2=yr;
    }
    bool in(int xl,int yl,int xr,int yr)
    {
        return xl>=x && yl>=y && xr<=x2 && yr<=y2;
    }
};
re c[N];
int xl,yl,xr,yr;
int num;//矩形的编号
int lson[N],rson[N],cnt[N];//cnt表示编号为i的矩阵内含线的条数
int n,q;
void update_from_root(int u)
{
    cnt[u]++;
    if(lson[u]==-1)
    {
        lson[u]=++num,rson[u]=++num;
        c[lson[u]].read(xl,yl,c[u].x2,c[u].y2);
        c[rson[u]].read(c[u].x,c[u].y,xr,yr);
    }
    else
    {
      if(c[lson[u]].in(xl,yl,xr,yr)) update_from_root(lson[u]);
       else update_from_root(rson[u]);
    }
}
int query(int u)
{
    if(lson[u]==-1) return cnt[u];
    if(c[lson[u]].in(xl,yl,xr,yr)) return query(lson[u]);
    if(c[rson[u]].in(xl,yl,xr,yr)) return query(rson[u]);
    return cnt[u];
}
int main()
{
    while(~scanf("%d%d%d%d",&xl,&yl,&xr,&yr))
    {
        memset(cnt,0,sizeof cnt);
        memset(lson,-1,sizeof lson);
        memset(rson,-1,sizeof rson);
        c[1].read(xl,yl,xr,yr);
        scanf("%d%d",&n,&q);
        int n1=n;
        num=1;
        while(n--)
        {
            scanf("%d%d%d%d",&xl,&yl,&xr,&yr);
            if(xl>xr) {int tmp=xl;xl=xr;xr=tmp;}
            if(yl>yr){int tmp=yl;yl=yr;yr=tmp;}
            update_from_root(1);
        }
        while(q--)
        {
            scanf("%d%d%d%d",&xl,&yl,&xr,&yr);
            if(xl>xr) {int tmp=xl;xl=xr;xr=tmp;}
            if(yl>yr){int tmp=yl;yl=yr;yr=tmp;}
            printf("%d\n",n1+1-query(1));
        }
    }
    return 0;
}