[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
#include
#include
#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 ij?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].xdep[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

 

你可能感兴趣的:(dp)