树上求和 牛客

已于 2024-03-12 19:55:21 修改
阅读量277 收藏 5
点赞数 7
文章标签： c++ 算法 数据结构 dfs
于 2024-03-12 19:54:31 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/theskyinblue/article/details/136661224
版权

题目:
给你一棵根为1的有N个节点的树，以及Q次操作。
每次操作诸如:
1 x y：将节点x所在的子树的所有节点的权值加上y
2 x：询问x所在子树的所有节点的权值的平方和，答案模23333后输出

输入：

第一行两个整数N,Q
第二行N个整数，第i个表示节点i的初始权值
接下来N-1行每行两个整数u,v，表示u和v之间存在一条树边
接下来Q行每行一个操作，格式如题目描述

输出描述:
对于每个询问操作，输出一行一个整数，表示答案在模23333后的结果

分析:
很模版的一道题有很多方法可以写，这里我用的是树链剖分+线段树。两编dfs，第一遍获得子树大小,每个节点父亲，深度。
第二遍获得dfn序，重链top（不需要主要是因为在练板子所以写了）以及入栈时间(tmp)所对应的点，相当于逆映射吧。

这题不是很难，只是我写的有很多问题记录一下。取余运算没有每个都写导致爆了long long,其实改一下就行，我比较懒就开了int128,然后修改和查询操作因为我没有记录每个点出去的时间，我是用的一个siz数组存子树大小，在区间修改个查询的时候应该是dfn[x]-----dfn[x]+siz[x]-1，一开始没有减一导致答案错误,还是记录出去的位置更方便

模运算没有处理好的代码

#include<bits/stdc++.h> 
using namespace std;
using i64 = long long;
using i128 = __int128;
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);

const int mod = 23333;
const int maxn = 1e5+10;
int n,q;
int a[maxn];
vector<int>g[maxn];

int fa[maxn],siz[maxn],dep[maxn],son[maxn];
void dfs1(int x,int f){
	fa[x]=f;
	siz[x]=1;
	dep[x]=dep[f]+1;
	for(auto y:g[x]){
		if(y==f) continue;
		dfs1(y,x);
		siz[x]+=siz[y];
		if(siz[y]>siz[son[x]]) son[x]=y;
	}
}

int tmp=0;
int dfn[maxn],ff[maxn],top[maxn];
void dfs2(int x,int f,int tp){
	dfn[x]=++tmp;
	ff[tmp]=x;
	top[x]=tp;
	if(son[x]!=0) dfs2(son[x],x,tp);
	for(auto y:g[x]){
		if(y==f||y==son[x]) continue;
		dfs2(y,x,y);
	}
}

struct ty{
	i64 sum,squared;
}tree[maxn<<2];
int lazy[maxn];

void pushup(int p,int l,int r){
	tree[p].sum = (tree[p*2].sum+tree[p*2+1].sum)%mod;
	tree[p].squared=(tree[p*2].squared+tree[p*2+1].squared)%mod;
}
void build(int p,int l,int r){
	if(l==r){
		tree[p].sum=a[ff[l]]%mod;
		tree[p].squared=(a[ff[l]]%mod*a[ff[l]]%mod)%mod;
		return;
	}
	int mid = (l+r)>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	pushup(p,l,r);
}

void pushdown(int p,int l,int r){
	int num = lazy[p]%mod;
	int mid = (l+r)>>1;
	lazy[p*2]+=num;
	lazy[p*2+1]+=num;
	tree[p*2].squared+=(tree[p*2].sum%mod*2LL*num%mod+(mid-l+1)*num%mod*num%mod)%mod;
	tree[p*2].sum+=(mid-l+1)*num%mod;
	tree[p*2+1].squared+=(tree[p*2+1].sum%mod*2LL*num%mod+(r-mid)*num%mod*num%mod)%mod;
	tree[p*2+1].sum+=(r-mid)*num%mod;
	lazy[p]=0;
}

void change(int p,int l,int r,int x,int y,int num){
	if(x<=l&&r<=y){
		lazy[p]+=num%mod;
		tree[p].squared+=(2LL*num%mod*tree[p].sum%mod+num%mod*num%mod*(r-l+1)%mod)%mod;
		tree[p].sum+=(r-l+1)*num%mod;
		return;
	}
	if(lazy[p]!=0) pushdown(p,l,r);
	int mid = (l+r)>>1;
	if(x<=mid) change(p*2,l,mid,x,y,num);
	if(y>mid) change(p*2+1,mid+1,r,x,y,num);
	pushup(p,l,r);
}

