1602: [Usaco2008 Oct]牧场行走|Lca
2016-01-14 07:58
357 查看
刷水~~~~~
随便打了个树链剖分求lca
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<algorithm>
#include<iostream>
#define T 2222
#define lowbit(x) (x&(-x))
using namespace std;
int sc()
{
int i=0;char c=getchar();
while(c>'9'||c<'0')c=getchar();
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i;
}
int head[T],lst[T],nxt[T],v[T];
int fa[T],top[T],size[T],y[T],deep[T];
int n,m,tot;
void insert(int x,int y,int z)
{
lst[++tot]=y;
nxt[tot]=head[x];
v[tot]=z;
head[x]=tot;
}
void dfs1(int x,int f)
{
size[x]=1;
deep[x]=deep[f]+1;
fa[x]=f;
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=f)
{
y[lst[i]]=y[x]+v[i];
dfs1(lst[i],x);
size[x]+=size[lst[i]];
}
}
void dfs2(int x,int htp)
{
top[x]=htp;
int k=0;
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=fa[x]&&size[lst[i]]>size[k]) k=lst[i];
if(!k)return; dfs2(k,htp);
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=fa[x]&&lst[i]!=k)
dfs2(lst[i],lst[i]);
}
int Lca(int x,int y)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
x=fa[top[x]];
}
return deep[x]<deep[y]?x:y;
}
int ask(int a,int b)
{
return y[a]+y[b]-2*y[Lca(a,b)];
}
int main()
{
n=sc(),m=sc();
for(int i=1;i<n;i++)
{
int x=sc(),y=sc(),z=sc();
insert(x,y,z);
insert(y,x,z);
}
dfs1(1,0);
dfs2(1,1);
for(int i=1;i<=m;i++)
{
int x=sc(),y=sc();
printf("%d\n",ask(x,y));
}
return 0;
}
随便打了个树链剖分求lca
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<algorithm>
#include<iostream>
#define T 2222
#define lowbit(x) (x&(-x))
using namespace std;
int sc()
{
int i=0;char c=getchar();
while(c>'9'||c<'0')c=getchar();
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i;
}
int head[T],lst[T],nxt[T],v[T];
int fa[T],top[T],size[T],y[T],deep[T];
int n,m,tot;
void insert(int x,int y,int z)
{
lst[++tot]=y;
nxt[tot]=head[x];
v[tot]=z;
head[x]=tot;
}
void dfs1(int x,int f)
{
size[x]=1;
deep[x]=deep[f]+1;
fa[x]=f;
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=f)
{
y[lst[i]]=y[x]+v[i];
dfs1(lst[i],x);
size[x]+=size[lst[i]];
}
}
void dfs2(int x,int htp)
{
top[x]=htp;
int k=0;
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=fa[x]&&size[lst[i]]>size[k]) k=lst[i];
if(!k)return; dfs2(k,htp);
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=fa[x]&&lst[i]!=k)
dfs2(lst[i],lst[i]);
}
int Lca(int x,int y)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
x=fa[top[x]];
}
return deep[x]<deep[y]?x:y;
}
int ask(int a,int b)
{
return y[a]+y[b]-2*y[Lca(a,b)];
}
int main()
{
n=sc(),m=sc();
for(int i=1;i<n;i++)
{
int x=sc(),y=sc(),z=sc();
insert(x,y,z);
insert(y,x,z);
}
dfs1(1,0);
dfs2(1,1);
for(int i=1;i<=m;i++)
{
int x=sc(),y=sc();
printf("%d\n",ask(x,y));
}
return 0;
}
相关文章推荐
- hdu-5405
- hybz1036 树的统计Count
- poj3237 Tree
- hybz2243 染色
- 【bzoj1036】【树链剖分】【ZJOI2008】树的统计Count
- 4034: [HAOI2015]T2 (树链剖分)
- SPOJ 375. Query on a tree (基于边的树链剖分)
- 1036: [ZJOI2008]树的统计Count (基于点的树链剖分)
- 树链剖分简单入门
- poj 3237 树链剖分+线段树
- poj 3237 树链剖分
- SPOJ QTREE 1-3题解
- 【HAOI2015】【BZOJ4034】树上操作T2
- 【BZOJ4127】Abs
- FZU 2082 树链剖分 区间最大和
- 树链剖分模版
- codeforces #329 D. Happy Tree Party (LCA+并查集 || 树链剖分)
- HDU 5458 Stability 树链剖分
- HDU 5274(树链剖分)
- SPOJ Query on a tree (树链剖分)