P2590 [ZJOI2008]树的统计

怎么表达我此时的心情呢，重构代码到我想把它吃下去

P2590 [ZJOI2008]树的统计

题解

线段树维护树链剖分

1.单点修改

   不需要pushdown

2.区间最大值

   取答案，初始化ans最小值

3.区间求和

   点权和，最后dfn[x]就不用+1了

   如果是边权和押到点上，那就需要+1啦

   

讲讲锅在哪里！！！

建树！！！ build！！！

由于建树的时候是在 dfs 序的基础上建的，所以区间的端点都是 dfs 序，如果取对应的值，不能直接用端点标号，而是应该用 端点标号dfs序 对应的 树上节点 来取值 QAQ

 

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdlib>
#include<queue>

using namespace std;

typedef long long ll;

inline int read()
{
    int ans=0;
    char last=' ',ch=getchar();
    while(ch<'0'||ch>'9') last=ch,ch=getchar();
    while(ch>='0'&&ch<='9') ans=ans*10+ch-'0',ch=getchar();
    if(last=='-') ans=-ans;
    return ans; 
}

const int maxn=30010,inf=-2147483644;
int n,q,w[maxn];
string s;
int head[maxn],to[maxn<<2],nxt[maxn<<2],cnt=0;
inline void addedge(int u,int v)
{
    to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt;
    to[++cnt]=u;nxt[cnt]=head[v];head[v]=cnt;
}
int fa[maxn],size[maxn],son[maxn],dfn[maxn],que[maxn],dep[maxn],top[maxn],tim=0;
int zd[maxn<<2],sum[maxn<<2];

inline void dfs1(int u,int f)
{
    fa[u]=f;
    size[u]=1;
    son[u]=0;
    for(int i=head[u];i;i=nxt[i]){
        int v=to[i];
        if(v==f) continue;
        dfs1(v,u);
        size[u]+=size[v];
        if(size[v]>size[son[u]]) son[u]=v;
    }
}

inline void dfs2(int u,int f)
{
    dfn[u]=++tim;
    que[tim]=u;
    dep[u]=dep[f]+1;
    top[u]=(son[f]==u?top[f]:u);
    if(son[u]) dfs2(son[u],u);
    for(int i=head[u];i;i=nxt[i]){
        int v=to[i];
        if(v==f||v==son[u]) continue;
        dfs2(v,u);
    }
}

inline void build(int k,int l,int r)
{
    if(l==r) //注意这里建树！！！ 
    {
        zd[k]=w[que[l]];    
        sum[k]=w[que[l]];
        return ;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    zd[k]=max(zd[k<<1],zd[k<<1|1]);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

inline void modify(int k,int l,int r,int p,int v)
{
    if(l==r) {
        zd[k]=v;
        sum[k]=v;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid) modify(k<<1,l,mid,p,v);
    if(mid<p) modify(k<<1|1,mid+1,r,p,v);
    zd[k]=max(zd[k<<1],zd[k<<1|1]);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

int querymax(int k,int l,int r,int x,int y)
{
    int ans=inf;
    if(y<l||r<x||l>r||x>y) return ans; 
    if(x<=l&&r<=y) return zd[k];
    int mid=(l+r)>>1;
    if(x<=mid) ans=max(ans,querymax(k<<1,l,mid,x,y));
    if(mid<y) ans=max(ans,querymax(k<<1|1,mid+1,r,x,y));
    return ans;
}

int qusmax(int x,int y)
{
    int ans=inf;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        ans=max(ans,querymax(1,1,n,dfn[top[x]],dfn[x]));
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    ans=max(ans,querymax(1,1,n,dfn[x],dfn[y]));
    return ans;
}

int querysum(int k,int l,int r,int x,int y)
{
    int ans=0;
    if(y<l||r<x||l>r||x>y) return 0;
    if(x<=l&&r<=y) return sum[k];
    int mid=(l+r)>>1;
    if(x<=mid) ans+=querysum(k<<1,l,mid,x,y);
    if(mid<y) ans+=querysum(k<<1|1,mid+1,r,x,y);
    return ans;
}

int qussum(int x,int y)
{
    int ans=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        ans+=querysum(1,1,n,dfn[top[x]],dfn[x]);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    ans+=querysum(1,1,n,dfn[x],dfn[y]);
    return ans;
}

int main()
{
    n=read();
    int x,y;
    for(int i=1;i<n;i++)
    {
        x=read(),y=read();
        addedge(x,y);
    }
    for(int i=1;i<=n;i++) w[i]=read();
    q=read();
    
    dfs1(1,0);
    dfs2(1,0);
    build(1,1,n);
    
    for(int i=1;i<=q;i++)
    {
        cin>>s;
        x=read();y=read();
        if(s[1]=='H') modify(1,1,n,dfn[x],y);
        if(s[1]=='M') printf("%d
",qusmax(x,y));
        if(s[1]=='S') printf("%d
",qussum(x,y));
    }
    
    return 0;
}

---

In The End

感谢善良的Sy&Lz帮忙出数据&看代码

我太难了QAQ

没错我重构以后也没看出毛病