倍增lca

woacioawdo

已于 2022-02-01 13:35:21 修改

阅读量374

点赞数

文章标签： c++

于 2022-01-31 21:46:57 首次发布

本文链接：https://blog.csdn.net/qq_52004482/article/details/122761817

版权

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网

题目描述

小A这次来到一个景区去旅游，景区里面有N个景点，景点之间有N-1条路径。小A从当前的一个景点移动到下一个景点需要消耗一点的体力值。但是景区里面有两个景点比较特殊，它们之间是可以直接坐观光缆车通过，不需要消耗体力值。而小A不想走太多的路，所以他希望你能够告诉它，从当前的位置出发到他想要去的那个地方，他最少要消耗的体力值是多少。

输入描述:

第一行一个整数N代表景区的个数。
接下来N-1行每行两个整数u,v代表从位置u到v之间有一条路径可以互相到达。
接下来的一行两个整数U,V表示这两个城市之间可以直接坐缆车到达。
接下来一行一个整数Q，表示有Q次询问。
接下来的Q行每行两个整数x,y,代表小A的位置在x，而他想要去的地方是y。

输出描述:

对于每个询问下x,y输出一个结果，代表x到y消耗的最少体力对于每个询问下x,y输出一个结果，代表x到y消耗的最少体力对于每个询问下x,y输出一个结果，代表x到y消耗的最少体力

输入

复制

输出

复制

1
0

题解

#include<iostream>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
#define int long long
const int maxn=3e5+5;
const int INF=1e9+5;
int head[maxn<<1];int d[maxn];
int fa[maxn][20];
struct node{
int to,next;
}edge[maxn<<1];
int cnt1=0;
void add_edge(int from,int to)
{
cnt1++;
edge[cnt1].to=to;
edge[cnt1].next=head[from];
head[from]=cnt1;
}
int lca(int a,int b)
{
if(d[a]<d[b]) swap(a,b);
for(int i=19;i>=0;i--)
{
if(d[fa[a][i]]>=d[b]) a=fa[a][i];
}
if(a==b) return a;
for(int i=19;i>=0;i--)
{
if(fa[a][i]!=fa[b][i])
{
a=fa[a][i];
b=fa[b][i];
}
}
return fa[a][0];
}
queue<int> qu;
signed main()
{
int n;scanf("%lld",&n);
memset(head,-1,sizeof(head));
memset(fa,-1,sizeof(fa));
for(int i=1;i<=n;i++) d[i]=INF;
for(int i=0;i<n-1;i++)
{
int from,to;
scanf("%lld%lld",&from,&to);
add_edge(from,to);
add_edge(to,from);
}
int a,b;scanf("%lld%lld",&a,&b);
d[1]=0;qu.push(1);
while(!qu.empty())
{
int from=qu.front();qu.pop();
for(int i=head[from];i!=-1;i=edge[i].next)
{
int to=edge[i].to;
if(d[to]>d[from]+1)
{
d[to]=d[from]+1;
fa[to][0]=from;
qu.push(to);
}
}
}
for(int k=1;k<20;k++)
{
for(int j=1;j<=n;j++)
{
if(fa[j][k-1]!=-1)
{
fa[j][k]=fa[fa[j][k-1]][k-1];
}
}
}
int q;scanf("%lld",&q);
while(q--)
{
int from,to;scanf("%lld%lld",&from,&to);
int v=lca(from,to);
int ans=INF;
ans=min(ans,d[from]+d[to]-2*d[v]);
int b1=lca(from,b);int a1=lca(a,to);
ans=min(ans,d[from]+d[b]-2*d[b1]+d[to]+d[a]-2*d[a1]);
a1=lca(from,a),b1=lca(b,to);
ans=min(ans,d[from]+d[a]-2*d[a1]+d[to]+d[b]-2*d[b1]);
printf("%lld\n",ans);
}
}

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网

题目描述

已知有 nnn 个节点，有 n−1n-1n−1 条边，形成一个树的结构。

给定一个根节点 kkk，每个节点都有一个权值，节点i的权值为 viv_ivi。

给 mmm 个操作，操作有两种类型：

1 a x ：表示将节点 aaa 的权值加上 xxx

2 a ：表示求 aaa 节点的子树上所有节点的和（包括 aaa 节点本身）

输入描述:

第一行给出三个正整数 n,m,kn,m,kn,m,k，表示树的节点数、操作次数、和这棵树的根节点.

第二行给出 nnn 个正整数，第 iii 个正整数表示第 iii 个节点的权值 valival_ivali

下面 n−1n-1n−1 行每行两个正整数 u,vu,vu,v，表示边的两个端点

接下来 mmm 行，每行给出一个操作

输出描述:

对于每个类型为 2 的操作,输出一行一个正整数,表示以 aaa 为根的子树的所有节点的权值和

示例1

输入

复制5 6 1 1 2 3 4 5 1 3 1 2 2 4 2 5 1 2 10 1 3 10 1 4 5 1 5 1 2 3 2 2

