首页 > 其他分享 >[题解]AT_abc222_f [ABC222F] Expensive Expense

[题解]AT_abc222_f [ABC222F] Expensive Expense

时间:2024-06-22 10:57:13浏览次数:13  
标签:int 题解 tree fa modify abc222 inline ABC222F id

板子题,模拟赛场切了。

思路

线段树换根板子题。

因为需要求每一个点的答案,所以定义 \(dp_i\) 表示以 \(i\) 为根的最长距离。

考虑将一个点 \(v\) 转化为根,树的形态会发生什么变化(假设 \(v\) 的父亲节点是 \(u\))。

发现在 \(v\) 子树中的节点,距离都会减少 \(w_{u \to v}\),其它节点都会加 \(w_{u \to v}\)。这个变化显然是可以将 DFS 序剖出来,然后用线段树优化的。

因为不能选择自己作为终点,所以查询最大值的时候避开即可。

Code

#include <bits/stdc++.h>  
#define re register  
#define int long long  
  
using namespace std;  
  
const int N = 2e5 + 10,M = 4e5 + 10;  
int n;  
int dp[N];  
int idx,h[N],ne[M],e[M],w[M],p[N];  
int num,f[N],d[N],sz[N],wson[N],id[N],tp[N];  
  
inline int read(){  
    int r = 0,w = 1;  
    char c = getchar();  
    while (c < '0' || c > '9'){  
        if (c == '-') w = -1;  
        c = getchar();  
    }  
    while (c >= '0' && c <= '9'){  
        r = (r << 3) + (r << 1) + (c ^ 48);  
        c = getchar();  
    }  
    return r * w;  
}  
  
inline void add(int a,int b,int c){  
    ne[idx] = h[a];  
    e[idx] = b;  
    w[idx] = c;  
    h[a] = idx++;  
}  
  
struct chain{  
    #define ls(u) (u << 1)  
    #define rs(u) (u << 1 | 1)  
  
    struct node{  
        int l,r;  
        int Max,tag;  
    }tr[N << 2];  
  
    inline void calc(int u,int k){  
        tr[u].Max += k;  
        tr[u].tag += k;  
    }  
  
    inline void pushup(int u){  
        tr[u].Max = max(tr[ls(u)].Max,tr[rs(u)].Max);  
    }  
  
    inline void pushdown(int u){  
        if (tr[u].tag){  
            calc(ls(u),tr[u].tag);  
            calc(rs(u),tr[u].tag);  
            tr[u].tag = 0;  
        }  
    }  
  
    inline void build(int u,int l,int r){  
        tr[u] = {l,r};  
        if (l == r) return;  
        int mid = l + r >> 1;  
        build(ls(u),l,mid);  
        build(rs(u),mid + 1,r);  
        pushup(u);  
    }  
  
    inline void modify(int u,int l,int r,int k){  
        if (l <= tr[u].l && tr[u].r <= r){  
            calc(u,k);  
            return;  
        }  
        pushdown(u);  
        int mid = tr[u].l + tr[u].r >> 1;  
        if (l <= mid) modify(ls(u),l,r,k);  
        if (r > mid) modify(rs(u),l,r,k);  
        pushup(u);  
    }  
  
    inline int query(int u,int l,int r){  
        if (l <= tr[u].l && tr[u].r <= r) return tr[u].Max;  
        pushdown(u);  
        int res = 0;  
        int mid = tr[u].l + tr[u].r >> 1;  
        if (l <= mid) res = max(res,query(ls(u),l,r));  
        if (r > mid) res = max(res,query(rs(u),l,r));  
        return res;  
    }  
  
    inline void modify_node(int x,int k){  
        modify(1,id[x],id[x],k);  
    }  
  
    inline void modify_tree(int x,int k){  
        modify(1,id[x],id[x] + sz[x] - 1,k);  
    }  
  
    inline int query_sec(int l,int r){  
        if (l > r) return 0;  
        return query(1,l,r);  
    }  
  
    #undef ls  
    #undef rs  
}tree;  
  
inline void dfs1(int u,int fa){  
    sz[u] = 1;  
    f[u] = fa;  
    d[u] = d[fa] + 1;  
    for (re int i = h[u];~i;i = ne[i]){  
        int j = e[i];  
        if (j == fa) continue;  
        dfs1(j,u);  
        if (sz[j] > sz[wson[u]]) wson[u] = j;  
        sz[u] += sz[j];  
    }  
}  
  
inline void dfs2(int u,int fa,int top){  
    num++;  
    id[u] = num;  
    tp[u] = top;  
    if (!wson[u]) return;  
    dfs2(wson[u],u,top);  
    for (re int i = h[u];~i;i = ne[i]){  
        int j = e[i];  
        if (j == fa || j == wson[u]) continue;  
        dfs2(j,u,j);  
    }  
}  
  
inline void dfs_val(int u,int fa,int d){  
    tree.modify_node(u,d + p[u]);  
    for (re int i = h[u];~i;i = ne[i]){  
        int j = e[i];  
        if (j == fa) continue;  
        dfs_val(j,u,d + w[i]);  
    }  
}  
  
inline void dfs_get(int u,int fa){  
    for (re int i = h[u];~i;i = ne[i]){  
        int j = e[i];  
        if (j == fa) continue;  
        tree.modify_tree(1,w[i]);  
        tree.modify_tree(j,-2 * w[i]);  
        dp[j] = max(tree.query_sec(1,id[j] - 1),tree.query_sec(id[j] + 1,n));  
        dfs_get(j,u);  
        tree.modify_tree(1,-w[i]);  
        tree.modify_tree(j,2 * w[i]);  
    }  
}  
  
