bzoj 4765: 普通计算姬

题意:

求树上一段连续编号的子树大小的和,带修改。

题解:

分块。
预处理一个东西 g[x][i] 表示x这个点对i这个块的影响。
然后就可以暴力搞了。
对于零散的点,就用树状数组维护dfs序的区间和。
貌似是 nn log n 的,不过就是能过。
还有以后打代码尽量注意,少递归,少递归,少递归!
最后要用 unsigned long long
code:

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL unsigned long long
using namespace std;
struct node{
    int y,next;
}a[200010];int len=0,last[100010];
int pre[100010],n,m,g[100010][350],d[100010],rt;
int ys[100010],z=0,la[100010],ri[350];
LL tr[100010],f[350];
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void ins(int x,int y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
int lowbit(int x){return x&(-x);}
void change(int x,int k)
{
    for(int i=x;i<=n;i+=lowbit(i))
        tr[i]+=(LL)k;
}
LL get(int x)
{
    LL ans=0;
    for(int i=x;i>=1;i-=lowbit(i))
        ans+=tr[i];
    return ans;
}
void dfs(int x,int fa)
{
    ys[x]=la[x]=++z;
    for(int i=1;i<=pre[n];i++) g[x][i]=g[fa][i];
    g[x][pre[x]]++;
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        if(y==fa) continue;
        dfs(y,x);
        la[x]=la[y];
    }
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;i++) d[i]=read();
    for(int i=1;i<=n;i++)
    {
        int x,y;x=read();y=read();
        if(x==0) rt=y;
        else ins(x,y),ins(y,x);
    }
    int len=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        pre[i]=i/len+1;
        if(pre[i]!=pre[i-1]) ri[pre[i-1]]=i-1;
    }
    //for(int i=1;i<=pre[n];i++) dfs(rt,0,i,0);
    dfs(rt,0);
    for(int i=1;i<=n;i++)
    {
        change(ys[i],d[i]);
        for(int j=1;j<=pre[n];j++)
            f[j]+=(LL)d[i]*(LL)g[i][j];
    }
    ri[pre[n]]=n;
    while(m--)
    {
        int op;op=read();
        if(op==1)
        {
            int x,c;x=read();c=read();
            change(ys[x],c-d[x]);
            for(int i=1;i<=pre[n];i++) f[i]+=(LL)(c-d[x])*(LL)g[x][i];
            d[x]=c;
        }
        else
        {
            int l,r;l=read();r=read();
            int k=l;LL ans=0;
            while(k<=r)
            {
                if((k==1||pre[k]!=pre[k-1])&&ri[pre[k]]<=r) ans+=f[pre[k]],k=ri[pre[k]];
                else ans+=(get(la[k])-get(ys[k]-1));
                k++;
            }
            printf("%llu\n",ans);
        }
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值