[cf1485e]Move and Swap(dp)

看起来很像树形dp，如果没有交换操作的话，显然有dp[u]=max(dp[v])+max(abs(a1[u']-a1[u]))其中v为u的儿子，u'与u处于同一深度。

然后考虑交换操作可能会怎样影响结果。假如是u'和u交换，用于更新dp[u]，那么计算出的结果应该是dp[u]=max(dp[v'])+abs(a1[u']-a1[u])。

这样思路就有了，连边后首先做一次深搜，记录每个节点的深度，每层的最大值和最小值。再将节点按深度从大到小排序做，因为由上面第二个方程可以看出，我们不仅需要每个节点最大儿子，还需要和它同层的节点最大的儿子。第一种更新逐点做就可以。对于第二种，是要在同层内记录最大的m1=maxson[u]+a1[u]和m2=maxson[u]-a1[u]，然后这一层都统计过一遍后，再统一更新一次dp[u]=max(dp[u], m1-a1[u], m2+a1[u])。

大概就这样吧。记得多组数据，每次要初始化。然后记得开long long……哭了

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=200010;
struct edge{
	int y,next;
}data[N*2];
struct node{
	int x,id;
}a1[N];
int t,n,num,h[N],dep[N],d,mn[N],mx[N],fa[N];
ll dp[N],m1,m2,mson[N];
inline int read(){
	int x=0,f=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return f?-x:x;
}
inline ll min1(ll i,ll j){return i<j?i:j;}
inline ll max1(ll i,ll j){return i>j?i:j;}
inline void addedge(int u,int v){
	data[++num].y=v,data[num].next=h[u],h[u]=num;
	data[++num].y=u,data[num].next=h[v],h[v]=num;
}
void dfs1(int u,int f){
	fa[u]=f;
	for(int i=h[u],v;i!=-1;i=data[i].next){
		v=data[i].y;
		if(v==f)continue;
		dep[v]=dep[u]+1;
		d=max1(dep[v],d);
		if(a1[v].x>mx[dep[v]])mx[dep[v]]=a1[v].x;
		if(a1[v].x<mn[dep[v]])mn[dep[v]]=a1[v].x;
		dfs1(v,u);
	}
}
inline bool cmp1(node i,node j){return dep[i.id]>dep[j.id];}
int main(){
	t=read();
	dep[0]=a1[0].id=0;
	for(int i=1;i<=200000;++i)mn[i]=1e9+10,mx[i]=0;
	while(t--){
		n=read();num=0;
		dep[1]=d=1;
		for(int i=1;i<=n;++i)h[i]=-1;
		for(int i=2,x;i<=n;++i){
			x=read();addedge(x,i);
		}
		a1[1].id=1,a1[1].x=0;
		for(int i=2;i<=n;++i){a1[i].id=i;a1[i].x=read();}
		dfs1(1,0);
		sort(a1+1,a1+n+1,cmp1);
		for(int k=1,jj=1,u;k<=n;++k){
			if(dep[a1[k].id]!=dep[a1[k-1].id]){
				for(int j=jj;j<k;++j){
					dp[a1[j].id]=max1(dp[a1[j].id],m1-a1[j].x);
					dp[a1[j].id]=max1(dp[a1[j].id],m2+a1[j].x);
				}
				jj=k;m1=m2=-1e9;
			}
			u=a1[k].id;mson[u]=0;
			for(int i=h[u],v;i!=-1;i=data[i].next){
				v=data[i].y;
				if(v==fa[u])continue;
				mson[u]=max1(mson[u],dp[v]);
			}
			dp[u]=mson[u]+max1(1ll*mx[dep[u]]-a1[k].x,1ll*a1[k].x-mn[dep[u]]);
			m1=max1(mson[u]+a1[k].x,m1);
			m2=max1(mson[u]-a1[k].x,m2);
		}
		printf("%lld\n",dp[1]);
		for(int i=1;i<=d;++i)mn[i]=1e9+10,mx[i]=0;
	}
	return 0;
}