signed main(){  
    memset(h,-1,sizeof(h));  
    n = read();  
    for (re int i = 1;i < n;i++){  
        int a,b,c;  
        a = read();  
        b = read();  
        c = read();  
        add(a,b,c);  
        add(b,a,c);  
    }  
    for (re int i = 1;i <= n;i++) p[i] = read();  
    dfs1(1,0);  
    dfs2(1,0,1);  
    tree.build(1,1,n);  
    dfs_val(1,0,0);  
    dp[1] = max(tree.query_sec(1,id[1] - 1),tree.query_sec(id[1] + 1,n));  
    dfs_get(1,0);  
    for (re int i = 1;i <= n;i++) printf("%lld\n",dp[i]);  
    return 0;  
}  

标签:int,题解,tree,fa,modify,abc222,inline,ABC222F,id
From: https://www.cnblogs.com/WaterSun/p/18261950

相关文章

  • [题解]AT_abc217_g [ABC217G] Groups
    思路定义\(dp_{i,j}\)表示将前\(i\)个数,正好分为\(j\)组的方案数。那么,我们对\(i\)号元素进行分类讨论:将\(i\)放入原本就存在的组中,因为在同一个组中不能存在两个数\(x,y\),使得\(x\bmodm=y\bmodm\)。所以对于\(i\),如果它是\(m\)的倍数,则在\(1\simi-......
  • [题解]AT_abc217_f [ABC217F] Make Pair
    思路区间DP好题,合并的时候十分毒瘤。首先,定义\(dp_{i,j}\)表示合并\([i,j]\)区间不同的方案的数量。不难发现,如果区间长度为奇数(即\(j-i+1\)为奇数),一定无法合并。然后,如果\(i,j\)是朋友关系,有\(dp_{i,j}=dp_{i+1,j-1}\)。接着,我们可以枚举一个中间点\(......
  • [题解]AT_abc216_f [ABC216F] Max Sum Counting
    思路首先,不难发现,对于本题将\(a,b\)合成一个序列,并按照\(a_i\)排序的答案不会发生变化。所以,我们可以直接排序,那么,我们当前枚举到的\(a_i\)就是当前的\(\max(a_i)\)。定义\(dp_{i,j,0/1}\)表示在\(1\simi\)中,选择的\(b_i\)之和为\(j\),并且第\(i\)个数不选/选......
  • [题解]AT_abc215_g [ABC215G] Colorful Candies 2
    思路定义\(vis_i\)表示数\(i\)在序列中出现的次数。如果我们选出\(k\)个数,答案就是(其中\(m\)表示\(\max(c_i)\)):\[\sum_{i=1}^m\frac{\binom{n}{x}-\binom{n-vis_i}{k}}{\binom{n}{x}}\]显然,我们只枚举序列中存在的元素,时间复杂度\(\Theta(n^2)\),过不......
  • [题解]AT_abc195_d [ABC195D] Shipping Center
    思路一个简单的贪心,对于每一次操作,我们假设我们能用盒子的大小的数组处理成\(a\)。那么,我们可以对\(a\)进行从小到大排序。然后,对于我们所有的箱子,我们可以以\(w\)为关键字,从小到大排序。接着,我们可以进行暴力枚举,对于\(a_i\),我们要取的必定为\(\max_{w_j\leqa_i}(v_j......
  • [题解]AT_abc158_e [ABC158E] Divisible Substring
    思路首先发现一个事情,任意一个子串都可以由\(s\)的某一个后缀的后面删除一些字符得到。因此假如\(s\)的某一个后缀的值为\(x\),那么我们可以减去后面的我们不用的数字\(a\),然后除以\(10\)的若干次幂得到,即\(\frac{x-a}{10^n}\)。于是得到:\[\frac{x-a}{10^n}\equi......
  • [题解]AT_abc153_f [ABC153F] Silver Fox vs Monster
    模拟赛最后\(15\)分钟想到的做法。思路首先有一个显然的贪心策略:我们放炸弹的地方要尽可能的使这个炸弹能影响到更多的怪上。那么我们可以将对于一个怪\(i\)能够影响到它的区间表示出来\([\max(1,l_i-d),a_i+r]\)。然后将这些区间排个序,可以粗略画出这样的图:根据上......
  • [题解]AT_abc151_e [ABC151E] Max-Min Sums
    思路考虑将\(\max\)和\(\min\)的贡献分开计算。显然我们对这个序列进行一次排序不会影响最终的答案,因此我们可以先排序一下。然后有一个很经典的trick,就是你枚举每一个数\(x\),将\(x\)令为最大值(最小值)。因为我们先前排序过一次,因此我们可以轻易的计算出比\(x\)小(大)的......
  • [题解]AT_abc236_e [ABC236E] Average and Median
    思路直接将输出的答案分为两个分考虑。(1)考虑二分+DP。设当前二分出的平均数为\(x\),如果合法,那么有(其中\(p\)为选出数下标的集合):\[\frac{a_{p_1}+a_{p_2}+\dots+a_{p_k}}{k}\geqx\]即:\[\frac{(a_{p_1}-x)+(a_{p_2}-x)+\dots+(a_{p_......
  • 2020C++等级考试二级真题题解
     202012数组指定部分逆序重放c++ #include<iostream>usingnamespacestd;intmain(){  inta[110];  intn,k;  cin>>n>>k;  for(inti=0;i<n;i++){    cin>>a[i];  }  for(inti=0;i<k/2;i++){......