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

来源：互联网发布：php $a & 7 变量编辑：程序博客网时间：2024/05/17 22:12
比较不错的题，因为每次都是加一个正数，所以每个数只会由负变正一次。
每次变成正数的时候，暴力修改就可以了。
维护一下区间最大的负数和负数的个数，如果这次没有负数变成正数，那么直接修改，否则暴力修改。
#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;}
0 0