[BZOJ1602][Usaco2008 Oct]牧场行走

1602: [Usaco2008 Oct]牧场行走

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 2065  Solved: 1084
[Submit][Status][Discuss]

Description

N头牛(2<=n<=1000)别人被标记为1到n,在同样被标记1到n的n块土地上吃草,第i头牛在第i块牧场吃草。 这n块土地被n-1条边连接。 奶牛可以在边上行走,第i条边连接第Ai,Bi块牧场,第i条边的长度是Li(1<=Li<=10000)。 这些边被安排成任意两头奶牛都可以通过这些边到达的情况,所以说这是一棵树。 这些奶牛是非常喜欢交际的,经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。

Input

*第一行:两个被空格隔开的整数:N和Q

 *第二行到第n行:第i+1行有两个被空格隔开的整数:AI,BI,LI

*第n+1行到n+Q行:每一行有两个空格隔开的整数:P1,P2,表示两头奶牛的编号。

Output

*第1行到第Q行:每行输出一个数,表示那两头奶牛之间的距离。

Sample Input

4 2
2 1 2
4 3 2
1 4 3
1 2
3 2

Sample Output

2
7
 
离线询问树上点对距离
#include <cstdio>
const int maxn = 1000 + 10, maxq = 1000 + 10;
struct Edge{
    int to, val, next;
    Edge(){}
    Edge(int _t, int _v, int _n): to(_t), val(_v), next(_n){}
}e[maxn * 2];
int efir[maxn] = {0}, ecnt = 0;
inline void eins(int u, int v, int w){
    e[++ecnt] = Edge(v, w, efir[u]); efir[u] = ecnt;
    e[++ecnt] = Edge(u, w, efir[v]); efir[v] = ecnt;
}
struct Query{
    int to, next, id;
    Query(){}
    Query(int _t, int _n, int _i): to(_t), next(_n), id(_i){}
}q[maxq * 2];
int qfir[maxn] = {0}, qcnt = 0;
inline void qins(int u, int v, int id){
    q[++qcnt] = Query(v, qfir[u], id); qfir[u] = qcnt;
    q[++qcnt] = Query(u, qfir[v], id); qfir[v] = qcnt;
}
int fa[maxn];
int Find(int x){
    return x == fa[x] ? x : fa[x] = Find(fa[x]);
}
int ai[maxq], bi[maxq], lca[maxq];
int ance[maxn], dis[maxn];
bool mark[maxn] = {false};
void dfs(int u, int f){
    ance[u] = u;
    for(int v, i = efir[u]; i; i = e[i].next){
        v = e[i].to;
        if(v == f) continue;        
        dis[v] = dis[u] + e[i].val;
        dfs(v, u);
        fa[Find(v)] = u;
    }
    mark[u] = true;
    for(int v, i = qfir[u]; i; i = q[i].next){
        v = q[i].to;
        if(mark[v]) lca[q[i].id] = ance[Find(v)];
    }
}
int main(){
    int n, q;
    scanf("%d %d", &n, &q);
    for(int u, v, w, i = 1; i < n; i++){
        scanf("%d %d %d", &u, &v, &w);
        eins(u, v, w);
    }
    for(int i = 1; i <= q; i++){
        scanf("%d %d", ai + i, bi + i);
        qins(ai[i], bi[i], i);
    }
    for(int i = 1; i <= n; i++) fa[i] = i;
    dfs(1, 0);
    for(int i = 1; i <= q; i++)
        printf("%d\n", dis[ai[i]] + dis[bi[i]] - 2 * dis[lca[i]]);
    return 0;
} 

 

posted @ 2017-09-02 22:43  Elder_Giang  阅读(125)  评论(0编辑  收藏  举报