【codevs 2370】小机房的树

来源:互联网 发布:印度软件产值 编辑:程序博客网 时间:2024/05/21 08:55

题目描述 Description
小机房有棵焕狗种的树,树上有N个节点,节点标号为0到N-1,有两只虫子名叫飘狗和大吉狗,分居在两个不同的节点上。有一天,他们想爬到一个节点上去搞基,但是作为两只虫子,他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量(从父亲节点爬到此节点也相同),他们想找出一条花费精力最短的路,以使得搞基的时候精力旺盛,他们找到你要你设计一个程序来找到这条路,要求你告诉他们最少需要花费多少精力

输入描述 Input Description
第一行一个n,接下来n-1行每一行有三个整数u,v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ,v 表示两只虫子所在的节点
输出描述 Output Description
一共有m行,每一行一个整数,表示对于该次询问所得出的最短距离。

样例输入 Sample Input
3

1 0 1

2 0 1

3

1 0

2 0

1 2

样例输出 Sample Output
1

1

2

数据范围及提示 Data Size & Hint
1<=n<=50000, 1<=m<=75000, 0<=c<=1000

暴力
先进行预处理
然后让两个节点跳到同一层,在一起往上跳,都是一次跳一步。
节点标号为0到N-1
完了。

#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#include<algorithm>using namespace std;const int maxn=50001;int first[maxn<<1],next[maxn<<1],tot;int deep[maxn],dist[maxn],fa[maxn][30];int n,m;struct edge{    int f,t,v;}l[maxn<<1];void build(int f,int t,int v){    l[++tot]=(edge){f,t,v};    next[tot]=first[f];    first[f]=tot;    return;}void init(int q,int h){    for(int i=first[h];i;i=next[i])    {        int at=l[i].t,av=l[i].v;        if(at==q) continue;        fa[at][0]=h;        deep[at]=deep[h]+1;        dist[at]=dist[h]+av;        init(h,at);    }    return;}int lca(int x,int y){    if(deep[x]<deep[y]) swap(x,y);    int i;    for(i=x;i>0;i=fa[i][0])    {        if(deep[i]==deep[y]) break;    }    if(i==y) return y;    while(1)    {        i = fa[i][0];        y = fa[y][0];        if(i==y)        break;    }    return i;}int main(){    int a,b,c;    scanf("%d",&n);    for(int i=1;i<n;i++)//节点标号为0到N-1    {        scanf("%d%d%d",&a,&b,&c);        build(a,b,c);        build(b,a,c);    }    init(0,0);    scanf("%d",&m);    for(int i=1;i<=m;i++)    {        scanf("%d%d",&a,&b);        cout<<dist[a]+dist[b]-(dist[lca(a,b)]<<1)<<'\n';    }    return 0;}

下面是正式的倍增~

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>using namespace std;const int maxn=50000+1;int first[maxn<<1],next[maxn<<1],tot=0;int deep[maxn],dist[maxn],fa[maxn][25];int n,m;struct edge{    int f,t,v;}l[maxn<<1];void build(int f,int t,int v){    l[++tot]=(edge){f,t,v};    next[tot]=first[f];    first[f]=tot;    return;}void init(int q,int h){    for(int i=first[h];i;i=next[i])    {        int at=l[i].t,av=l[i].v;        if(at==q) continue;        fa[at][0]=h;        deep[at]=deep[h]+1;        dist[at]=dist[h]+av;        init(h,at);    }    return;}void make_lca(){    for(int j=1;j<=log2(n);j++)    for(int i=1;i<=n;i++)//for(int i=1;i<n;i++)    fa[i][j]=fa[fa[i][j-1]][j-1];}int lca(int x,int y){    if(deep[x]<deep[y]) swap(x,y);    for(int i=log2(n);i>=0;i--)    {        if(deep[fa[x][i]]>=deep[y])//注意条件~        x=fa[x][i];    }    if(x==y) return y;    for(int i=log2(n);i>=0;i--)    {        if(fa[x][i]!=fa[y][i])//重要条件!!        {            x=fa[x][i];            y=fa[y][i];        }    }    return fa[x][0];//注意这里跟暴力不一样}int main(){    int a,b,c;    scanf("%d",&n);     for(int i=1;i<n;i++)    {        scanf("%d%d%d",&a,&b,&c);        build(a,b,c);        build(b,a,c);    }    init(0,0);    make_lca();//!!!!!!!!!!!    scanf("%d",&m);    while(m--)    {        scanf("%d%d",&a,&b);        cout<<dist[a]+dist[b]-(dist[lca(a,b)]<<1)<<endl;    }    return 0;}
1 0
原创粉丝点击