题面

传送门

思路

本来以为这道题可以LCT维护子树信息直接做的,后来发现这样会因为splay形态改变影响子树权值平方和,是splay本身的局限性导致的

所以只能另辟蹊径

首先,我们考虑询问点都在1的情况

考虑一次修改带来的影响:

假设当前节点的值变动量为$delta$,修改节点为$u$

那么对于所有位于路径$(1,u)$上的节点而言,它们的子树和以及子树平方和都会有改变

设$sum(u)$表示子树点权和,$sumsqr(u)$表示点权和的平方

那么$\forall v \in (1,u)$,$sum(v)+=delta$,$sumsqr(v)+=delta\ast 2\ast sum(v)+delta\ast delta$

又可见$ans=\sum_{u} sumsqr(u)$

那么$ans$在这次事件中的变化量可以表示如下:

设$len$为路径$(1,u)$的长度

那么$ans+=len\ast delta\ast delta+delta\ast 2\ast \sum_{v \in (1,u)} sum(v)$

所以,我们可以使用一棵线段树维护$sum(u)$的值,树链剖分一下,只需要支持区间修改和区间查询

接下来考虑询问点在$x$的情况

可以发现,如果我们考虑$ans(1)$到$ans(x)$中各个位置的贡献,容易发现,依然只有$v\in (1,x)$的节点贡献改变了

设$a_i$表示节点$i$在以1为根(也就是我们的树剖维护的东西)时的子树点权和,$b_i$表示以$x$为根的时候的点权和,$sum$为总点权和

可以得到:

$ans(x)=ans(1)-\sum_{v\in (1,x)} a_v^2 +\sum_{v\in (1,x)} b_v^2$

同时,$a_i$和$b_i$有一个性质:

$a_1=b_x=sum=a_v+b_{fa(v)}$,其中$v \neq 1,x$

那么化简上面式子

$ans(x)=ans(1)-\sum_{v\in (1,x),v\neq 1}a_v^2 + sum_{v\in(1,x),v\neq x}b_v^2$

为了方便,我们用$a_i$表示路径上的第$i$个点,$b_i$同理

$ans(x)=ans(1)-\sum_{i=2}{len}a_i2 + sum_{i=2}{len}(sum-a_i)2$

$ans(x)=ans(1)+(len-1)\ast (len-1)\ast sum-2\ast (len-1)ast \sum_{i=2}^{len}a_i$

这样就可以算了,依然是树链剖分+线段树解决

注意最后面这个式子,如果为了方便直接查询到根,可以把$len-1$提出来,并把后面的$i=2...len$变成$1...len$,在前面把一个$len-1$变成$len+1$,即:

