[2021 HDU多校] HDU 6962 I love tree

题目描述：

给定一棵树，然后有两个操作。
第一个操作从U,V这个路径上的点权值依次加上路径次序的平方。
第二个操作是查询点U的权值和。

题目分析：

首先可以树剖把路径分为logn段。
对于dfn从[h,t]这段，可以维护一个二次函数$(x-h{)}^2$
展开后得到$x^2+h^2-2\ast x\ast h$
分别用三个线段树维护每个项的和就可以了
时间复杂度$O(n(logn{)}^2)$

题目链接：

HDU 6962

AC代码：

#include <iostream>
#include <cstdio>
#define int long long
const int maxm=1e6+100; 
int deep[maxm],son[maxm],size[maxm],fa[maxm],top[maxm];
int head[maxm],to[maxm<<1],net[maxm<<1],cnt;
int dfn[maxm],tot,n,m;
struct node{
	int tag[maxm*4];
	inline void col(int o,int num){tag[o]+=num;}
	inline void pushdown(int o)
	{
		if(!tag[o]) return;
		col((o<<1),tag[o]),col((o<<1)|1,tag[o]);
		tag[o]=0;
	}
	void change(int o,int l,int r,int ql,int qr,int num)
	{
		if(ql>r||qr<l) return;
		if(ql<=l&&r<=qr)
		{
			col(o,num);
			return;
		}
		pushdown(o);
		int mid=(l+r)>>1;
		change((o<<1),l,mid,ql,qr,num),change((o<<1)|1,mid+1,r,ql,qr,num);
	}
	int ask(int o,int l,int r,int ind)
	{
		if(l>=r) return tag[o];
		int mid=(l+r)>>1;
		pushdown(o);
		return ind<=mid?ask((o<<1),l,mid,ind):ask((o<<1)|1,mid+1,r,ind);
	}
}st[3];
inline void addedge(int x,int y)
{
    to[++cnt]=y,net[cnt]=head[x],head[x]=cnt;
}
int dfs1(int now,int fax,int dep)
{
    fa[now]=fax,deep[now]=dep,size[now]=1;
    int maxson=-1;
    for(int i=head[now];i;i=net[i])
    if(to[i]!=fax)
    {
        size[now]+=dfs1(to[i],now,dep+1);
        if(maxson<size[to[i]]) son[now]=to[i],maxson=size[to[i]];
    }
    return size[now];
}
void dfs2(int now,int topx)
{
    top[now]=topx,dfn[now]=++tot;
    if(!son[now]) return;
    dfs2(son[now],topx);
    for(int i=head[now];i;i=net[i])
    if(!top[to[i]]) dfs2(to[i],to[i]);
}
inline int lca(int u,int v)
{ 
    while(top[u]!=top[v])
    {
        if(deep[top[u]]<deep[top[v]]) std::swap(u,v);
        u=fa[top[u]];
    }
    return deep[u]<deep[v]?u:v;
}
inline void tree_change(int u,int v,int k)
{
	int p1=1,p2=k;
	while(top[u]!=top[v])
	{
		if(deep[top[u]]>deep[top[v]])
		{
			p1-=deep[top[u]]-deep[u];
			st[0].change(1,1,n,dfn[top[u]],dfn[u],(p1+dfn[top[u]])*(p1+dfn[top[u]]));
			st[1].change(1,1,n,dfn[top[u]],dfn[u],(p1+dfn[top[u]]));
			st[2].change(1,1,n,dfn[top[u]],dfn[u],1);
			u=fa[top[u]];
			p1++;
		}
		else
		{
			p2+=deep[top[v]]-deep[v];
			st[0].change(1,1,n,dfn[top[v]],dfn[v],(-p2+dfn[top[v]])*(-p2+dfn[top[v]]));
			st[1].change(1,1,n,dfn[top[v]],dfn[v],(-p2+dfn[top[v]]));
			st[2].change(1,1,n,dfn[top[v]],dfn[v],1);
			v=fa[top[v]];
			p2--;
		}
	}
	if(deep[u]<=deep[v])
	{
		p2+=deep[u]-deep[v];
		st[0].change(1,1,n,dfn[u],dfn[v],(-p2+dfn[u])*(-p2+dfn[u]));
		st[1].change(1,1,n,dfn[u],dfn[v],(-p2+dfn[u]));
	    st[2].change(1,1,n,dfn[u],dfn[v],1);
	}
	else
	{
		p1-=deep[v]-deep[u];
		st[0].change(1,1,n,dfn[v],dfn[u],(p1+dfn[v])*(p1+dfn[v]));
		st[1].change(1,1,n,dfn[v],dfn[u],(p1+dfn[v]));
	    st[2].change(1,1,n,dfn[v],dfn[u],1);
	}
}
signed main()
{
	scanf("%lld",&n);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%lld%lld",&u,&v);
		addedge(u,v),addedge(v,u);
	}
	dfs1(1,0,1),dfs2(1,1);
	scanf("%lld",&m);
	for(int i=1;i<=m;i++)
	{
		int opt,u,v;
		scanf("%lld",&opt);
		if(opt==1)
		{
			scanf("%lld%lld",&u,&v);
			int x=lca(u,v);
			tree_change(u,v,deep[u]+deep[v]-2*deep[x]+1);
		}
		else
		{
			scanf("%lld",&u);
			printf("%lld\n",st[0].ask(1,1,n,dfn[u])+dfn[u]*dfn[u]*st[2].ask(1,1,n,dfn[u])-2*dfn[u]*st[1].ask(1,1,n,dfn[u]));
		}
	}
	return 0; 
}