[USACO12FEB]附近的牛Nearby Cows

题目链接：QAQ

题目大意：给出一棵n个点的树，每个点上有\(C_i\)头牛，问每个点k步范围内各有多少头牛。

Solution:

首先看看数据范围，\(n\le1e5,k\le20\)，那我们可以尝试设\(f[n][k]\)

设\(d[i][j]\)表示从i点开始向下k的范围内有多少头牛，\(f[i][j]\)代表从i点开始j范围内有多少头牛

\(d[i][j]\)可以很容易算出来，对于一个点\(u\),\(d[u][j]=C_u+\sum_{v\in son_u}d[v][j-1]\)

而\(f[i][j]\)也可以得出，\(f[v][j]=f[fa[v]][j-1]-d[v][j-2]+d[v][j]\)

最后输出\(f[i][k]\)就行了

Code:

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+1;
int n,k,cnt,head[N];
int a[N],d[N][21],f[N][21];
struct Edge{int nxt,to;}edge[N<<1];
void ins(int x,int y){edge[++cnt].nxt=head[x];edge[cnt].to=y;head[x]=cnt;
}
void dfs(int x,int fa,int u){d[x][u]+=a[x];for(int i=head[x];i;i=edge[i].nxt){int y=edge[i].to;if(y==fa) continue;d[x][u]+=d[y][u-1];dfs(y,x,u);}
}
void calc(int x,int fa,int u){int w=u-2>=0?d[x][u-2]:0;f[x][u]=f[fa][u-1]-w+d[x][u];for(int i=head[x];i;i=edge[i].nxt)if(edge[i].to!=fa) calc(edge[i].to,x,u);
}
void work(int u){for(int i=head[1];i;i=edge[i].nxt)calc(edge[i].to,1,u);
}
int read(){int x=0,f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}while(isdigit(ch)){x=x*10+ch-48;ch=getchar();}return x*f;
}
int main(){n=read(),k=read();for(int i=1;i<n;i++){int x=read(),y=read();ins(x,y),ins(y,x);}for(int i=1;i<=n;i++)a[i]=read(),d[i][0]=a[i];if(!k){for(int i=1;i<=n;i++)printf("%d\n",a[i]);return 0;}for(int i=1;i<=k;i++) dfs(1,0,i),f[1][i]=d[1][i];for(int i=1;i<=n;i++) f[i][0]=a[i];for(int i=1;i<=k;i++) work(i);for(int i=1;i<=n;i++) printf("%d\n",f[i][k]);return 0;
}

转载于:https://www.cnblogs.com/NLDQY/p/11117911.html

[USACO12FEB]附近的牛Nearby Cows相关推荐

P3047 [USACO12FEB]附近的牛Nearby Cows
题目描述 Farmer John has noticed that his cows often move between nearby fields. Taking this into accoun ...
洛谷P2017 [USACO09DEC]晕牛Dizzy Cows [拓扑排序]
题目传送门晕牛Dizzy Cows 题目背景 Hzwer 神犇最近又征服了一个国家,然后接下来却也遇见了一个难题. 题目描述 The cows have taken to racing each o ...
USACO12FEB Nearby Cows
完全没有思路,看了题解还有些迷设f[i][j]表示和节点\(i\)距离为\(j\)的节点的总数,\(son\)为\(i\)的儿子,\(num\)为儿子的数量则转移方程为f[i][j] = f[s] ...
P2017 [USACO09DEC]晕牛Dizzy Cows
图论日常不会系列... 题意:给定有向边和无向边,然后给每一条无向边定向,使得到的图无环. 我本来想缩一下点的,但是越想越晕. 然后就翻了题解,恍然大悟... 其实只需要给只有有向边的图跑一次topo ...
linux删除多余日志,linux 删除日志
https://jingyan.baidu.com/album/c1a3101e73129ade656deb9d.html?picindex=2 里面的 ls -s 可以看到目录 https://zh ...
USACO Clumsy Cows
洛谷 P3056 [USACO12NOV]笨牛Clumsy Cows 洛谷传送门 JDOJ 2323: USACO 2012 Nov Silver 1.Clumsy Cows JDOJ传送门 Desc ...
cocos creator小游戏案例之趣味套牛
趣味套牛 0. 思路搭建基本背景奔跑的牛套牛的绳子套牛成功的判定结算重新开始特效,(音效,粒子) 1. 基本背景按钮添加 button 组件 transition: sprite 2 ...
NOIP复健计划——动态规划
树形DP [POI2011] DYN-Dynamite 二分 K K K check(mid): 能否选出 m m m个点,使得 ∀ i 为关键点, M i n j i s s e l e c t e ...
动态规划--from zpz
版权声明:copy from zpz,我可能要修改 https://blog.csdn.net/qq_40828060/article/details/83064425 文章目录前言记忆化搜索动 ...
动态规划，且学且放弃
文章目录前言记忆化搜索动态规划的基本解题思路背包问题 01背包 (待处理)P1489 猫狗大战 01背包的空间优化问题 P1048 采药 P1510 精卫填海 P1566 加等式 P1504 ...

[USACO12FEB]附近的牛Nearby Cows

Solution:

Code:

[USACO12FEB]附近的牛Nearby Cows相关推荐

最新文章

热门文章