$ans(x)=ans(1)+(len-1)\ast ((len+1)\ast sum-2\ast \sum_{i=1}^{len}a_i)$

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cassert>
#define ll long long
using namespace std;
inline int read(){
int re=0,flag=1;char ch=getchar();
while(!isdigit(ch)){
if(ch=='-') flag=-1;
ch=getchar();
}
while(isdigit(ch)) re=(re<<1)+(re<<3)+ch-'0',ch=getchar();
return re*flag;
}
int n,first[200010],dep[200010],siz[200010],son[200010],top[200010],pos[200010],back[200010],fa[200010],clk,cnte;
struct edge{
int to,next;
}a[400010];
ll w[200010],s[200010],ans1;
inline void adde(const int u,const int v){
a[++cnte]=(edge){v,first[u]};first[u]=cnte;
a[++cnte]=(edge){u,first[v]};first[v]=cnte;
}
void dfs1(const int u,const int f){
int i,v,maxn=0;
dep[u]=dep[f]+1;fa[u]=f;
siz[u]=1;son[u]=0;s[u]=w[u];
for(i=first[u];~i;i=a[i].next){
v=a[i].to;if(v==f) continue;
dfs1(v,u);
siz[u]+=siz[v];s[u]+=s[v];
if(maxn<siz[v]) son[u]=v,maxn=siz[v];
}
}
void dfs2(const int u,const int t){
int i,v;
pos[u]=++clk;back[clk]=u;top[u]=t;
if(son[u]) dfs2(son[u],t);
for(i=first[u];~i;i=a[i].next){
v=a[i].to;if(v==fa[u]||v==son[u]) continue;
dfs2(v,v);
}
}
ll len[800010],sum[800010],lazy[800010];
void update(int num){
sum[num]=sum[num<<1|1]+sum[num<<1];
}
void push(int l,int r,int num){
if(l==r||!lazy[num]) return ;
int mid=(l+r)>>1;
sum[num<<1]+=(ll)(mid-l+1)*lazy[num];
sum[num<<1|1]+=(ll)(r-mid)*lazy[num];
lazy[num<<1]+=lazy[num];
lazy[num<<1|1]+=lazy[num];
lazy[num]=0;
}
int ql,qr;ll val;
void build(const int l,const int r,const int num){
if(l==r){sum[num]=s[back[l]];return;}
const int mid=(l+r)>>1;
build(l,mid,num<<1);build(mid+1,r,num<<1|1);
update(num);
}
void change(const int l,const int r,const int num){
if(l>=ql&&r<=qr){
sum[num]+=(r-l+1)*val;
lazy[num]+=val;
return;
}
push(l,r,num);
const int mid=(l+r)>>1;
if(mid>=ql) change(l,mid,num<<1);
if(mid<qr) change(mid+1,r,num<<1|1);
update(num);
}
ll query(const int l,const int r,const int num){
if(l>=ql&&r<=qr) return sum[num];
push(l,r,num);
const int mid=(l+r)>>1;ll re=0;
if(mid>=ql) re+=query(l,mid,num<<1);
if(mid<qr) re+=query(mid+1,r,num<<1|1);
return re;
}
void add(int u,const int v){
int f;val=v;
while(u){
f=top[u];
ql=pos[f];qr=pos[u];
change(1,n,1);
u=fa[f];
}
}
ll ask(int u){
int f;ll re=0;
while(u){
f=top[u];
ql=pos[f];qr=pos[u];
re+=query(1,n,1);
u=fa[f];
}
return re;
}
int main(){
n=read();int Q=read(),i,t1,t2,t3;ll s1;
memset(first,-1,sizeof(first));
for(i=1;i<n;i++){
t1=read();t2=read();
adde(t1,t2);
}
for(i=1;i<=n;i++) w[i]=read();
dfs1(1,0);dfs2(1,0);build(1,n,1);
for(i=1;i<=n;i++) ans1+=s[i]*s[i];
while(Q--){
t3=read();t1=read();
if(t3==1){
t2=read();
t2=t2-w[t1];w[t1]+=t2;
ans1+=t2*t2*dep[t1]+t2*2*ask(t1);
add(t1,t2);
}
else{
ql=qr=1;s1=query(1,n,1);
printf("%lld\n",ans1+s1*((dep[t1]+1)*s1-2*ask(t1)));
}
}
}