输出

复制13 27

13
27

链接：https://ac.nowcoder.com/acm/contest/27836/C
来源：牛客网

题目描述

已知有 nnn 个节点，有 n−1n-1n−1 条边，形成一个树的结构。

给定一个根节点 kkk，每个节点都有一个权值，节点i的权值为 viv_ivi。

给 mmm 个操作，操作有两种类型：

1 a x ：表示将节点 aaa 的权值加上 xxx

2 a ：表示求 aaa 节点的子树上所有节点的和（包括 aaa 节点本身）

输入描述:

第一行给出三个正整数 n,m,kn,m,kn,m,k，表示树的节点数、操作次数、和这棵树的根节点.

第二行给出 nnn 个正整数，第 iii 个正整数表示第 iii 个节点的权值 valival_ivali

下面 n−1n-1n−1 行每行两个正整数 u,vu,vu,v，表示边的两个端点

接下来 mmm 行，每行给出一个操作

输出描述:

对于每个类型为 2 的操作,输出一行一个正整数,表示以 aaa 为根的子树的所有节点的权值和

示例1

输入

复制5 6 1 1 2 3 4 5 1 3 1 2 2 4 2 5 1 2 10 1 3 10 1 4 5 1 5 1 2 3 2 2

输出

复制13 27

13
27

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define int long long
const int maxn=1e6+5;
int n;
int past[maxn],now[maxn],val[maxn],top[maxn],depth[maxn],fa[maxn],son[maxn],s[maxn],head[maxn<<1],ctv[maxn];
struct node1{
int to,next;
}edge[maxn<<1];
int cc;
void add_edge(int from,int to)
{
cc++;
edge[cc].to=to;
edge[cc].next=head[from];
head[from]=cc;
}
struct node{
int sum;
}tree[maxn*4];
void Pushup(int rt)
{
tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void build(int rt,int l,int r)
{
if(l==r){
tree[rt].sum=val[past[l]];
return ;
}
int m=(l+r)/2;
build(rt<<1,l,m);
build(rt<<1|1,m+1,r);
Pushup(rt);
}
void change(int rt,int l,int r,int pos,int c)
{
if(l==r){
tree[rt].sum+=c;
return ;
}
int m=(l+r)/2;
if(pos<=m) change(rt<<1,l,m,pos,c);
if(pos>m) change(rt<<1|1,m+1,r,pos,c);
Pushup(rt);
}
int query(int rt,int l,int r,int L,int R)
{
int ans=0;
if(l>=L&&r<=R)
{
return tree[rt].sum;
}
int m=(l+r)/2;
if(L<=m) ans+=query(rt<<1,l,m,L,R);
if(R>m) ans+=query(rt<<1|1,m+1,r,L,R);
return ans;
}
void init()
{
memset(head,-1,sizeof(head));
}
void dfs1(int root)
{
depth[root]=depth[fa[root]]+1;
s[root]=1;
for(int i=head[root];i!=-1;i=edge[i].next)
{
int to=edge[i].to;
if(to!=fa[root])
{
fa[to]=root;
dfs1(to);
s[root]+=s[to];
if(s[son[root]]<s[to])
{
son[root]=to;
}
}
}
}
int id=0;
void dfs2(int root,int f)
{
top[root]=f;
now[root]=++id;
past[id]=root;
if(son[root]) dfs2(son[root],f);
for(int i=head[root];i!=-1;i=edge[i].next)
{
int to=edge[i].to;
if(to!=fa[root]&&to!=son[root])
{
dfs2(to,to);
}
}
ctv[root]=id;
}
signed main()
{
int m;int k;
scanf("%lld%lld%lld",&n,&m,&k);
init();
for(int i=1;i<=n;i++) scanf("%lld",&val[i]);
for(int i=0;i<n-1;i++)
{
int from,to; scanf("%lld%lld",&from,&to);
add_edge(from,to);
add_edge(to,from);
}
dfs1(k);dfs2(k,k);
build(1,1,n);
while(m--)
{
int op;scanf("%lld",&op);
if(op==1)
{
int a,x;scanf("%lld%lld",&a,&x);
change(1,1,n,now[a],x);
}
if(op==2)
{
int a;scanf("%lld",&a);
printf("%lld\n",query(1,1,n,now[a],ctv[a]));
}
}
}

woacioawdo

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
倍增lca

链接：登录—专业IT笔试面试备考平台_牛客网来源：牛客网题目描述小A这次来到一个景区去旅游，景区里面有N个景点，景点之间有N-1条路径。小A从当前的一个景点移动到下一个景点需要消耗一点的体力值。但是景区里面有两个景点比较特殊，它们之间是可以直接坐观光缆车通过，不需要消耗体力值。而小A不想走太多的路，所以他希望你能够告诉它，从当前的位置出发到他想要去的那个地方，他最少要消耗的体力值是多少。输入描述:第一行一个整数N代表景区的个数。接下来N-1行每行两个整数u,v代表从位置u到v之间有.
复制链接

扫一扫