P3047 [USACO12FEB] Nearby Cows G

合集 - 洛谷(99)

1.P8755 [蓝桥杯 2021 省 AB2] 负载均衡2023-11-222.P1102 A-B 数对的三种解法2023-11-263.P1090 [NOIP2004 提高组] 合并果子2023-11-264.P3379 【模板】最近公共祖先（LCA）2023-11-285.P8599 [蓝桥杯 2013 省 B] 带分数2023-11-286.P1433 吃奶酪2023-11-297.P3385 【模板】负环2023-12-068.P1576 最小花费2023-12-069.Zuma2023-12-0610.P3205 [HNOI2010] 合唱队2023-12-0611.P1220 关路灯2023-12-0712.P4170 [CQOI2007] 涂色（天赋哥不要点进来）2023-12-1113.P8805 [蓝桥杯 2022 国 B] 机房2023-12-1214.P3386 【模板】二分图最大匹配2023-12-2015.P2197 【模板】Nim 游戏2023-12-2016.P1036 [NOIP2002 普及组] 选数2023-12-2117.T397291 【模板】拓扑排序（加强版）2023-12-2618.P2865 [USACO06NOV] Roadblocks G2023-12-2619.P1168 中位数2024-01-0220.P1631 序列合并2024-01-0221.P3372 【模板】线段树 12024-01-0222.P1352 没有上司的舞会2024-01-0223.P2015 二叉苹果树2024-01-0324.P2014 [CTSC1997] 选课2024-01-0325.P1613 跑路2024-01-0326.P2911 [USACO08OCT] Bovine Bones G2024-01-0427.P4316 绿豆蛙的归宿2024-01-0428.P3870 [TJOI2009] 开关2024-01-1029.P3373 【模板】线段树 22024-01-1430.P1558 色板游戏2024-01-1531.P1638 逛画展2024-01-1632.P1043 [NOIP2003 普及组] 数字游戏2024-01-2633.P1147 连续自然数和2024-01-2634.P1063 [NOIP2006 提高组] 能量项链2024-01-2735.P4342 [IOI1998] Polygon2024-01-2736.P1122 最大子树和2024-02-0537.P2016 战略游戏2024-02-0538.P2585 [ZJOI2006] 三色二叉树2024-02-0739.P1273 有线电视网2024-02-0740.P2986 [USACO10MAR] Great Cow Gathering G2024-02-0741.P1114 “非常男女”计划2024-02-0842.P5026 Lycanthropy2024-02-0843.P4231 三步必杀2024-02-0944.P2985 [USACO10FEB] Chocolate Eating S2024-02-0945.P4090 [USACO17DEC] Greedy Gift Takers P2024-02-1046.P4933 大师2024-02-1047.P1082 [NOIP2012 提高组] 同余方程2024-02-1148.P3811 【模板】模意义下的乘法逆元2024-02-1249.P9725 [EC Final 2022] Chase Game 22024-02-1250.P1330 封锁阳光大学2024-02-1251.P8674 [蓝桥杯 2018 国 B] 调手表2024-02-1252.P10111 [GESP202312 七级] 纸牌游戏2024-02-1453.P9975 [USACO23DEC] Cowntact Tracing 2 B2024-02-1554.P8783 [蓝桥杯 2022 省 B] 统计子矩阵2024-02-1555.P8786 [蓝桥杯 2022 省 B] 李白打酒加强版2024-02-1556.P8807 [蓝桥杯 2022 国 C] 取模2024-02-1557.P8732 [蓝桥杯 2020 国 ABC] 答疑2024-02-1558.P8725 [蓝桥杯 2020 省 AB3] 画中漂流2024-02-1559.P3958 [NOIP2017 提高组] 奶酪2024-02-1560.P4084 [USACO17DEC] Barn Painting G2024-02-1561.P1892 [BOI2003] 团伙2024-02-1662.P1198 [JSOI2008] 最大数2024-02-1663.P1204 [USACO1.2] 挤牛奶Milking Cows2024-02-1664.P5490 【模板】扫描线2024-02-1765.P8784 [蓝桥杯 2022 省 B] 积木画2024-02-1766.P9325 [CCC 2023 S2] Symmetric Mountains2024-02-1767.Iva & Pav2024-02-1768.P9889 [ICPC2018 Qingdao R] Plants vs. Zombies2024-02-1769.P9847 [ICPC2021 Nanjing R] Crystalfly2024-02-1870.E. Anna and the Valentine's Day Gift2024-02-1971.F. Chat Screenshots2024-02-1972.G. One-Dimensional Puzzle2024-02-2073.P2899 [USACO08JAN] Cell Phone Network G2024-02-2074.P2023 [AHOI2009] 维护序列2024-02-2075.P1531 I Hate It2024-02-2076.P5057 [CQOI2006] 简单题2024-02-2077.P2357 守墓人2024-02-2078.P8775 [蓝桥杯 2022 省 A] 青蛙过河2024-02-2179.P1656 炸铁路2024-02-2280.Watering an Array2024-02-2381.Dora and Search2024-02-2482.P1137 旅行计划2024-02-2483.P2835 刻录光盘2024-02-2484.P1197 [JSOI2008] 星球大战2024-02-2485.P3388 【模板】割点（割顶）2024-02-2486.P8435 【模板】点双连通分量2024-02-2587.P8436 【模板】边双连通分量2024-02-2588.P2860 [USACO06JAN] Redundant Paths G2024-02-2589.P3067 [USACO12OPEN] Balanced Cow Subsets G2024-02-2690.P4799 [CEOI2015 Day2] 世界冰球锦标赛2024-02-2691.P6492 [COCI2010-2011#6] STEP2024-02-2692.P1653 [USACO04DEC] Cow Ski Area G2024-02-2693.P1040 [NOIP2003 提高组] 加分二叉树2024-03-0394.P5322 [BJOI2019] 排兵布阵2024-03-0395.P2946 [USACO09MAR] Cow Frisbee Team S2024-03-0396.P1156 垃圾陷阱2024-03-0397.P1064 [NOIP2006 提高组] 金明的预算方案2024-03-04

