题意
给你一棵包含N个节点的树,设每条边一开始的边权为0,现在有两种操作:
1)给出参数U,V,C,表示把U与V之间的路径上的边权变成C(保证C≥0)
2)给出参数U,V,C,表示把U与V之间的路径上的边权加上C。但是如果U至V之间路径某条边的边权加上C小于0,那么C=这条边的边权的相反数。
你需要统计出每次一操作过后树中边权为0的边有多少条。
解析
边权树链剖分。对于操作1直接修改,对于操作2,先检测树链上最小值,看要不要改C,然后再加。每次查的话如果Tn[i]=0就累加答案。注意事项的话就是合并的时候看两边最小值是否一样。还有修改和覆盖的关系要搞清。
#include <cstdio>
#include <algorithm>
#define Rep( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i<=(i##_END);i++)
#define For( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i++)
#define Lop( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i>=(i##_END);i--)
#define Dnt( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i--)
using std :: max;
using std :: min;
const int maxx = 200000 + 25;
const int Inf = (unsigned)(-1) >> 1;
typedef int Array[maxx];
int n,m,x,y,z,num,tot,ans,f;
int head[maxx],to[maxx<<1],nxt[maxx<<1],w[maxx<<1];
int Tn[maxx<<2],cnt[maxx<<2],Add[maxx<<2],st[maxx<<2];
Array top,size,ftr,son,dpt,rnk;
namespace Cute{
void Ins(int x,int y){
to[++num] = y;nxt[num] = head[x];head[x] = num;
}
void Dfs(int x){
size[x] = 1;
for(int i=head[x];i;i=nxt[i]){
if(to[i] == ftr[x]) continue;
dpt[to[i]] = dpt[x] + 1;
ftr[to[i]] = x;
Dfs(to[i]);size[x] += size[to[i]];
if(size[to[i]] > size[son[x]]) son[x] = to[i];
}
}
void __Dfs(int x,int brn){
top[x] = brn;rnk[x] = ++tot;
if(son[x]) __Dfs(son[x],brn);
for(int i=head[x];i;i=nxt[i])
if(to[i] != ftr[x] && to[i] != son[x])
__Dfs(to[i],to[i]);
}
}
namespace Segment_Tree{
void upt(int i){
if(Tn[i<<1] == Tn[i<<1|1]) {Tn[i] = Tn[i<<1];cnt[i] = cnt[i<<1] + cnt[i<<1|1];return;}
Tn[i] = min(Tn[i<<1],Tn[i<<1|1]);
cnt[i] = Tn[i<<1] < Tn[i<<1|1]? cnt[i<<1] : cnt[i<<1|1];
}
void Build(int i,int l,int r){
st[i] = -1;
if(l == r) {Tn[i] = 0;cnt[i] = 1;return;}
int mid = (l+r) >> 1;
Build(i<<1,l,mid);Build(i<<1|1,mid+1,r);
upt(i);
}
void pushdown(int i,int l,int r){
int mid = (l+r) >> 1;
if(st[i] != -1){
st[i<<1] = st[i];Add[i<<1] = 0;Tn[i<<1] = st[i];cnt[i<<1] = (mid-l+1);
st[i<<1|1] = st[i];Add[i<<1|1] = 0;Tn[i<<1|1] = st[i];cnt[i<<1|1] = (r-mid);
st[i] = -1;
}
if(Add[i]){
Add[i<<1] += Add[i];Tn[i<<1] += Add[i];
Add[i<<1|1] += Add[i];Tn[i<<1|1] += Add[i];
Add[i] = 0;
}
}
void modify(int i,int x,int y,int l,int r,int f,int k){
if(x <= l && r <= y){
if(f == 1) {Tn[i] += k;Add[i] += k;return;}
if(f == 2) {st[i] = k;Tn[i] = k;Add[i] = 0;cnt[i] = (r-l+1);return;}
}
int mid = (l+r) >> 1;
pushdown(i,l,r);
if(x <= mid) modify(i<<1,x,y,l,mid,f,k);
if(y > mid) modify(i<<1|1,x,y,mid+1,r,f,k);
upt(i);
}
int Query(int i,int x,int y,int l,int r){
int ans = Inf;
if(x <= l && r <= y) return Tn[i];
int mid = (l+r) >> 1;
pushdown(i,l,r);
if(x <= mid) ans = min(ans,Query(i<<1,x,y,l,mid));
if(y > mid) ans = min(ans,Query(i<<1|1,x,y,mid+1,r));
return ans;
}
int Getcnt(int i,int x,int y,int l,int r){
if(x <= l && r <= y) return Tn[i]==0? cnt[i] : 0;
int ans = 0;
int mid = (l+r) >> 1;
pushdown(i,l,r);
if(x <= mid) ans += Getcnt(i<<1,x,y,l,mid);
if(y > mid) ans += Getcnt(i<<1|1,x,y,mid+1,r);
return ans;
}
}
namespace Sb_Linux{
using namespace Segment_Tree;
int Get(int x,int y){
int ans = Inf;
while(top[x] != top[y]){
if(dpt[top[x]] > dpt[top[y]]) std :: swap(x,y);
ans = min(ans,Query(1,rnk[top[y]],rnk[y],1,n));
y = ftr[top[y]];
}
if(rnk[x] > rnk[y]) std :: swap(x,y);
if(x != y) ans = min(ans,Query(1,rnk[x]+1,rnk[y],1,n));
return ans;
}
void C(int x,int y,int f,int k){
if(f == 1){
int tmp = Get(x,y);
if(tmp + k < 0) k = -tmp;
}
while(top[x] != top[y]){
if(dpt[top[x]] > dpt[top[y]]) std :: swap(x,y);
modify(1,rnk[top[y]],rnk[y],1,n,f,k);
y = ftr[top[y]];
}
if(rnk[x] > rnk[y]) std :: swap(x,y);
if(x != y) modify(1,rnk[x]+1,rnk[y],1,n,f,k);
}
}
using namespace Cute;
using namespace Segment_Tree;
using namespace Sb_Linux;
int main(){
scanf("%d%d",&n,&m);
For( i , 1 , n ) scanf("%d%d",&x,&y),Ins(x,y),Ins(y,x);
Dfs(1);__Dfs(1,1);
Build(1,1,n);
while( m-- ){
scanf("%d%d%d%d",&f,&x,&y,&z);
if(f == 1) C(x,y,2,z);
if(f == 2) C(x,y,1,z);
printf("%d\n",Getcnt(1,2,n,1,n));
}
return 0;
}