ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

【积累】【树】2020牛客多校 A National Pandemic(树剖)

2020-08-07 22:01:16  阅读:518  来源: 互联网

标签:树剖 int National 多校 dep maxn ans tid ll


2020牛客多校 A National Pandemic

题意

一棵树 ,三种操作:

1,一个中心城市 x,所有城市 y 的值+=w-dist(x,y)

2,将城市x的值与0取min

3,询问单点的值。

官方题解

对于1操作,我们考虑一次修改对y来说会增加w-dis(x,y)

w-dis(x,y)=w-(dep[x]+dep[y]-2*dep[lca])=w-dep[x]-dep[y]+2dep[lca]

所以,对于每次1操作,我们将其到根上所有点的cnt+=2,询问的时候那部分就是求它到根的权值和。

个人

不是代码不会敲,公式也会,但是就是想不到做法。看了题解才:哦哦哦哦哦!

唉,比赛时想了半天 一开始觉得是树剖,想做,结果不知道怎么下手,然后绕开树剖继续想。也就没有然后了。

个人代码:

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int mod=998244353;
const int inf=0x3f3f3f3f;
const int maxn=5e4+5;
const double ep=1e-6;

void read(int&x)
{
    char c;
    while(!isdigit(c=getchar()));x=c-'0';
    while(isdigit(c=getchar()))x=(x<<3)+(x<<1)+c-'0';
}
int n;
vector<int>p[maxn];
int siz[maxn],dep[maxn],fad[maxn],son[maxn];
int top[maxn],tid[maxn],rnk[maxn],cnt;
void dfs1(int u,int fa,int d)
{
    dep[u]=d;fad[u]=fa;siz[u]=1;son[u]=0;
    for(int v:p[u])
    {
        if(v==fa)continue;
        dfs1(v,u,d+1);
        siz[u]+=siz[v];
        if(siz[v]>siz[son[u]])son[u]=v;
    }
}
void dfs2(int u,int t)
{
    top[u]=t;tid[u]=++cnt;rnk[cnt]=u;
    if(!son[u])return;
    dfs2(son[u],t);
    for(int v:p[u])
        if(v!=son[u]&&v!=fad[u])
            dfs2(v,v);
}
struct Tr{
    ll sum[maxn<<2],ly[maxn<<2];
    inline void init(){mem(sum,0);mem(ly,0);}
    inline void pdown(int k,int l,int r){
        int mid=(l+r)>>1;
        sum[k<<1]+=ly[k]*(mid-l+1);sum[k<<1|1]+=ly[k]*(r-mid);
        ly[k<<1]+=ly[k];ly[k<<1|1]+=ly[k];
        ly[k]=0;
    }
    void update(int k,int l,int r,int L,int R,int v){
        if(L>r||R<l)return;
        if(L<=l&&r<=R){
            sum[k]+=1ll*v*(r-l+1);ly[k]+=v;
            return;
        }
        if(ly[k])pdown(k,l,r);
        int mid=(l+r)>>1;
        update(k<<1,l,mid,L,R,v);update(k<<1|1,mid+1,r,L,R,v);
        sum[k]=sum[k<<1]+sum[k<<1|1];
//        printf("update(%d %d)=%lld\n",l,r,sum[k]);
    }
    ll query(int k,int l,int r,int L,int R){
        if(L>r||R<l)return 0;
        if(L<=l&&r<=R)return sum[k];
        if(ly[k])pdown(k,l,r);
        int mid=(l+r)>>1;
        return query(k<<1,l,mid,L,R)+query(k<<1|1,mid+1,r,L,R);
    }
    void tupdate(int u,int v,int w){
        while(top[u]!=top[v]){
            if(dep[top[u]]<dep[top[v]])swap(u,v);
            update(1,1,n,tid[top[u]],tid[u],w);
            u=fad[top[u]];
        }
        if(dep[u]>dep[v])swap(u,v);
        update(1,1,n,tid[u],tid[v],w);
    }
    ll tquery(int u,int v){
        ll ans=0;
        while(top[u]!=top[v]){
            if(dep[top[u]]<dep[top[v]])swap(u,v);
            ans+=query(1,1,n,tid[top[u]],tid[u]);
            u=fad[top[u]];
        }
        if(dep[u]>dep[v])swap(u,v);
//        printf("lastans=%lld,query<%d %d>\n",ans,tid[u],tid[v]);
        ans+=query(1,1,n,tid[u],tid[v]);
        return ans;
    }
}st;
ll f[maxn];
int main()
{
    int T;
    read(T);
    while(T--)
    {
        int m,u,v;ll all=0,tim=0;
        read(n);read(m);
        for(int i=1;i<=n;i++)p[i].clear(),f[i]=0;
        for(int i=1;i<n;i++)
        {
            read(u);read(v);
            p[u].push_back(v);p[v].push_back(u);
        }
        cnt=0;dfs1(1,0,1);dfs2(1,1);st.init();
//        for(int i=1;i<=n;i++)printf("%d ",tid[i]);putchar(10);
        int op,x,w;
        while(m--)
        {
            read(op);
            if(op==1){
                read(x);read(w);all+=w-dep[x];tim++;
                st.tupdate(1,x,2);

            }else if(op==2){
                read(x);
                ll ans=all-f[x]-tim*dep[x];
                ans+=st.tquery(1,x);
                if(ans>0)f[x]+=ans;

            }else{
                read(x);
                ll ans=all-f[x]-tim*dep[x];
                ans+=st.tquery(1,x);
//                printf("(%lld)",st.tquery(1,x));
                printf("%lld\n",ans);
            }
        }
    }
}

标签:树剖,int,National,多校,dep,maxn,ans,tid,ll
来源: https://www.cnblogs.com/kkkek/p/13455458.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有