CodeForces - 1485E Move and Swap(树形dp)

题目链接：点击查看

题目大意：给出一棵有根树，每个节点都有权值，满足所有叶子结点到根节点的距离相同，初始时在根节点有两个硬币，分别是红色和蓝色，每次可以执行如下操作：

将红色硬币移动到任意一个子节点
将蓝色硬币移动到下一层的任意一个节点
（可选）交换红色硬币和蓝色硬币
本次操作的贡献是红色硬币和蓝色硬币点权之差的绝对值

问如何移动可以使得贡献之和最大

题目分析：不算难的一道树形 dpdpdp，因为蓝色硬币可以是下一层的任意节点，所以只需要关注红色硬币移动的路线就可以了，不妨设 dpidp_idpi 为红色硬币在点 iii 时移动到叶子结点的最大贡献和，选择自底向上转移，这样就将原树中的节点划分成不同层次之间的转移了。

分成两种情况讨论：

不交换硬币：那么 dpu=dpv+∣av−am∣dp_u=dp_v+|a_v-a_m|dpu=dpv+∣av−am∣，因为不交换的话，红色硬币下一步一定到达点 uuu 的子节点 vvv，而 mmm 是蓝色节点的位置，我们可以选择同层中任意节点来充当
交换硬币：那么 dpu=dpm+∣av−am∣dp_u=dp_m+|a_v-a_m|dpu=dpm+∣av−am∣，类比于上面的情况，蓝色节点一定是 vvv 了，而同层的任意一个节点可以充当红色节点

乍一看转移的复杂度不是很可观，但是这些东西是可以预处理的，第一种情况我们只需要维护一下 ama_mam 的最小值和最大值辅助转移即可，第二种情况需要预处理一下 dpm−amdp_m-a_mdpm−am 和 dpm+amdp_m+a_mdpm+am 的最大值辅助转移就好啦

代码：

// Problem: E. Move and Swap
// Contest: Codeforces - Codeforces Round #701 (Div. 2)
// URL: https://codeforces.com/contest/1485/problem/E
// Memory Limit: 256 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)// #pragma GCC optimize(2)
// #pragma GCC optimize("Ofast","inline","-ffast-math")
// #pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<list>
#include<unordered_map>
#define lowbit(x) x&-x
using namespace std;
typedef long long LL;
typedef unsigned long long ull;
template<typename T>
inline void read(T &x)
{T f=1;x=0;char ch=getchar();while(0==isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}while(0!=isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();x*=f;
}
template<typename T>
inline void write(T x)
{if(x<0){x=~(x-1);putchar('-');}if(x>9)write(x/10);putchar(x%10+'0');
}
const LL inf=0x3f3f3f3f3f3f3f3f;
const int N=1e6+100;
int fa[N],max_deep;
LL dp[N],a[N];
vector<int>node[N],d[N];
void dfs(int u,int f,int dep) {max_deep=max(max_deep,dep);fa[u]=f;d[dep].push_back(u);for(auto v:node[u]) {dfs(v,u,dep+1);}
}
void init(int n) {for(int i=1;i<=n;i++) {node[i].clear();d[i].clear();}memset(dp,0,sizeof(LL)*(n+5));max_deep=0;
}
int main()
{#ifndef ONLINE_JUDGE
//  freopen("data.in.txt","r",stdin);
//  freopen("data.out.txt","w",stdout);
#endif
//  ios::sync_with_stdio(false);int w;cin>>w;while(w--) {int n;read(n);init(n);for(int i=2;i<=n;i++) {int fa;read(fa);node[fa].push_back(i);}for(int i=2;i<=n;i++) {read(a[i]);}dfs(1,-1,1);for(int i=max_deep;i>1;i--) {//不交换LL mmax=-inf,mmin=inf;for(auto u:d[i]) {mmax=max(mmax,a[u]);mmin=min(mmin,a[u]);}for(auto u:d[i]) {dp[fa[u]]=max(dp[fa[u]],dp[u]+max(llabs(a[u]-mmax),llabs(a[u]-mmin)));}//交换LL mmax1=-inf,mmax2=-inf;for(auto u:d[i]) {mmax1=max(mmax1,dp[u]+a[u]);mmax2=max(mmax2,dp[u]-a[u]);}for(auto u:d[i]) {dp[fa[u]]=max(dp[fa[u]],mmax1-a[u]);dp[fa[u]]=max(dp[fa[u]],mmax2+a[u]);}}cout<<dp[1]<<endl;}return 0;
}

CodeForces - 1485E Move and Swap(树形dp)相关推荐

CodeForces - 1324F Maximum White Subtree(树形dp)
题目链接:点击查看题目大意:给出 n 个点组成的树,每个点都有一个颜色,非黑即白,现在问对于每个点而言,选出一个连通块,使得白色点的个数与黑色点的个数做差最大题目分析:记录一下div3的第一次ak ...
codeforces1485 E. Move and Swap（dp）
E. Move and Swap Heltion 由于红色硬币向下一层走的时候只能走儿子,而蓝色无限制(对后续操作无影响),于是考虑下面表示状态表示:fuf_ufu表示当前是红色硬币,向下一层走后 ...
Codeforces Round #701 (Div. 2) E. Move and Swap 思维 + dp
传送门文章目录题意: 思路: 题意: 思路: 由于是按层来的,所以我们肯定先按照层来分组. 定义dp[i]dp[i]dp[i]为红棋在位置iii的时候的最大得分和. 先考虑不换的情况,我们对于每个 ...
Codeforces 1276D/1259G Tree Elimination (树形DP)
题目链接 http://codeforces.com/contest/1276/problem/D 题解我什么DP都不会做,吃枣药丸-- 设\(f_{u,j}\)表示\(u\)子树内,\(j=0\) ...
E. Xenon's Attack on the Gangs,Codeforces Round #614 (Div. 2),树形dp
E. Xenon's Attack on the Gangs http://codeforces.com/contest/1293/problem/E On another floor of the ...
【CodeForces - 697D】Puzzles（树形dp，期望dp）
题目大意: 给定一棵树,从1开始,按DFS的方式访问这棵树每次从父亲节点随机访问儿子,问每个节点被访问到的时间的期望输入:第一行一个数n,代表n个节点.第二行n-1个数p2,p3,p4,p5.. ...
Codeforces Round #263 (Div. 2) D. Appleman and Tree 树形dp
链接: http://codeforces.com/contest/462/problem/D 题意: 给定n个点的树, 0为根,下面n-1行表示每个点的父节点最后一行n个数表示每个点的颜色,0为 ...
codeforces 816 E. Karen and Supermarket（树形dp）
题目链接:http://codeforces.com/contest/816/problem/E 题意:有n件商品,每件有价格ci,优惠券di,对于i>=2,使用di的条件为:xi的优惠券需要被 ...
【Codeforces Round #614（div2）】E-Xenon's Attack on the Gangs（树形dp）
一.题目链接 https://codeforces.com/contest/1293/problem/E 二.题意给n个结点,n-1条无向边.即一棵树.我们需要给这n-1条边赋上0~n-2不重复的值 ...

CodeForces - 1485E Move and Swap(树形dp)

CodeForces - 1485E Move and Swap(树形dp)相关推荐

最新文章

热门文章