【bzoj4127】Abs 线段树+树链剖分

比较不错的题,因为每次都是加一个正数,所以每个数只会由负变正一次。

每次变成正数的时候,暴力修改就可以了。

维护一下区间最大的负数和负数的个数,如果这次没有负数变成正数,那么直接修改,否则暴力修改。


#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#define maxn 100010 
#define inf 1000000000
 
using namespace std;
 
struct yts
{
    int l,r;
    long long ans,tag,data;
    long long mx;
    int cnt;
}t[4*maxn];
 
int w[maxn],rank[maxn],seq[maxn];
int fa[maxn][20],dep[maxn],g[maxn],size[maxn],dd[maxn];
bool vis[maxn];
int next[2*maxn],st[maxn],head[maxn],to[2*maxn],a[maxn],e[maxn];
int n,m,tot,num,T;
 
void addedge(int x,int y)
{
    num++;to[num]=y;next[num]=st[x];st[x]=num;
}
 
void dfs(int x)
{
    e[++tot]=x;
    size[x]=1;dd[x]=0;
    for (int p=st[x];p;p=next[p])
      if (to[p]!=g[x])
      {
        g[to[p]]=x;dep[to[p]]=dep[x]+1;
        dfs(to[p]);
        if (size[to[p]]>size[dd[x]]) dd[x]=to[p];
        size[x]+=size[to[p]];
      }
}
 
int go_up(int x,int d)
{
    if (d<0) return 0;
    for (int i=0;i<=17;i++)
      if (d&(1<<i)) x=fa[x][i];
    return x;
}
 
int LCA(int x,int y)
{
    if (dep[x]>dep[y]) x=go_up(x,dep[x]-dep[y]);
    else y=go_up(y,dep[y]-dep[x]);
    if (x==y) return x;
    for (int i=17;i>=0;i--)
      if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return g[x];
}
 
void add(int i,long long d)
{
    t[i].tag+=d;
    t[i].ans+=((long long)t[i].r-t[i].l+1-2*t[i].cnt)*d;
    if (t[i].mx && t[i].mx+d>=0) printf("Wrong %lld %lld\n",t[i].mx,d);
    if (t[i].mx) t[i].mx+=d;
}
 
void release(int i)
{
    if (t[i].l==t[i].r) return;
    if (t[i].tag)
    {
        add(i*2,t[i].tag);
        add(i*2+1,t[i].tag);
        t[i].tag=0;
    }
}
 
int gmax(int x,int y)
{
    if (x>=0) return y;
    if (y>=0) return x;
    return x>y?x:y;
}
 
void update(int i)
{
    t[i].ans=t[i*2].ans+t[i*2+1].ans;
    t[i].cnt=t[i*2].cnt+t[i*2+1].cnt;
    t[i].mx=gmax(t[i*2].mx,t[i*2+1].mx);
}
 
void build(int i,int l,int r)
{
    t[i].l=l;t[i].r=r;
    if (l==r) 
    {
        t[i].ans=abs(a[l]);
        t[i].cnt=(a[l]<0?1:0);
        t[i].mx=min(a[l],0);
        return;
    }
    int mid=(l+r)/2;
    if (l<=mid) build(i*2,l,mid);
    if (mid<r) build(i*2+1,mid+1,r);
    update(i);
}
 
void dfs(int i,int d)
{
    if (t[i].mx==0 || t[i].mx+d<0) {add(i,d);return;}
    if (t[i].l==t[i].r)
    {
        t[i].ans=t[i].mx+d;
        t[i].cnt=0;
        t[i].mx=0;
        return;
    }
    release(i);
    dfs(i*2,d);dfs(i*2+1,d);
    update(i);
}
 
void modify(int i,int l,int r,int d)
{
    if (l<=t[i].l && t[i].r<=r)
    {
        dfs(i,d);
        return;
    }
    release(i);
    int mid=(t[i].l+t[i].r)/2;
    if (l<=mid) modify(i*2,l,r,d);
    if (mid<r) modify(i*2+1,l,r,d);
    update(i);
}
  
long long query(int i,int l,int r)
{
    if (l<=t[i].l && t[i].r<=r) return t[i].ans;
    release(i);
    int mid=(t[i].l+t[i].r)/2;
    long long ans=0;
    if (l<=mid) ans+=query(i*2,l,r);
    if (mid<r) ans+=query(i*2+1,l,r);
    update(i);
    return ans;
}
 
int main()
{
    scanf("%d%d",&n,&T);
    for (int i=1;i<=n;i++) scanf("%d",&w[i]);
    num=0;
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);
    }
    dfs(1);
    memset(vis,0,sizeof(vis));
    int qwer=0;
    for (int i=1;i<=n;i++)
      if (!vis[e[i]])
      {
        int k=e[i];
        while (k)
        {
            seq[++qwer]=k;vis[k]=1;head[k]=e[i];k=dd[k];
        }
      }
    for (int i=1;i<=n;i++) rank[seq[i]]=i;
    for (int i=1;i<=n;i++) a[rank[i]]=w[i];
    build(1,1,n);
    for (int i=1;i<=n;i++) fa[i][0]=g[i];
    for (int j=1;j<=17;j++)
      for (int i=1;i<=n;i++)
        fa[i][j]=fa[fa[i][j-1]][j-1];
    while (T--)
    {
        int op,x,y,z;
        scanf("%d%d%d",&op,&x,&y);
        if (op==1)
        {
            scanf("%d",&z);
            int lca=LCA(x,y),w=go_up(y,dep[y]-dep[lca]-1);
            while (dep[head[x]]>dep[lca])
            {
                modify(1,rank[head[x]],rank[x],z);x=g[head[x]];
            }
            modify(1,rank[lca],rank[x],z);
            while (dep[head[y]]>dep[lca])
            {
                modify(1,rank[head[y]],rank[y],z);y=g[head[y]];
            }
            if (w) modify(1,rank[w],rank[y],z);
        }
        else
        {
            long long ans=0;
            int lca=LCA(x,y);
            while (dep[head[x]]>dep[lca])
            {
                ans+=query(1,rank[head[x]],rank[x]);x=g[head[x]];
            }
            ans+=query(1,rank[lca],rank[x]);
            while (dep[head[y]]>dep[lca])
            {
                ans+=query(1,rank[head[y]],rank[y]);y=g[head[y]];
            }
            ans+=query(1,rank[lca],rank[y]);
            ans-=query(1,rank[lca],rank[lca]);
            printf("%lld\n",ans);
        }
    }
    return 0;
}


你可能感兴趣的:(【bzoj4127】Abs 线段树+树链剖分)