ZR1050
ZR1050
http://www.zhengruioi.com/problem/1030
题目大意:
给定一棵带点权的树,求所有联通块的点权和的平方的和 \(n \le 10^5\)
题解
首先,关于平方的和或者和的平方我们一般都是考虑新加进来一个点会产生什么样的贡献
比如这一题,我们考虑合并两个集合的代价会发生什么样的变化
首先,设\(a,b\)分别为合并的两个集合的点权和
那么由$a^2 \(变为\)(a + b) ^ 2$的区别是代价多了个\(2ab\)和\(b^2\)
也就是说我们在DP的过程中需要维护所有联通块的点权和进行转移
接下,直接暴力计算是不可以的,我们需要对所有的联通块统一计算,发现对于合并每一对联通块的代价都是上面的东西
乘法分配律之后
我们发现还要额外记录当前点为根的联通块个数
设\(f_i\)表示以\(i\)点为根的联通块的个数
设\(g_i\)表示\(i\)点为根的所有联通块的点权和
设\(h_i\)表示以\(i\)点为根的所有联通块的点权和的平方的和
那么首先根据上面,我们在树形DP的时候可以合并两个联通块
\]
首先本来原有的联通块的代价肯定是要保留,另外,对于\(h_x\)可以被合并\(f_y\)次,同理\(h_y\)会被合并\(f_y\)次
最后由于\(g\)数组记录的本来就是总贡献,因此直接合并即可
同理我们可以得到\(g\)的转移
\]
最后更新答案即可
总结
对于这种和联通块有关的题目一般可以通过单独算贡献或者DP的方式解决
特别是和平方相关的,要考虑新加进来的数的贡献
代码
#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cctype>
#include<vector>
#include<ctime>
#include<cmath>
#include<set>
#include<map>
#define LL long long
#define pii pair<int,int>
#define mk make_pair
#define fi first
#define se second
using namespace std;
const int N = 2e5 + 3;
const LL mod = 998244353;
int n;
LL w[N];
LL f[N],g[N],h[N];
vector <int> G[N];
inline int read(){
int v = 0,c = 1;char ch = getchar();
while(!isdigit(ch)){
if(ch == '-') c = -1;
ch = getchar();
}
while(isdigit(ch)){
v = v * 10 + ch - 48;
ch = getchar();
}
return v * c;
}
inline void dfs(int x,int fa){//h:平方的和
f[x] = 1;g[x] = w[x];
h[x] = w[x] * w[x] % mod;
for(int i = 0;i < (int)G[x].size();++i){
int y = G[x][i];
if(y == fa) continue;
dfs(y,x);
h[x] = (h[x] + h[x] * f[y] % mod + 2 * g[x] * g[y] % mod + h[y] * f[x] % mod) % mod;
g[x] = (g[x] + g[x] * f[y] + g[y] * f[x]) % mod;
f[x] = (f[x] + f[x] * f[y]) % mod;
}
}
int main(){
n = read();
for(int i = 1;i <= n;++i) w[i] = read();
for(int i = 1;i < n;++i){
int x = read(),y = read();
G[x].push_back(y);
G[y].push_back(x);
}
dfs(1,0);
LL ans = 0;
for(int i = 1;i <= n;++i) ans = (ans + h[i]) % mod;
cout << ans;
return 0;
}
ZR1050的更多相关文章
随机推荐
- Hibernate中的Session对象 标签: hibernatesession 2017-01-22 22:10 238人阅读 评论(
Hibernate中的Session 大家在看hibernate视频的时候一定都发现了,每次要操作数据库,总是要新建一个session对象,Hibernate在对资料库进行操作之前,必须先取得Sess ...
- jq方法的注意点
当jq方法里面引用的ajax方法和其它方法时,就需要把ajax改为同步,通过ajax方法返回值来判断下一步执行那个方法,你不做判断,浏览器编译执行的时候不会不会按你想的从上之下执行下来. 当安卓手机跟 ...
- redis :read error on connection
最近做了一个多人竞拍的小功能 因为以前没做过 所以踩了很多坑用的是 mysql + php + redis 实现的竞拍功能 这里先说一下踩得第二个坑redis 的原因 真是欲哭无泪 解决完一个 ...
- js原生复习2.0
// 1.闭包的作用// 实现共有变量,函数累加器的实现// 可以做缓存以及储存结构// 可以实现封装,实现属性私有化// 模块开发,防止全局污染// var name = 123;// var in ...
- input禁止复制、粘贴、剪切
<input type="text" autocomplete="off"> <!-- autocomplete="off" ...
- Spring Data JPA 查询结果返回至自定义实体
本人在实际工作中使用Spring Data Jpa框架时,一般查询结果只返回对应的Entity实体.但有时根据实际业务,需要进行一些较复杂的查询,比较棘手.虽然在框架上我们可以使用@Query注解执行 ...
- 12-2 js基础
一 数据类型 <!DOCTYPE html> <html lang="en"> <head> <meta charset="UT ...
- 在springmvc中 @RequestMapping(value={"", "/"})是什么意思
这个意思是说请求路径 可以为空或者/ 我给你举个例子:比如百度知道的个人中心 访问路径是 http://zhidao.baidu.com/ihome,当然你也可以通过 http://zhidao.ba ...
- 深度解读Helm 3: 犹抱琵琶半遮面
自去年年初开始放风Helm v3将要开始开发,到去年年底KubeConn 上海被一堆人追问到底啥时候发版本.今年五月份,Helm v3 终于发布了第一个alpha版本,让我们来一窥新版本的Helm 到 ...
- php parse_url linux 解析问题
耕毅 解析url函数parse_url() (PHP 4, PHP 5, PHP 7) parse_url — 解析 URL,返回其组成部分 mixed parse_url ( string $url ...