hdu 2586:题意:输入n个点的n-1条边的树,m组询问任意点 a b之间的最短距离

思路:LCA中的Tarjan算法,RMQ还不会。。

#include <stdio.h>
#include <iostream>
#include <string.h>
using namespace std;
const int N = ;
const int M = ; int head[N]; //树边邻接表的表头
int __head[N]; //保存询问的邻接表的表头
struct edge{ //保存边
int u,v,w,next;
}e[*N];
struct ask{ //保存询问
int u,v,lca,next;
}ea[M];
int dir[N]; //保存点到树根的距离
int fa[N]; //并查集,保存集合的代表元素
int ance[N]; //保存集合的组合,注意对象是集合而不是元素
bool vis[N]; //遍历时的标记数组 inline void add_edge(int u,int v,int w,int &k) //保存边
{
e[k].u = u; e[k].v = v; e[k].w = w;
e[k].next = head[u]; head[u] = k++;
u = u^v; v = u^v; u = u^v;
e[k].u = u; e[k].v = v; e[k].w = w;
e[k].next = head[u]; head[u] = k++;
} inline void add_ask(int u ,int v ,int &k) //保存询问
{
ea[k].u = u; ea[k].v = v; ea[k].lca = -;
ea[k].next = __head[u]; __head[u] = k++;
u = u^v; v = u^v; u = u^v; ///看上去深奥。。其实就是swap(u,v);
ea[k].u = u; ea[k].v = v; ea[k].lca = -;
ea[k].next = __head[u]; __head[u] = k++;
} int Find(int x)
{
return x == fa[x] ? x : fa[x] = Find(fa[x]);
}
void Union(int u ,int v)
{
fa[v] = fa[u]; //可写为 fa[Find(v)] = fa[u];
} void Tarjan(int u)
{
vis[u] = true;
ance[u] = fa[u] = u; //可写为 ance[Find(u)] = fa[u] = u;
for(int k=head[u]; k!=-; k=e[k].next)
if( !vis[e[k].v] )
{
int v = e[k].v , w = e[k].w;
dir[v] = dir[u] + w;
Tarjan(v);
Union(u,v);
ance[Find(u)] = u;
}
for(int k=__head[u]; k!=-; k=ea[k].next)
if( vis[ea[k].v] )
{
int v = ea[k].v;
ea[k].lca = ea[k^].lca = ance[Find(v)];
}
} int main()
{
int tcase;
scanf("%d",&tcase);
while(tcase--){
int n,q;
scanf("%d%d",&n,&q);
memset(head,-,sizeof(head));
memset(__head,-,sizeof(__head));
int tot = ;
for(int i=; i<n; i++) //建树
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,w,tot);
}
tot = ;
for(int i=; i<q; i++) //拆开保存询问
{
int u,v;
scanf("%d%d",&u,&v);
add_ask(u,v,tot);
}
memset(vis,,sizeof(vis));
dir[] = ;
Tarjan();
for(int i=; i<q; i++)
{
int s = i * , u = ea[s].u , v = ea[s].v , lca = ea[s].lca;
printf("%d\n",dir[u]+dir[v]-*dir[lca]);
}
} return ;
}

hdu 2874:和上题一样都是求两点之间的最短距离,但是有多棵树,所以存在不连通的情况(用father判断一下就OK),,然后华丽丽的 超内存,拿别人的代码也是MLE

