poj 1330

来源:互联网 发布:淘宝店加权重有哪些 编辑:程序博客网 时间:2024/06/06 01:30

LCA裸题。
这里我用tarjan(dfs+并查集)的方法进行离线处理。
这里写图片描述
这是第一个例子的图,题目中求的是16和7的LCA,那么这个过程是这样的。
我们从根节点,即8号开始搜索,每次计算完LCA(u)以后,那么我们就处理了以u为根节点的子树,比如我们计算了4,6,15,7那么这棵子树就为一个集合,然后它们的根节点为4,然后再从4往右边走,当我们走到16时,发现7已经被处理过了,说明此时它们的第一个交叉点就是此时它们最近公共祖先。

#include<stdio.h>#include<iostream>#include<string>#include<string.h>#include<algorithm>#include<iomanip>#include<vector>#include<time.h>#include<queue>#include<stack>#include<iterator>#include<math.h>#include<stdlib.h>#include<limits.h>#include<map>#include<set>//#define ONLINE_JUDGE#define eps 1e-8#define INF 0x7fffffff#define FOR(i,a) for((i)=0;i<(a);(i)++)#define MEM(a) (memset((a),0,sizeof(a)))#define sfs(a) scanf("%s",a)#define sf(a) scanf("%d",&a)#define sfI(a) scanf("%I64d",&a)#define pf(a) printf("%d\n",a)#define pfI(a) printf("%I64d\n",a)#define pfs(a) printf("%s\n",a)#define sfd(a,b) scanf("%d%d",&a,&b)#define sft(a,b,num) scanf("%d%d%d",&a,&b,&num)#define for1(i,a,b) for(int i=(a);i<b;i++)#define for2(i,a,b) for(int i=(a);i<=b;i++)#define for3(i,a,b)for(int i=(b);i>=a;i--)#define MEM1(a) memset(a,0,sizeof(a))#define MEM2(a) memset(a,-1,sizeof(a))#define ll __int64const double PI=acos(-1.0);template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}template<class T> inline T Min(T a,T b){return a<b?a:b;}template<class T> inline T Max(T a,T b){return a>b?a:b;}using namespace std;int n,m;#define M 110#define N 100010#define Mod 1000000007#define p(x,y) make_pair(x,y)const int size = 10010;vector<int> node[size],que[size];int vis[size],anc[size];int fa[size],rank[size];int deg[size];void Init(){    for(int i=0;i<=n;i++){      //初始化        node[i].clear();    //存储点        que[i].clear();         //存储询问        vis[i]=0;       //记录以i为根的子树是否已经被访问过了        anc[i]=0;       //记录i的祖先是谁        fa[i]=i;        //记录i属于哪颗子树        rank[i]=1;          //记录每颗子树的秩        deg[i]=0;       //记录每个点的入度    }}int find(int x){        //并查集路径压缩    return x==fa[x]?fa[x]:fa[x]=find(fa[x]);}void Union(int x,int y){    //并查集合并操作    int fx = find(x);    int fy = find(y);    if(fx != fy){        if(rank[fx]<=rank[fy]){            fa[fx] = fy;            rank[fy] += rank[fx];        }else{            fa[fy] = fx;            rank[fx] += rank[fy];        }    }}void LCA(int u){            //dfs+并查集 Tarjan算法求LCA    anc[u] = u;         //第一次遍历到时赋初值    int size = (int)node[u].size();    for(int i=0;i<size;i++){        int v = node[u][i];        LCA(v);             //处理以u为节点的子树        Union(u,v);     //将u节点和子节点合并        anc[find(v)] = u;   //将u所有子节点的父节点指向u    }    vis[u] = 1;     //标记u的整个子树已经访问过了    size = (int)que[u].size();    for(int i=0;i<size;i++){        if(vis[que[u][i]]){     //如果另外一个点也被访问过了            printf("%d\n",anc[find(que[u][i])]);    //那么就输出它们的LCA            continue;        }    }}int main() {#ifndef ONLINE_JUDGE    freopen("in.txt", "r", stdin);//  freopen("out.txt", "w", stdout);#endif    int t;    sf(t);    while(t--){        scanf("%d",&n);        Init();        int u,v;        for(int i=1;i<n;i++){            scanf("%d%d",&u,&v);            node[u].push_back(v);   //我们将图按照有向图来处理            deg[v]++;        }        scanf("%d%d",&u,&v);        //询问u和v之间的LCA是哪个节点        que[u].push_back(v);        que[v].push_back(u);        for(int i=1;i<=n;i++){  //找出度数为0的开始dfs            if(deg[i] == 0){                LCA(i);                break;            }        }    }    return 0;}
0 0