hdoj 2586 How far away ? 【Tarjan离线LCA】

来源:互联网 发布:长视频软件 编辑:程序博客网 时间:2024/06/16 05:17

题目:hdoj 2586 How far away ?


题意:给出一个有权树,求任意两点的之间的距离。


分析:思想就是以一个点 root 作为跟变成有根数,然后深搜处理处所有点到跟的距离。求要求的两个点的LCA(最近公共祖先),

然后ans = dis【x】 + dis【y】 - 2 * dis【LCA(x,y)】,可以画图分析一下就知道。


求LCA我用的是Tarjan离线lca,由于询问次数很多,所以这个比较快。


AC代码:

#include <iostream>#include <cstdio>#include <cstring>#include <queue>#include <vector>const int N = 44000;#define INF 1000000000using namespace std;struct Node{    int to,val;};vector<Node> v[N];vector<int> query[N],num[N];int ans[N];bool vis[N];int dis[N],father[N];void isit(int n){    for(int i = 0;i<=n;i++)    {        v[i].clear();        query[i].clear();        num[i].clear();        vis[i] = false;        dis[i] = 0;        father[i] = i;    }}int find(int x){    if(x==father[x])        return x;    return father[x] = find(father[x]);}void Union(int x,int y){    x = find(x);    y = find(y);    if(x!=y)        father[y] = x; //让孩子节点指向父节点}void Tarjan(int o,int val){    vis[o] = true;    dis[o] = val;    for(int i = 0; i < v[o].size();i++)    {        Node tmp = v[o][i];        if(vis[tmp.to])            continue;        Tarjan(tmp.to,val+tmp.val);        Union(o,tmp.to);    }    for(int i=0;i<query[o].size();i++)    {        int tmp = query[o][i];        if(vis[tmp])            ans[num[o][i]] = dis[o] + dis[tmp] - 2*dis[find(tmp)];    }}int main(){    int T;    scanf("%d",&T);    while(T--)    {        int n,m;        scanf("%d%d",&n,&m);        isit(n);        for(int i = 1;i < n; i++)        {            int x,y,z;            scanf("%d%d%d",&x,&y,&z);            v[x].push_back((Node){y,z});            v[y].push_back((Node){x,z});        }        for(int i=0;i<m;i++)        {            int x,y;            scanf("%d%d",&x,&y);            query[x].push_back(y);            num[x].push_back(i);            query[y].push_back(x);            num[y].push_back(i);        }        Tarjan(1,0);        for(int i = 0;i<m;i++)            printf("%d\n",ans[i]);    }    return 0;}


0 0
原创粉丝点击