98.P3047 [USACO12FEB] Nearby Cows G2024-03-04

99.P2466 [SDOI2008] Sue 的小球2024-03-04

收起

原题链接

题解

核心技巧：两次搜索
第一次搜索：
搜索出 $f[now][i]$ 以 $now$ 为根节点的子树且距离根节点恰好为 $i$ 的节点的个数

搜索完了之后，把范围 $k$ 以内的累加

第二次搜索：
由于整棵树的根节点的 $f$ 等于整棵树里距离不大于 $k$ 的节点个数，即已经符合题目要求
那么我们可以状态转移，子节点的 $f[k]$ 加上父节点的 $f[k-1]$ 再减去 子节点的 $f[k-2]$ （容斥，注意这里要倒着来）

code

#include<bits/stdc++.h>
using namespace std;
vector<int> G[100005];
int f[100005][25]={0};
int deep[100005]={0};
int a[100005]={0};
int n,k;
void ss1(int now)
{
    for(auto next:G[now])
    {
        if(!deep[next])
        {
            deep[next]=deep[now]+1;
            ss1(next);
            for(int i=1;i<=k;i++) f[now][i]+=f[next][i-1];
        }   
    }
}

void ss2(int now)
{
    for(auto next:G[now])
    {
        if(deep[next]>deep[now])
        {
            for(int i=k;i>=2;i--) f[next][i]=f[next][i]-f[next][i-2]+f[now][i-1];
            f[next][1]+=f[now][0];
            ss2(next);
        }
    }
}
int main()
{
    cin>>n>>k;

    for(int i=1;i<n;i++)
    {
        int x,y;
        cin>>x>>y;
        G[x].push_back(y);
        G[y].push_back(x);
    }

    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        f[i][0]=a[i];
    }

    deep[1]=1;
    ss1(1);

    for(int i=1;i<=n;i++)
        for(int j=1;j<=k;j++) f[i][j]+=f[i][j-1];

    ss2(1);

    for(int i=1;i<=n;i++) cout<<f[i][k]<<endl;
    return 0;
}