POJ 1986 - Distance Queries(LCA‘ 在线算法RMQ)

原创 2015年07月10日 17:30:46

题目:http://poj.org/problem?id=1986

题意:N个节点M条边的一棵树,M条边的信息,K个询问,求出两个端点之间的距离。

思路:LCA的在线算法RMQ。

该算法详细解析见http://blog.csdn.net/y990041769/article/details/40887469

AC.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 40005;
int rmq[2*maxn];

struct ST {
    int mm[2*maxn];
    int dp[2*maxn][30];
    void init(int n) {
        mm[0] = -1;
        for(int i = 1; i <= n; ++i) {
            mm[i] = ((i&(i-1)) == 0) ? mm[i-1]+1: mm[i-1];
            dp[i][0] = i;
        }
        for(int j = 1; j <= mm[n]; ++j) {
            for(int i = 1; i+(1<<j) - 1 <= n; i++) {
                dp[i][j] = rmq[dp[i][j-1]] < rmq[dp[i+(1<<(j-1))][j-1]]?
                                            dp[i][j-1]: dp[i+(1<<(j-1))][j-1];
            }
        }
    }
    int query(int a, int b) {
        if(a > b) swap(a, b);
        int k = mm[b-a+1];

        int p = dp[a][k], q = dp[b-(1<<k)+1][k];
        return rmq[p] <= rmq[q]? p: q;
    }
};

struct Edge {
    int to, next, w;
};
Edge edge[maxn*2];

int tot, head[maxn];
int F[maxn*2], P[maxn], dis[maxn];
bool flag[maxn];
int cnt;

ST st;
void init()
{
    tot = 0;
    memset(flag, 0,sizeof(flag));
    memset(head, -1, sizeof(head));
}
void addedge(int u, int v, int w)
{
    edge[tot].w = w;
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot++;
}

void dfs(int u, int pre, int dep)
{
    F[++cnt] = u;
    rmq[cnt] = dep;
    P[u] = cnt;
    for(int i = head[u]; i != -1; i = edge[i].next) {
        int v = edge[i].to;
        if(v == pre) continue;
        dis[v] = dis[u] + edge[i].w;
        dfs(v, u, dep+1);
        F[++cnt] = u;
        rmq[cnt] = dep;
    }
}
void LCA_init(int root, int node_num)
{
    cnt = 0;
    dis[1] = 0;
    dfs(root, root, 0);
    st.init(2*node_num-1);
}
int query_lac(int u, int v)
{
    return F[st.query(P[u], P[v])];
}

int main()
{
    //freopen("in", "r", stdin);
    int V, N, u, v;
    while(~scanf("%d %d", &V, &N)) {
        int u, v, w;
        char r;
        init();

        for(int i = 0; i < N; ++i) {
            scanf("%d%d%d %c", &u, &v, &w, &r);
            addedge(u, v, w);
            addedge(v, u, w);
            flag[v] = true;
        }
        int root;
        for(int i = 1; i <= V; ++i) {
            if(!flag[i]) {
                root = i;
                break;
            }
        }

        LCA_init(root, V);

        int K;
        scanf("%d", &K);
        while(K--) {
            int u, v;
            scanf("%d %d", &u, &v);
            int c = query_lac(u, v);
            //printf("c = %d\n", c);
            //printf("%d %d %d\n", dis[u], dis[v], dis[c]);
            printf("%d\n", dis[u]+dis[v] - 2*dis[c]);
        }
    }
    return 0;
}


相关文章推荐

POJ 1986 Distance Queries (在线LCA转RMQ)

题目地址:POJ 1986 纯模板题。输入的最后一个字母是多余的,完全不用管。还有注意询问的时候有相同点的情况。 代码如下:#include #include #include #inclu...

POJ1986 Distance Queries【最近公共祖先】【Tarjan-LCA算法】

题目大意:John是一个农场主,他的牛很懒,拒绝按照John选的路走。John不得不找一条 最短的路。这道题的输入前半部分和POJ1984"Navigation Nightmare"相同。在每组数据 ...

POJ--1986[Distance Queries] DFS+RMQ(最近公共祖先)

题目大意:给你一棵树,求任意两结点u,v之间的距离。   源代码: /*DFS+RMQ(最近公共祖先)*/ /* 注: (1):Index[x]:表示标号x的结点最早出现的位置 (2):R[]...

POJ 1986 Distance Queries [LCA]

#include #include #include #include using namespace std; const int maxn = 40010; const int max...
  • ahfywff
  • ahfywff
  • 2012年05月26日 16:28
  • 365

poj 1986 Distance Queries(LCA求两点距离模版)

Distance Queries Time Limit: 2000MS   Memory Limit: 30000K Total Submissions: 7673   Accepted:...
  • WEYuLi
  • WEYuLi
  • 2013年08月19日 22:07
  • 615

POJ1986 Distance Queries(LCA最近公共祖先)

Distance Queries Time Limit: 2000MS   Memory Limit: 30000K Total Submissions: 10604 ...

[POJ 1986] Distance Queries (LCA)

Distance Queries LCA问题: http://www.cppblog.com/Icyflame/archive/2009/07/04/88987.html Tarjin (离...
  • SIOFive
  • SIOFive
  • 2014年02月12日 15:19
  • 665

POJ 1986 Distance Queries(查询两点距离,LCA)

题目链接: POJ 1986 Distance Queries 题意: 给一个连通的nn个节点的树,有QQ次查询,每次输出两点间距离。 数据范围:n≤40000n\leq 40000 ...
  • Ramay7
  • Ramay7
  • 2016年08月13日 13:35
  • 194

【POJ1986】Distance Queries(lca的应用)

题目链接题目大意: 给你n个点,m条边(是一颗树 q个查询。 每个查询给出两个点,求两点的距离。感觉上树链剖分也可做,不难但是写起来麻烦。这里用倍增lca来做。代码比较好写。(其他lca的写法也...

poj1986Distance Queries(LCA)

题目大意: 其实就是一颗三叉树
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:POJ 1986 - Distance Queries(LCA‘ 在线算法RMQ)
举报原因:
原因补充:

(最多只允许输入30个字)