#include <stdio.h>
#include <iostream>
#include <string.h>
using namespace std;
const int N = ;
const int M = ; int head[N];
int __head[N];
struct edge{
int u,v,w,next;
}e[*N];
struct ask{
int u,v,lca,next;
}ea[*M];
int dir[N];
int fa[N];
int ance[N];
bool vis[N]; inline void add_edge(int u,int v,int w,int &k) //保存边
{
e[k].u = u; e[k].v = v; e[k].w = w;
e[k].next = head[u]; head[u] = k++;
u = u^v; v = u^v; u = u^v;
e[k].u = u; e[k].v = v; e[k].w = w;
e[k].next = head[u]; head[u] = k++;
} inline void add_ask(int u ,int v ,int &k) //保存询问
{
ea[k].u = u; ea[k].v = v; ea[k].lca = -;
ea[k].next = __head[u]; __head[u] = k++;
u = u^v; v = u^v; u = u^v; ///看上去深奥。。其实就是swap(u,v);
ea[k].u = u; ea[k].v = v; ea[k].lca = -;
ea[k].next = __head[u]; __head[u] = k++;
} int Find(int x)
{
return x == fa[x] ? x : fa[x] = Find(fa[x]);
}
void Union(int u ,int v)
{
int x = Find(u);
int y = Find(v);
fa[x] = y;
} void Tarjan(int u)
{
vis[u] = true;
ance[u] = fa[u] = u; //可写为 ance[Find(u)] = fa[u] = u;
for(int k=head[u]; k!=-; k=e[k].next)
if( !vis[e[k].v] )
{
int v = e[k].v , w = e[k].w;
dir[v] = dir[u] + w;
Tarjan(v);
Union(u,v);
ance[Find(u)] = u; //可写为ance[u] = u; //甚至不要这个语句都行
}
for(int k=__head[u]; k!=-; k=ea[k].next)
if( vis[ea[k].v] )
{
int v = ea[k].v;
ea[k].lca = ea[k^].lca = ance[Find(v)];
}
} int main()
{
int k,n,q;
while(scanf("%d%d%d",&n,&k,&q)!=EOF){ memset(head,-,sizeof(head));
memset(__head,-,sizeof(__head));
int tot = ;
for(int i=; i<k; i++) //建树
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,w,tot);
}
tot = ;
for(int i=; i<q; i++) //拆开保存询问
{
int u,v;
scanf("%d%d",&u,&v);
add_ask(u,v,tot);
}
memset(vis,,sizeof(vis));
for(int i=;i<=n;i++){ ///访问每个节点
if(!vis[i]){
dir[i]=;
Tarjan(i);
}
}
for(int i=; i<q; i++)
{
int s = i * , u = ea[s].u , v = ea[s].v , lca = ea[s].lca; ///s代表偶数边,奇偶只是方向不同罢了,所以取一个就行
if(fa[u]!=fa[v]) printf("Not connected\n"); ///父亲结点不同当然就不是同一棵子树了
else printf("%d\n",dir[u]+dir[v]-*dir[lca]);
}
} return ;
}

最新文章

  1. Aspose.Words简单生成word文档
  2. 命名规范(数据库,c#)
  3. php获取数组第一个值 current()
  4. java学习笔记(2):获取文件名和自定义文件过滤器
  5. Leetcode Unique Binary Search Trees
  6. SharePoint 2010 ——自定义上传页面与多文件上传解决方案
  7. Jdk1.8+Eclipse+MySql+Tomcat开发Java应用的环境搭建
  8. POJ 2482 Stars in Your Window (线段树+扫描线+区间最值,思路太妙了)
  9. linux-rm
  10. Design Pattern Iterator 迭代器设计模式
  11. The certificate that was used has a trust chain that cannot be verified问题
  12. PAT乙级 1034
  13. 正则表达式入门+实战(c#实现)
  14. linux(centos)常用命令
  15. laravel整合workerman做聊天室
  16. react 环境安装
  17. 大家多开发点uwp吧
  18. Vim 利剑常磨,见血封喉
  19. [04] 高级映射 association和collection
  20. OpenGL chapter4 基础变换

热门文章

  1. CF724E Goods transportation
  2. PHP汉字转拼音
  3. 【计数】【UVA11401】 Triangle Counting
  4. HDU1166:敌兵布阵(线段树模板)
  5. C++派生类继承的理解
  6. 彻底解决_OBJC_CLASS_$_某文件名&quot;, referenced from:问题
  7. js_同步和异步
  8. python产生随机样本数据
  9. mysql增删
  10. Python3 高阶函数