HDU 5221 Occupation

算是一个比较裸的树链剖分。。。但是为什么比赛的时候没人做。。。
树上每个点都有权值
3个操作
1° ans += u~v路径上点权值之和
2° 将点u的权值从ans中剔除
3° ans += 以u为root的子树的点权之和

1和2都是树链剖分的基本操作。至于3,如果你是dfs建图的话,那么你剖出来的点的序号终究是他们的dfs序,所以一个u的子树在序中的范围就是id[u] + size[u] - 1。那么第三个操作也就是很简单的操作了。

读入写错了,WA了一个晚上….

AL//      whn6325689
//      Mr.Phoebe
//      http://blog.csdn.net/u013007900
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>
#include <functional>
#include <numeric>
#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

#define eps 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LLINF 1LL<<62
#define speed std::ios::sync_with_stdio(false);

typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef complex<ld> point;
typedef pair<int, int> pii;
typedef pair<pii, int> piii;
typedef vector<int> vi;

#define CLR(x,y) memset(x,y,sizeof(x))
#define CPY(x,y) memcpy(x,y,sizeof(x))
#define clr(a,x,size) memset(a,x,sizeof(a[0])*(size))
#define cpy(a,x,size) memcpy(a,x,sizeof(a[0])*(size))

#define mp(x,y) make_pair(x,y)
#define pb(x) push_back(x)
#define lowbit(x) (x&(-x))

#define MID(x,y) (x+((y-x)>>1))
#define getidx(l,r) (l+r | l!=r)
#define ls getidx(l,mid)
#define rs getidx(mid+1,r)
#define lson l,mid
#define rson mid+1,r
#define root 1,n

template<class T>
inline bool read(T &n)
{
    T x = 0, tmp = 1;
    char c = getchar();
    while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
    if(c == EOF) return false;
    if(c == '-') c = getchar(), tmp = -1;
    while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
    n = x*tmp;
    return true;
}
template <class T>
inline void write(T n)
{
    if(n < 0)
    {
        putchar('-');
        n = -n;
    }
    int len = 0,data[20];
    while(n)
    {
        data[len++] = n%10;
        n /= 10;
    }
    if(!len) data[len++] = 0;
    while(len--) putchar(data[len]+48);
}
//-----------------------------------

const int MAXN=100010;

struct Edge
{
    int to,next;
} e[MAXN<<1];
int head[MAXN],tot;
int size[MAXN],p[MAXN],dep[MAXN],son[MAXN];
int n;

void init()
{
    CLR(head,-1);
    tot=0;
}

void addedge(int u,int v)
{
    e[tot].to=v;
    e[tot].next=head[u];
    head[u]=tot++;
}

void dfs(int u,int depth=0,int fa=0)
{
    p[u]=fa;
    dep[u]=depth;
    size[u]=1;
    son[u]=-1;
    for(int i=head[u]; ~i; i=e[i].next)
    {
        int v=e[i].to;
        if(v==fa)    continue;
        dfs(v,depth+1,u);
        size[u]+=size[v];
        if(son[u]==-1 || size[v]>size[son[u]])
            son[u]=v;
    }
}

int dfn[MAXN],top[MAXN],timestamp;
int val[MAXN],valsum[MAXN];

void getid(int u,int fa)
{
    dfn[u]=++timestamp;
    top[u]=fa;
    valsum[timestamp]=val[u]+valsum[timestamp-1];
    if(son[u]!=-1)
        getid(son[u],fa);
    for(int i=head[u]; ~i; i=e[i].next)
    {
        int v=e[i].to;
        if(v==p[u] || v==son[u])    continue;
        getid(v,v);
    }
}

int sum[MAXN<<1];
bool la[MAXN<<1];

inline void pushup(int l,int r)
{
    int mid=MID(l,r);
    sum[getidx(l,r)]=sum[ls]+sum[rs];
}

void pushdown(int l,int r)
{
    int idx=getidx(l,r),mid=MID(l,r);
    la[idx]=false;
    la[ls]=la[rs]=true;
    sum[ls]=valsum[mid]-valsum[l-1],sum[rs]=valsum[r]-valsum[mid];
}

void update(int L,int R,int l,int r)
{
    int idx=getidx(l,r);
    if(L<=l && r<=R)
    {
        la[idx]=true;
        sum[idx]=valsum[r]-valsum[l-1];
        return;
    }
    else if(l<=R && L<=r)
    {
        if(la[idx])
            pushdown(l,r);
        int mid=MID(l,r);
        update(L,R,l,mid);
        update(L,R,mid+1,r);
        pushup(l,r);
    }

}

void update(int u,int v,int n)
{
    while(top[u]!=top[v])
    {
        if(dep[top[u]]>dep[top[v]])
            swap(u,v);
        update(dfn[top[v]],dfn[v],1,n);
        v=p[top[v]];
    }
    if(dep[u]>dep[v])
        swap(u,v);
    update(dfn[u],dfn[v],1,n);
}


void del(int pos,int l,int r)
{
    int idx=getidx(l,r);
    if(l==pos && pos==r)
    {
        sum[idx]=0;
        return ;
    }
    if(la[idx])
        pushdown(l,r);
    int mid=MID(l,r);
    if(pos<=mid)
        del(pos,l,mid);
    else
        del(pos,mid+1,r);
    pushup(l,r);
}

int main()
{
    int T;
    read(T);
    while(T--)
    {
        init();
        read(n);
        for(int i=1; i<=n; i++)
            read(val[i]);
        for(int i=1,u,v; i<n; i++)
        {
            read(u),read(v);
            addedge(u,v);
            addedge(v,u);
        }
        dfs(1);
        timestamp=0;
        valsum[0]=0;
        getid(1,1);
        int q;
        read(q);
        CLR(sum,0);
        CLR(la,0);
        while(q--)
        {
            int op, x, y;
            read(op),read(x);
            if(op==1)
            {
                read(y);
                update(x,y,n);
            }
            else if(op==2)
                del(dfn[x],1,n);
            else if(op==3)
                update(dfn[x],dfn[x]+size[x]-1,1,n);
            printf("%d\n", sum[getidx(1,n)]);
        }

    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值