E. Alternating Tree 树点分治|树形DP
题意:给你一颗树,然后这颗树有n*n条路径,a->b和b->a算是一条,然后路径的权值是 vi*(-1)^(i+1) 注意是点有权值。
从上头往下考虑是点分治,从下向上考虑就是树形DP,分成三类路径:1.指定点单点 2.跨过指定点3.没跨过指定点但不是单点
细节要好好打磨一下,然后还是用long long比较稳 ,1LL*还是有点危险.
点分治:
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e5+;
const int P=1e9+;
vector<int>G[N];
bool vis[N];
int SQ,SO,Q,O,q[N],o[N],rt,m,nq[N],no[N],sz[N],v[N],mx[N],ans;
void dfs(int u,int fa,int S){
sz[u]=mx[u]=;
q[u]=o[u]=no[u]=nq[u]=;
for(int i=;i<(int)G[u].size();++i) {
int v=G[u][i];
if(vis[v]||v==fa) continue;
dfs(v,u,S);
sz[u]+=sz[v];
mx[u]=max(mx[u],sz[v]);
}
mx[u]=max(mx[u],S-sz[u]);
if(m>mx[u]) rt=u,m=mx[u];
}
void solve(int u,int nv,int f,int fa,int last){
int t=(nv+f*v[u])%P;
if(f==-) o[fa]=(o[fa]+t)%P,O=(O+t)%P,++no[fa],++SO;
else q[fa]=(q[fa]+t)%P,Q=(Q+t)%P,++nq[fa],++SQ;
for(int i=;i<(int)G[u].size();++i) {
int v=G[u][i];
if(v==last||vis[v]) continue;
solve(v,t,-f,fa,u);
}
}
int work(int u,int S){
m=1e9+;
dfs(u,,S);
vis[rt]=;
Q=O=SQ=SO=;
for(int i=;i<(int)G[rt].size();++i) if(!vis[G[rt][i]]) solve(G[rt][i],v[rt],-,G[rt][i],rt);
for(int i=;i<(int)G[rt].size();++i) if(!vis[G[rt][i]]){
int t=G[rt][i];
ans=((ans-1LL*no[t]*(O-o[t])%P)%P-1LL*o[t]*(SO-no[t])%P)%P;
ans=(ans+1LL*no[t]*(SO-no[t])%P*v[rt]%P)%P;
ans=((ans+1LL*nq[t]*(Q-q[t])%P)%P+1LL*q[t]*(SQ-nq[t])%P)%P;
ans=(ans-1LL*nq[t]*(SQ-nq[t])%P*v[rt]%P)%P;
ans=(ans+(q[t]<<)%P)%P;
}
ans=((ans+v[rt])%P+P)%P;
int nt=rt;
for(int i=;i<(int)G[nt].size();++i) if(!vis[G[nt][i]])
work(G[nt][i],sz[G[nt][i]]>sz[nt]?S-sz[nt]:sz[G[nt][i]]);
}
int main(){
int n,x,y;
scanf("%d",&n);
for(int i=;i<=n;++i) scanf("%d",v+i);
for(int i=;i<n;++i) {
scanf("%d%d",&x,&y);
G[x].push_back(y);
G[y].push_back(x);
}
work(,n);
printf("%d\n",(ans+P)%P);
}
树形DP的思路是参考的这里
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e5+;
const int P=1e9+;
vector<int>G[N];
long long dp[N][],num[N][],ans,v[N];
void dfs(int u,int fa){
for(int i=;i<(int)G[u].size();++i) {
int vt=G[u][i];
if(vt==fa) continue;
dfs(vt,u);
dp[u][]=(((dp[u][]+dp[vt][])%P-1LL*v[u]*num[vt][]%P)%P+P)%P;
dp[u][]=(((dp[u][]+dp[vt][])%P+1LL*v[u]*num[vt][]%P)%P+P)%P;
num[u][]=(num[u][]+num[vt][])%P;
num[u][]=(num[u][]+num[vt][])%P;
}
for(int i=;i<(int)G[u].size();++i) {
int vt=G[u][i];
if(vt==fa) continue;
ans=(ans+2LL*(num[u][]-num[vt][])%P*dp[vt][]%P+1LL*(num[u][]-num[vt][])*v[u]%P*num[vt][]%P)%P;
ans=(ans+2LL*(num[u][]-num[vt][])%P*dp[vt][]%P-1LL*(num[u][]-num[vt][])*v[u]%P*num[vt][]%P)%P;
}
ans=(ans+(dp[u][]<<)%P+v[u])%P;
++num[u][];
num[u][]%=P;
dp[u][]=(dp[u][]+v[u])%P;
}
int main(){
int n,x,y;
scanf("%d",&n);
for(int i=;i<=n;++i) scanf("%lld",v+i);
for(int i=;i<n;++i) {
scanf("%d%d",&x,&y);
G[x].push_back(y);
G[y].push_back(x);
}
dfs(,);
printf("%lld\n",(ans+P)%P);
}
E. Alternating Tree 树点分治|树形DP的更多相关文章
- 算法笔记--树的直径 && 树形dp && 虚树 && 树分治 && 树上差分 && 树链剖分
树的直径: 利用了树的直径的一个性质:距某个点最远的叶子节点一定是树的某一条直径的端点. 先从任意一顶点a出发,bfs找到离它最远的一个叶子顶点b,然后再从b出发bfs找到离b最远的顶点c,那么b和c ...
- (中等) HDU 5293 Tree chain problem,树链剖分+树形DP。
Problem Description Coco has a tree, whose vertices are conveniently labeled by 1,2,…,n.There are ...
- [集训队作业2018]蜀道难——TopTree+贪心+树链剖分+链分治+树形DP
题目链接: [集训队作业2018]蜀道难 题目大意:给出一棵$n$个节点的树,要求给每个点赋一个$1\sim n$之内的权值使所有点的权值是$1\sim n$的一个排列,定义一条边的权值为两端点权值差 ...
- CF F - Tree with Maximum Cost (树形DP)给出你一颗带点权的树,dist(i, j)的值为节点i到j的距离乘上节点j的权值,让你任意找一个节点v,使得dist(v, i) (1 < i < n)的和最大。输出最大的值。
题目意思: 给出你一颗带点权的树,dist(i, j)的值为节点i到j的距离乘上节点j的权值,让你任意找一个节点v,使得dist(v, i) (1 < i < n)的和最大.输出最大的值. ...
- [codeforces161D]Distance in Tree(点分治/树形dp)
题意:求树上距离为k的点对个数: 解题关键:练习一下点分治不用容斥 而直接做的做法.注意先查询,后更新. 不过这个方法有个缺陷,每次以一个新节点为根,必须memset mp数组,或许使用map会好些, ...
- 『You Are Given a Tree 整体分治 树形dp』
You Are Given a Tree Description A tree is an undirected graph with exactly one simple path between ...
- [BZOJ2152]聪聪可可 点分治/树形dp
2152: 聪聪可可 Time Limit: 3 Sec Memory Limit: 259 MB Submit: 3602 Solved: 1858 [Submit][Status][Discu ...
- BZOJ 2152 / Luogu P2634 [国家集训队]聪聪可可 (点分治/树形DP)
题意 一棵树,给定边权,求满足两点之间的路径上权值和为3的倍数的点对数量. 分析 点分治板题,对每个重心求子树下面的到根的距离模3分别为0,1,2的点的个数就行了. O(3nlogn)O(3nlogn ...
- 2014 Super Training #9 E Destroy --树的直径+树形DP
原题: ZOJ 3684 http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3684 题意: 给你一棵树,树的根是树的中心(到其 ...
随机推荐
- nginx开启ssl并把http重定向到https的两种方式
1 简介 Nginx是一个非常强大和流行的高性能Web服务器.本文讲解Nginx如何整合https并将http重定向到https. https相关文章如下: (1)Springboot整合https原 ...
- Clickhouse 时区转换
Clickhouse 时区转换 ClickHouse是一个用于联机分析(OLAP)的列式数据库管理系统(DBMS). OLAP场景的关键特征 大多数是读请求 数据总是以相当大的批(> 1000 ...
- ACM一年记,总结报告(希望自己可以走得很远)
一. 知识点梳理 (一) 先从工具STL说起: 容器学习了:stack,queue,priority_queue,set/multiset,map/multimap,vector. 1.stack: ...
- unittest 中的方法调用时报错 ValueError: no such test method in <class 'mytestcase.MyTestCase'>: runTest
1.调用unittest中的方法时报错: ValueError: no such test method in <class 'mytestcase.MyTestCase'>: runTe ...
- 怎么成为一名WEB前端开发工程师
对于刚开始学的人来说,web 就是HTML+CSS+JavaScript其实我们是可以这样理解的.web工程师负责或参与Web产品的页面开发,包含PC端.移动APP内嵌移动端.微信小程序.web ...
- CC2530ADC应用
ADC单通道外部电压采集 需要设置一个上机位命令控制字符. 系统时钟初始化——32MHZ晶振 串口0函数初始化——设置串口对应引脚,波特率,清楚中断标志 串口0接收中断响应函数——U0DBUF将控制命 ...
- mysql优化–explain分析sql语句执行效率
Explain命令在解决数据库性能上是第一推荐使用命令,大部分的性能问题可以通过此命令来简单的解决,Explain可以用来查看SQL语句的执行效 果,可以帮助选择更好的索引和优化查询语句,写出更好的优 ...
- 一文教你快速学会在matlab的simulink中调用C语言进行仿真
本文介绍如何在matlab的simulink中嵌入C语言进行多输入多输出的仿真:matlab版本位2015b: 创作不易,如果本文帮到了您: 如果本文帮到了您,请帮忙点个赞
- [codeforces-543B]bfs求最短路
题意:给一个边长为1的无向图,求删去最多的边使得从a到b距离<=f,从c到d距离<=g,a,b,c,d,f,g都是给定的,求最多删去的边数. 思路:反过来思考,用最少的边构造两条从a到b, ...
- vscode调试webpack的启动和打包部署过程,nodejs调试
launch.json ``` { // 使用 IntelliSense 了解相关属性. // 悬停以查看现有属性的描述. // 欲了解更多信息,请访问: https://go.micr ...