BZOJ3631 松鼠的新家

BZOJ3631 松鼠的新家

题目传送门

题解

又是一道树剖题，对于这个访问顺序，我们用树剖把路径\(a[i],a[i+1]\)上的每个点都加1，但是这样会重复计算，所以我们每次路径修改的时候，不修改起点，最后把\(a[1]\)的贡献加上，并且把\(a[n]\)的贡献减掉就行了。

ｃｏｄｅ

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x%10+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('\n');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
/*================Header Template==============*/
const int maxn=3e5+500;
int n,tot;
int a[maxn];
struct edge {
    int to,nxt;
}E[maxn<<1];
int head[maxn],pos[maxn],idx[maxn],heav[maxn],res[maxn],sz[maxn],fa[maxn],dep[maxn],top[maxn];
/*==================Define Area================*/
void addedge(int u,int v) {
    E[++tot].to=v;E[tot].nxt=head[u];head[u]=tot;
    E[++tot].to=u;E[tot].nxt=head[v];head[v]=tot;
}

namespace SegmentTree {
    struct node {
        int l,r,sz,v;
        int lazy;
    }t[maxn<<2];
    #define ls(o) o<<1
    #define rs(o) o<<1|1
    void Update(int o) {
        t[o].v=t[ls(o)].v+t[rs(o)].v;
    }
    void Pushdown(int o) {
        if(!t[o].lazy) return ;
        t[ls(o)].v+=t[ls(o)].sz*t[o].lazy;t[rs(o)].v+=t[rs(o)].sz*t[o].lazy;
        t[ls(o)].lazy+=t[o].lazy;t[rs(o)].lazy+=t[o].lazy;
        t[o].lazy=0;
    }
    void Build(int o,int l,int r) {
        t[o].l=l;t[o].r=r;t[o].sz=r-l+1;
        if(l==r) return ;
        int mid=(l+r)>>1;
        if(mid>=l) Build(ls(o),l,mid);
        if(mid<r) Build(rs(o),mid+1,r);
        return ;
    }
    void Modify(int o,int l,int r,int v) {
        if(t[o].l>=l&&t[o].r<=r) {
            t[o].v+=t[o].sz*v;
            t[o].lazy+=v;
            return ;
        }
        Pushdown(o);
        int mid=(t[o].l+t[o].r)>>1;
        if(mid>=l) Modify(ls(o),l,r,v);
        if(mid<r) Modify(rs(o),l,r,v);
        Update(o);
        return ;
    }
    void Answer(int o) {
        if(t[o].l==t[o].r) {
            res[pos[t[o].l]]=t[o].v;
            return ;
        }
        Pushdown(o);
        Answer(ls(o));
        Answer(rs(o));
    }
}
using namespace SegmentTree;

namespace poufen {
    void Dfs1(int o,int ff,int deep) {
        fa[o]=ff;sz[o]=1;dep[o]=deep;
        int mxson=-1;
        for(int i=head[o];~i;i=E[i].nxt) {
            int to=E[i].to;
            if(to==ff) continue;
            Dfs1(to,o,deep+1);
            sz[o]+=sz[to];
            if(sz[to]>mxson) mxson=sz[to],heav[o]=to;
        }
    }
    int cnt=0;
    void Dfs2(int o,int topp) {
        idx[o]=++cnt;
        top[o]=topp;
        pos[cnt]=o;
        if(!heav[o]) return ;
        Dfs2(heav[o],topp);
        for(int i=head[o];~i;i=E[i].nxt) {
            int to=E[i].to;
            if(!idx[to]) Dfs2(to,to);
        } 
    }
    void TreeAdd(int x,int y) {
        Modify(1,idx[x],idx[x],-1);
        while(top[x]!=top[y]) {
            if(dep[top[x]]<dep[top[y]]) swap(x,y);
            Modify(1,idx[top[x]],idx[x],1);
            x=fa[top[x]];
        }
        if(dep[x]>dep[y]) swap(x,y);
        Modify(1,idx[x],idx[y],1);
        return ;
    }
}
using namespace poufen;

int main() {
    memset(head,-1,sizeof head);
    read(n);
    for(int i=1;i<=n;i++) {
        read(a[i]);
    }
    for(int i=1,u,v;i<n;i++) {
        read(u);read(v);
        addedge(u,v);
    }
    Dfs1(1,1,1);
    Dfs2(1,1);
    Build(1,1,n);
    for(int i=1;i<n;i++) {
        TreeAdd(a[i],a[i+1]);
    }
    Answer(1);
    res[a[1]]++;res[a[n]]--;
    for(int i=1;i<=n;i++) {
        printf("%d\n",res[i]);
    }
    return 0;
}

转载于:https://www.cnblogs.com/Apocrypha/p/9453661.html