[ DP 树状数组 dfs序 倍增 ] Codeforces856D Masha and Cactus

15 篇文章 0 订阅
3 篇文章 0 订阅

%%%LargestJN
fi 表示点 i 子树的答案,si 表示点 i 所有儿子的答案减去 fi
假如点 i 不在环上,答案就是 i 的所有儿子的 f 和。
假如点 i 在环上,对于所有LCA是 i 的边 (x,y),假如选了这条边,答案就是边权, x i y i 路径上 sj 之和加上点 i 所有儿子的答案之和。
si 可以用 dfs 序和树状数组维护。

#include<bits/stdc++.h>
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void Read(int& x){
    char c=nc();
    for(;c<'0'||c>'9';c=nc());
    for(x=0;c>='0'&&c<='9';x=(x<<3)+(x<<1)+c-48,c=nc());
}
#define N 200010
#define M 20
vector<int>g[N],a[N];
int i,j,k,n,m,p,c[N];
int f[N];
int s[N],t[N],w[N];
int L[N],R[N],num,F[N][M],d[N];
inline void Dfs(int x){
    L[x]=++num;d[x]=d[F[x][0]]+1;
    for(int j=1;j<M;j++)F[x][j]=F[F[x][j-1]][j-1];
    for(int i=0;i<g[x].size();i++)Dfs(g[x][i]);
    R[x]=num;
}
inline int Lca(int x,int y){
    if(d[x]<d[y])swap(x,y);
    for(int i=M-1;i>=0;i--)if(d[F[x][i]]>=d[y])x=F[x][i];
    if(x==y)return x;
    for(int i=M-1;i>=0;i--)
    if(F[x][i]!=F[y][i])x=F[x][i],y=F[y][i];
    return F[x][0];
}
inline void Update(int x,int y){
    for(;x<=n;x+=x&-x)c[x]+=y;
}
inline void Merge(int x,int y,int z){
    Update(x,z);Update(y+1,-z);
}
inline int Query(int x){
    int Ans=0;
    for(;x;x-=x&-x)Ans+=c[x];
    return Ans;
}
inline void Solve(int x){
    int S=0;
    for(int i=0;i<g[x].size();i++){
        Solve(g[x][i]);
        S+=f[g[x][i]];
    }
    f[x]=S;
    for(int i=0;i<a[x].size();i++){
        int y=a[x][i];
        f[x]=max(f[x],Query(L[s[y]])+Query(L[t[y]])+S+w[y]);
    }
    Merge(L[x],R[x],S-f[x]);
}
int main(){
    Read(n);Read(m);
    for(i=2;i<=n;i++)Read(F[i][0]),g[F[i][0]].push_back(i);
    Dfs(1);
    for(i=1;i<=m;i++)Read(s[i]),Read(t[i]),Read(w[i]),a[Lca(s[i],t[i])].push_back(i);
    Solve(1);
    cout<<f[1]<<endl;
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值