i64 query(int p,int l,int r,int x,int y){
	if(x<=l&&r<=y) return tree[p].squared%mod;
	int mid =(l+r)>>1;
	if(lazy[p]!=0)pushdown(p,l,r);
	i64 ans=0;
	if(x<=mid) ans+=query(p*2,l,mid,x,y)%mod;
	if(y>mid) ans+=query(p*2+1,mid+1,r,x,y)%mod;
	return ans;
}


int main(){
    ios;
    cin>>n>>q;
    for(int i = 1;i<=n;++i)cin>>a[i];
    for(int i =1;i<n;++i){
    	int x,y;cin>>x>>y;
    	g[x].push_back(y);
    	g[y].push_back(x);
    }
    dfs1(1,0);
    dfs2(1,0,1);
    build(1,1,n);
    for(int i = 1;i<=q;++i){
    	int op;
    	cin>>op;
    	if(op==1){
    		int x,y;
    		cin>>x>>y;
    		change(1,1,n,dfn[x],dfn[x]+siz[x]-1,y);
    	}else{
    		int x;cin>>x;
    		cout<<query(1,1,n,dfn[x],dfn[x]+siz[x]-1)%mod<<"\n";
    	}
    }
    return 0;

AC的代码(写的属实丑陋)

#include<bits/stdc++.h>
using namespace std;
using i64 = long long;
using i128 = __int128;
#define int __int128
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);

template <typename T>
inline T read() {
    T sum = 0, fl = 1;
    char ch = getchar();
    for (; !isdigit(ch); ch = getchar()) 
        if (ch == '-') fl = -1;
    for (; isdigit(ch); ch = getchar()) 
        sum = sum * 10 + (ch - '0'); // 修改此处
    return sum * fl;
}


template <typename T>
inline void write(T x) { // 修改返回类型为 void
    static int sta[45];
    int cnt = 0;
    if (x < 0) putchar('-'), x = -x; // 处理负数
    do {
        sta[cnt++] = x % 10, x /= 10;
    } while (x);
    while (cnt) putchar(sta[--cnt] + '0');
}


const int mod = 23333;
const int maxn = 1e6+10;
int n,q;
int a[maxn];
vector<int>g[maxn];

int fa[maxn],siz[maxn],dep[maxn],son[maxn];
void dfs1(int x,int f){
	fa[x]=f;
	siz[x]=1;
	dep[x]=dep[f]+1;
	for(auto y:g[x]){
		if(y==f) continue;
		dfs1(y,x);
		siz[x]+=siz[y];
		if(siz[y]>siz[son[x]]) son[x]=y;
	}
}

int tmp=0;
int dfn[maxn],ff[maxn],top[maxn];
void dfs2(int x,int f,int tp){
	dfn[x]=++tmp;
	ff[tmp]=x;
	top[x]=tp;
	if(son[x]!=0) dfs2(son[x],x,tp);
	for(auto y:g[x]){
		if(y==f||y==son[x]) continue;
		dfs2(y,x,y);
	}
}

struct ty{
	i64 sum,squared;
}tree[maxn<<2];
int lazy[maxn];

void pushup(int p,int l,int r){
	tree[p].sum = (tree[p*2].sum+tree[p*2+1].sum)%mod;
	tree[p].squared=(tree[p*2].squared+tree[p*2+1].squared)%mod;
}
void build(int p,int l,int r){
	if(l==r){
		tree[p].sum=a[ff[l]]%mod;
		tree[p].squared=(a[ff[l]]%mod*a[ff[l]]%mod)%mod;
		return;
	}
	int mid = (l+r)>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	pushup(p,l,r);
}

void pushdown(int p,int l,int r){
	int num = lazy[p]%mod;
	int mid = (l+r)>>1;
	lazy[p*2]+=num;
	lazy[p*2+1]+=num;
	tree[p*2].squared+=(tree[p*2].sum%mod*2LL*num%mod+(mid-l+1)*num%mod*num%mod)%mod;
	tree[p*2].sum+=(mid-l+1)*num%mod;
	tree[p*2+1].squared+=(tree[p*2+1].sum%mod*2LL*num%mod+(r-mid)*num%mod*num%mod)%mod;
	tree[p*2+1].sum+=(r-mid)*num%mod;
	lazy[p]=0;
}

