小机房的树 codevs 2370
题目描述 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
思路:
裸lca强行ac(极端数据会超时,但是我还是ac了)
来,上代码:
#include<cstdio> #include<algorithm> using namespace std; struct node { int to,dis,next; }; struct node edge[50000*2+1]; int n,m,head[50001],num=0,dfn[50001],flag=0; char ch; inline void read_int(int &x) { x=0;ch=getchar(); while(ch>'9'||ch<'0') ch=getchar(); while(ch>='0'&&ch<='9'){x=x*10+(int)(ch-'0');ch=getchar();} } inline void edge_add(int from,int to,int dis) { num++; edge[num].to=to; edge[num].dis=dis; edge[num].next=head[from]; head[from]=num; } void dfs(int now) { dfn[now]=++flag; for(int i=head[now];i;i=edge[i].next) if(!dfn[edge[i].to]) dfs(edge[i].to); } int lca(int now_1,int now_2) { int dis_1=0,dis_2=0; while(dfn[now_2]>dfn[now_1]) { for(int i=head[now_2];i;i=edge[i].next) { if(dfn[edge[i].to]<dfn[now_2]) { now_2=edge[i].to; dis_2+=edge[i].dis; } } } while(dfn[now_1]>dfn[now_2]) { for(int i=head[now_1];i;i=edge[i].next) { if(dfn[edge[i].to]<dfn[now_1]) { now_1=edge[i].to; dis_1+=edge[i].dis; } } } return dis_1+dis_2; } int main() { read_int(n); int from,to,dis; for(int i=1;i<n;i++) { read_int(from),read_int(to),read_int(dis); edge_add(from,to,dis); edge_add(to,from,dis); } read_int(m); dfs(1); for(int i=1;i<=m;i++) { read_int(from),read_int(to); if(dfn[from]>dfn[to]) swap(from,to); printf("%d\n",lca(from,to)); } return 0; }