比较不错的题,因为每次都是加一个正数,所以每个数只会由负变正一次。
每次变成正数的时候,暴力修改就可以了。
维护一下区间最大的负数和负数的个数,如果这次没有负数变成正数,那么直接修改,否则暴力修改。
#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;
}