标签:题意 USACO12FEB int 题解 个数 转移 Cows 节点 Nearby
题解
题意:给你一棵 n 个点的树,点带权,对于每个节点求出距离它不超过 k 的所有节点权值和 m。
数据范围 n<=1e5,k<=20;
思路:考虑换根dp的转移,加上容斥,设f[i][j]表示与i这个节点相距j的节点个数,g[i][j]表示子树内与i节点相差j的节点个数;
假设已经知道父节点的一些信息,求子节点的话,发现需要加上f[fa][j-1]-g[i][j-2]+g[i][j];
g数组不用转移,就ok了
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+7;
struct edge{
int v,nxt;
}e[N<<1];
int n,cnt,k;
int a[N],head[N],fa[N];
int f[N][30],d[N][30];
void add_edge(int u,int v){
e[++cnt]=(edge){v,head[u]};
head[u]=cnt;
}
void build(int u){
for(int i=head[u];i;i=e[i].nxt){
int to=e[i].v;
if(to!=fa[u]){
fa[to]=u;
build(to);
for(int j=1;j<=k;j++){
d[u][j]+=d[to][j-1];
}
}
}
}
void dfs(int u){
if(u==1){
for(int i=1;i<=k;i++){
f[u][i]=d[u][i];
}
for(int i=head[u];i;i=e[i].nxt){
int to=e[i].v;
if(to!=fa[u]){
dfs(to);
}
}
}else{
for(int i=1;i<=k;i++){
f[u][i]=f[fa[u]][i-1]-d[u][i-2]+d[u][i];
}
for(int i=head[u];i;i=e[i].nxt){
int to=e[i].v;
if(to!=fa[u]){
dfs(to);
}
}
}
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<n;i++){
int x,y;
scanf("%d%d",&x,&y);
add_edge(x,y);
add_edge(y,x);
}
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
f[i][0]=d[i][0]=a[i];
for(int j=1;j<=k;j++){
d[i][j]=a[i];
}
}
build(1);
// for(int i=1;i<=n;i++){
// for(int j=0;j<=k;j++){
// cout<<d[i][j]<<" ";
// }
// cout<<"\n";
// }
dfs(1);
for(int i=1;i<=n;i++) cout<<f[i][k]<<"\n";
}
标签:题意,USACO12FEB,int,题解,个数,转移,Cows,节点,Nearby 来源: https://www.cnblogs.com/Aswert/p/13871864.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。