[luogu3676] 小清新数据结构题 [树链剖分+线段树]的更多相关文章

  1. 【bzoj4127】Abs 树链剖分+线段树

    题目描述 给定一棵树,设计数据结构支持以下操作 1 u v d 表示将路径 (u,v) 加d 2 u v 表示询问路径 (u,v) 上点权绝对值的和 输入 第一行两个整数n和m,表示结点个数和操作数 ...

  2. 【Codeforces827D/CF827D】Best Edge Weight(最小生成树性质+倍增/树链剖分+线段树)

    题目 Codeforces827D 分析 倍增神题--(感谢T*C神犇给我讲qwq) 这道题需要考虑最小生成树的性质.首先随便求出一棵最小生成树,把树边和非树边分开处理. 首先,对于非树边\((u,v ...

  3. 【BZOJ-2325】道馆之战 树链剖分 + 线段树

    2325: [ZJOI2011]道馆之战 Time Limit: 40 Sec  Memory Limit: 256 MBSubmit: 1153  Solved: 421[Submit][Statu ...

  4. 【BZOJ2243】[SDOI2011]染色 树链剖分+线段树

    [BZOJ2243][SDOI2011]染色 Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的 ...

  5. POJ3237 (树链剖分+线段树)

    Problem Tree (POJ3237) 题目大意 给定一颗树,有边权. 要求支持三种操作: 操作一:更改某条边的权值. 操作二:将某条路径上的边权取反. 操作三:询问某条路径上的最大权值. 解题 ...

  6. bzoj4034 (树链剖分+线段树)

    Problem T2 (bzoj4034 HAOI2015) 题目大意 给定一颗树,1为根节点,要求支持三种操作. 操作 1 :把某个节点 x 的点权增加 a . 操作 2 :把某个节点 x 为根的子 ...

  7. 【POJ3237】Tree(树链剖分+线段树)

    Description You are given a tree with N nodes. The tree’s nodes are numbered 1 through N and its edg ...

  8. B20J_2836_魔法树_树链剖分+线段树

    B20J_2836_魔法树_树链剖分+线段树 题意: 果树共有N个节点,其中节点0是根节点,每个节点u的父亲记为fa[u].初始时,这个果树的每个节点上都没有果子(即0个果子). Add u v d ...

  9. B20J_2243_[SDOI2011]染色_树链剖分+线段树

    B20J_2243_[SDOI2011]染色_树链剖分+线段树 一下午净调这题了,争取晚上多做几道. 题意: 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成 ...

随机推荐

  1. 关于使用js下载图片

    使用js进行图片下载是很常见的需求,但是解决起来却不是那么顺利. 服务器返回了一个图片地址,网上一搜基本都是用a标签的download属性,但是兼容性实在是很差.这里推荐使用blob. 上代码: va ...

  2. mac安装mysql及workbench

    安装mysql 登录MySQL网站 打开网站Download MySQL Community Server,选择下方的dmg文件下载 点击download,此处为8.0.11版本 然后选择no tha ...

  3. struts2入门第一天----------配置环境

    放假之后有空就开始走上了三大框架的学习.第一个选择的框架是struts2.首先第一步当然是环境的配置.去apache官网把struts2下载下来.然后在自己的开发工具下创建一个web项目.在lib文件 ...

  4. hibernate映射实体类查询时数据库空字段赋值给实体类报错的问题

    因为一直报实体类空异常,去网上查了资料只查到了并没有查到数据库空值时不给实体类赋值的属性 异常 org.hibernate.InvalidMappingException: Could not par ...

  5. linux学习(1)——这是一个新的开始,加油吧少年

     (一)自己简单总结 学会使用简单命令 Tab:实现自动补全功能 Ctrl+D:退出当前终端 Ctrl+Z:暂停当前进程 Ctrl+L:清屏 Ctrl+A:可以让光标移动到最前列 Ctrl+E:可以让 ...

  6. 硬盘安装Windows Server 2008(解决系统盘符变成D盘)

    硬盘安装Windows 2008系统方法 操作系统最好用的无疑是server 2003,但是现在Server 2003支持的软件越来越少,很多是故意不支持Server 2003了, 像php5.5以上 ...

  7. C++ 二叉搜索树

    二叉搜索树利用其特有的二叉树性质,使其搜索更方便 源代码: struct node { int val; node *left, *right; }; //the function of insert ...

  8. poj 3111 卖珠宝问题 最大化平均值

    题意:有N件分别价值v重量w的珠宝,希望保留k件使得 s=v的和/w的和最大 思路:找到贡献最大的 设当前的s为mid(x) 那么贡献就是 v-w*x 排序 ,取前k个 bool operator&l ...

  9. 笔记-scrapy-去重

    笔记-scrapy-去重 1.      scrapy 去重 scrapy 版本:1.5.0 第一步是要找到去重的代码,scrapy在请求入列前去重,具体源码在scheduler.py: def en ...

  10. P1498 南蛮图腾

    P1498 南蛮图腾 题目描述 自从到了南蛮之地,孔明不仅把孟获收拾的服服帖帖,而且还发现了不少少数民族的智慧,他发现少数民族的图腾往往有着一种分形的效果(看Hint),在得到了酋长的传授后,孔明掌握 ...