首页 > 其他分享 >树上前缀和与差分(边权)

树上前缀和与差分(边权)

时间:2022-11-07 18:55:18浏览次数:62  
标签:最简 前缀 int 51 边权 42 差分 61

问题描述
有一棵n个点的树,每个点i有点树v[i],q个询问求点u到点v最简路径上所有点权之和
输入格式
第一行n,q表示有n个点,q个询问
第二行n个整数表示每个点的权
下面n-1每行三个整数x,y,z描述边的信息及边长
下面q行每行两个整数u,v表示询问的两个点路径长度
输出
q行,每行为u到v最简路径长度
样例

in
5 5
1 2 4
2 3 6
1 4 2
4 5 6
1 5
1 4
2 5
3 4
2 4
out
8
2
12
12
6

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
struct node{
	int to;
	int dis;
	int nxt;
}E[maxn<<1];
int h[maxn],dist[maxn],dep[maxn],f[maxn][25],lg[maxn],etot;
void addedge(int x,int y,int d)
{
	E[++etot].to=y;
	E[etot].dis=d;
	E[etot].nxt=h[x];
	h[x]=etot;
}
void dfs(int u,int fa)
{
	dep[u]=dep[fa]+1;
	f[u][0]=fa;
	for (int i=1;(1<<i)<=dep[u];i++)
	{
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for (int i=h[u];i;i=E[i].nxt)
	{
		int v=E[i].to;
		if (v!=fa)
		{
			dist[v]+=dist[u]+E[i].dis;
			dfs(v,u);
		}
	}
}
int lca(int x,int y)
{
	if (dep[x]<dep[y]) swap(x,y);
	int d=dep[x]-dep[y];
	while(d)
	{
		int k=lg[d];
		x=f[x][k];
		d=d-(1<<k);
	}	
	if (x==y) return x;
	for (int i=lg[dep[x]];i>=0;i--)
	{
		if (f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		} 
	}
	return f[x][0];
} 
int main()
{
	int n,q;
	cin>>n>>q;
	for (int i=1;i<n;i++)
	{
		int x,y,d;
		cin>>x>>y>>d;
		addedge(x,y,d);
		addedge(y,x,d);
	}
	dfs(1,0);
	for (int i=1;i<=q;i++)
	{
		int x,y;
		cin>>x>>y;
		int l=lca(x,y);
		cout<<dist[x]+dist[y]-2*dist[l]<<endl;
	}
}

  

 

标签:最简,前缀,int,51,边权,42,差分,61
From: https://www.cnblogs.com/smghj/p/16867026.html

相关文章

  • CodeTON Round 3 (C.差分维护,D.容斥原理)
    C.ComplementaryXOR题目大意:给你两个01串ab,问你是否可以通过一下两种操作在不超过n+5次的前提下将两个串都变为0,同时需要输出可以的操作方案选择一个区间[l,r]将......
  • 树上前缀和
    树上前缀和问题描述有一棵n个点的树,每个点i有点树v[i],q个询问求点u到点v最简路径上所有点权之和输入格式第一行n,q表示有n个点,q个询问第二行n个整数表示每个点的权下面......
  • 二维数组的前缀和
    二维数组的前缀和设二维数组,intarr[5][7];,以arr[1][1]作为作为矩形的左上角坐标,以此开始存储数据,数组最左边,最上边不存储数据,为空设二维数组,int......
  • R语言使用逻辑回归Logistic、单因素方差分析anova、异常点分析和可视化分类iris鸢尾花
    摘要本文将探讨Fisher和Anderson ​​鸢尾花​​数据集中呈现的三个变量之间的关系,特别是virginica和versicolor级别的因变量变量物种对预测变量花瓣长度和花瓣宽度......
  • 基础算法篇——前缀和与差分
    基础算法篇——前缀和与差分本次我们介绍基础算法中的快速排序,我们会从下面几个角度来介绍前缀和:前缀和介绍前缀和基本题型前缀和介绍首先我们来简单介绍一下前缀和......
  • 浏览器私有前缀
    浏览器私有前缀1、私有前缀-moz-:代表Firefox浏览器私有属性-ms-:代表ie浏览器私有属性-webkit-:代表Safari、Chrome浏览器私有属性-o-:代表Opera浏览器私有属性2、提......
  • P5369 [PKUSC2018]最大前缀和
    题意给定一个序列\(a\),求\(a\)的所有排列的最大前缀和的和。\(1\len\le20\)。Solution考虑到\(n\)很小的性质,想到状压。先考虑一手,最大前缀和应该满足什么条件......
  • Nt、Ki、Zw等前缀含义
    搜索到如下回答。 https://stackoverflow.com/questions/4770553/windows-native-api-when-and-why-use-zw-vs-nt-prefixed-api-callsWindows本机系统服务例程的名称......
  • 【XSY3333】魔力(差分,哈希)
    ”每个字符出现次数相等“可以使用差分简化:记录数组\(s_c\)为字符\(c\)的出现次数减去字符\(\texttt{"a"}\)的出现次数,那么条件等价于\(s\)数组全为\(0\)。维护......
  • 【Rényi差分隐私和零集中差分隐私(差分隐私变体)代码实现】差分隐私代码实现系列(九) 粥
    https://blog.csdn.net/qq_41691212/article/details/122515022  【Rényi差分隐私和零集中差分隐私(差分隐私变体)代码实现】差分隐私代码实现系列(九)粥粥粥少女的拧发......