void change(int p,int l,int r,int x,int y,int num){
	if(x<=l&&r<=y){
		lazy[p]+=num%mod;
		tree[p].squared+=(2LL*num%mod*tree[p].sum%mod+num%mod*num%mod*(r-l+1)%mod)%mod;
		tree[p].sum+=(r-l+1)*num%mod;
		return;
	}
	if(lazy[p]!=0) pushdown(p,l,r);
	int mid = (l+r)>>1;
	if(x<=mid) change(p*2,l,mid,x,y,num);
	if(y>mid) change(p*2+1,mid+1,r,x,y,num);
	pushup(p,l,r);
}

i64 query(int p,int l,int r,int x,int y){
	if(x<=l&&r<=y) return tree[p].squared%mod;
	int mid =(l+r)>>1;
	if(lazy[p]!=0)pushdown(p,l,r);
	i64 ans=0;
	if(x<=mid) ans+=query(p*2,l,mid,x,y)%mod;
	if(y>mid) ans+=query(p*2+1,mid+1,r,x,y)%mod;
	return ans;
}


signed main(){
    n = read<int>();
    q = read<int>();
    for(int i = 1;i<=n;++i)a[i] = read<int>();;
    for(int i =1;i<n;++i){
    	int x,y;
        x = read<int>();y = read<int>();
    	g[x].push_back(y);
    	g[y].push_back(x);
    }
    dfs1(1,0);
    dfs2(1,0,1);
    build(1,1,n);
    for(int i = 1;i<=q;++i){
    	int op;
    	op = read<int>();
    	if(op==1){
    		int x,y;
    		 x= read<int>();y = read<int>();
    		change(1,1,n,dfn[x],dfn[x]+siz[x]-1,y);
    	}else{
    		int x;x = read<int>();
    		write<int>(query(1,1,n,dfn[x],dfn[x]+siz[x]-1)%mod);
            printf("\n");
    	}
    }
    return 0;
}
Assassin's Creed
关注
  • 7
    点赞
  • 踩
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
树上求和【牛客小白月赛9-D】【树链剖分】
既然弱小，就只顾变强就是了
11-28 214
题目链接 还不懂树链剖分的小伙伴可以看这里，以及我的学习笔记都在里面   一道很基本的树链剖分，但是要推一个关于处理平方和的公式：X1^2+X2^2+......+Xn^2；有(x+y)^2=X^2+Y^2+2xy，那么多几个x、y也是成立的：(X1+X2+X3+......+Xn+Y)=X1^2+X2^2+......+Xn^2+2*(X1+X2+X3+......+Xn)*Y+Y^2。 ...
牛客练习赛----树上求和
weixin_43731005的博客
06-10 273
题目描述 有一棵包含n个节点和n-1条边的树，规定树链(u,v)为树上从u到v的简单路径。 树的每条边上都有一个正整数，这个正整数被称作这条边的颜色，规定一条树链的权值w(u,v)为这条树链上所有边的颜色的代数和。 而整棵树的权值为所有不同的树链的权值的代数和。 已知所有边的颜色集合恰好为1到n-1这n-1个不同的正整数，请你为每条边安排一种颜色，使得这棵树的权值尽量小，你不需要给出具体方案，只需要求出这个最小的权值即可。 输入描述: 测试数据第一行，是一个正整数n(1≤n≤10^5)，表示树的节点个数 接
牛客月赛9-树上求和-(dfs序+线段树)
m0_52398496的博客
03-05 187
D 题意： 就是给你一个以1为跟节点的树，m次操作，一种操作是让x所在的子树所有节点的权值都加上y，另一种是查询x所在子树所有节点的权值平方和。同时对23333取模。 思考： 一看是树上操作区间就知道是dfs序了，然后权值的平方和考过很多次了，就是手推一下公式就看出来了。(a+c)(a+c)+(b+c)(b+c)==aa+bb+2*(a+b)+2cc。所以还要维护一下区间和就行了。值得注意的呢是，如何把树的区间转化为线段树的区间，其实只要按每个节点的左边进行记录记录编号就行了。好比线段树的第一个区间是1到n
刷题记录:牛客NC19428树上求和 树链剖分+区间平方和
yingjiayu12的博客
03-06 104
刷题记录:牛客NC19428树上求和 树链剖分+区间平方和
牛客. 青蛙树1
08-03
青蛙树，也称为小根堆问题，是一类在数组中寻找...综上所述，青蛙树问题主要涉及到数组、动态规划、递归、卡特兰数、单调栈和线段树等数据结构和算法知识。通过巧妙地利用这些工具，可以有效地解决小根堆的计数问题。
牛客 – 王国(虚树+树的直径)
01-03
题目分析：真的很巧，上周刚学的虚树，读完这个题的第一反应就是可以用虚树简化题目，其实完全可以用树形dp跑一遍dfs出来，可奈何我dp不好，就只能用虚树来做了 题目中有两个点可以单独考虑，首先是权值相同的点，这...
牛客刷题总结.docx
01-28
"牛客刷题总结" 本资源摘要信息主要涵盖了 Java 语言的基础知识点、设计模式和事务处理。下面是详细的知识点解释： 一、Java 基础知识 1. Java 源程序只能有一个 public 类存在，且类名与文件名相同。这是因为 ...
牛客校招C++宝典.rar
03-17
《牛客校招C++宝典.rar》是一个针对C++编程语言在校招面试中的核心知识点进行整理的资源包。这份宝典涵盖了多个关键领域，包括计算机网络、算法、C++基础知识以及数据库知识，旨在帮助应聘者全面准备技术面试。下面...
关于牛客刷题总结全部汇总
02-21
关于牛客刷题总结全部汇总 本资源摘要信息主要讲述了 Java 语言的基本概念、设计模式、MVC 模式、数学函数、重载和重写的区别、多态、数组、静态成员和方法、事务属性等知识点。 一、Java 基本概念 1. Java 源...
(C++二叉树02) 翻转二叉树 对称二叉树 二叉树的深度
qq_44962949的博客
07-17 214
【代码】(C++二叉树02) 翻转二叉树 对称二叉树 二叉树的深度。
学懂C语言系列（三）：C语言基本语法
martian665的专栏
07-17 557
学懂C语言系列,C语言基本语法
C++ std::thread 管理线程生命周期：join 和 detach
m0_59680769的博客
07-16 1002
方法用于同步主线程和子线程。也就是说，当主线程调用子线程的join方法时，主线程会被阻塞，直到子线程完成其执行。这意味着join方法确保主线程等待子线程结束，然后继续执行后续代码。t.join()方法则是让子线程在后台独立运行，与主线程分离。分离后的线程（也称为“游离线程”）不能再被join，它的资源会在该线程完成后自动释放。主线程会继续执行而不会等待该子线程。如果把std::this_thread::sleep_for(std::chrono::seconds(3));
【C++】获取环境变量
Badme
07-18 4573
【C++】获取环境变量
C语言强化-1.数据结构概述
Annabelle___的博客
07-18 629
与408的关联：1. 逻辑结构和存储结构在选择题中会有涉及。2. 时间复杂度几乎是每一年大题必考内容！
【素养大赛c++复赛题-序列排序题解】图像模糊处理、吉利号码、战胜白蚁题解
qq_56949201的博客
07-16 199
第一行1个整数n,表示序列的大小。( 0 < n ≤ 1000) 第二行n个正整数，表示序列的每个数，每个数不大于100000000。对于给定的正整数序列，按照每个数的各位数和从大到小排序，各位数和相同的按照本身大小排序，大的在前，小的在后。来存储输入的数字序列，排序完成后输出序列，并在最后释放了动态分配的内存。输出按照题目要求排序后的序列。也可以使用动态分配的数组。
Windows 点云生成二维栅格图 [附C++完整代码实现]
weixin_43896283的博客
07-17 602
三维点云数据转换为二维栅格图
C语言中的指针函数和函数指针
letterljhx的博客
07-18 434
C语言中的指针函数和函数指针
【C语言】逗号运算符详解 - 《不起眼的 “逗号”》
最新发布
EleganceJiaBao的博客
07-19 659
逗号运算符是C语言中一个非常实用的运算符，但也常常被初学者忽视。本文将详细介绍逗号运算符的定义、用法和应用场景，通过示例和代码解释其工作原理。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值