poj3237 Tree [树链剖分]

这个题是spoj的改版

是在原来的题意上增加了区间取反操作

所以只需要在spoj375的基础上再线段树上增加一个取反标志

同时在维护一个区间最小值

因为在区间取反了以后 区间的最大值就是区间原来的最小值

嗯 就这样就可以了

#include <iostream>#include <stdio.h>#include <string.h>using namespace std;struct Node{int next,num;}edge[20001];struct Node1{int x,y,v;}a[10001];int cnt;int dep[10001],num[10001],fa[10001],head[10001],siz[10001],son[10001],topfa[10001],id[10001];void add(int x,int y){edge[cnt].num=y;edge[cnt].next=head[x];head[x]=cnt++;edge[cnt].num=x;edge[cnt].next=head[y];head[y]=cnt++;}void dfs1(int x,int f,int d){dep[x]=d;fa[x]=f;siz[x]=1;son[x]=0;for(int i=head[x];i!=-1;i=edge[i].next){int v=edge[i].num;if(v==f) continue;dfs1(v,x,d+1);siz[x]+=siz[v];if(siz[son[x]]<siz[v]) son[x]=v;}}int nu;void dfs2(int x,int f){topfa[x]=f;id[x]=++nu;if(son[x]) dfs2(son[x],f);for(int i=head[x];i!=-1;i=edge[i].next){int v=edge[i].num;if(v==fa[x]||v==son[x]) continue;dfs2(v,v);}}struct Tree{int l,r,flag,min,max;}t[40001];int max(int x,int y){if(x<y) return y;return x;}int min(int x,int y){if(x>y) return y;return x;}void swap(int &x,int &y){int c=x;x=y;y=c;}void build(int l,int r,int x){//cout << l << " " << r << " " << x << endl;int ls=x<<1,rs=ls|1;t[x].l=l,t[x].r=r;t[x].flag=0;if(l==r) {t[x].max=num[l];t[x].min=num[l];return;}int mid=(l+r)>>1;build(l,mid,ls);build(mid+1,r,rs);t[x].max=max(t[ls].max,t[rs].max);t[x].min=min(t[ls].min,t[rs].min);}void updata(int x,int ls,int rs){if(t[ls].flag){t[x].max=-t[ls].min;t[x].min=-t[ls].max;}else{t[x].max=t[ls].max;t[x].min=t[ls].min;}if(t[rs].flag){t[x].max=max(t[x].max,-t[rs].min);t[x].min=min(t[x].min,-t[rs].max);}else{t[x].max=max(t[x].max,t[rs].max);t[x].min=min(t[x].min,t[rs].min);}}void change(int l,int v,int x){int ls=x<<1,rs=ls|1;if(l==t[x].l&&l==t[x].r) { t[x].flag=0;t[x].max=v;t[x].min=v; return;}if(t[x].flag) {t[ls].flag^=1;t[rs].flag^=1;t[x].flag=0;}int mid=(t[x].l+t[x].r)>>1;if(l<=mid) change(l,v,ls);if(l>mid) change(l,v,rs);updata(x,ls,rs);}void change_flag(int l,int r,int x){int ls=x<<1,rs=ls|1;if(l==t[x].l&&r==t[x].r) {t[x].flag^=1;return;}int mid=(t[x].l+t[x].r)>>1;if(t[x].flag) {t[ls].flag^=1;t[rs].flag^=1;t[x].flag=0;}if(r<=mid) change_flag(l,r,ls);else if(l>mid) change_flag(l,r,rs);else{change_flag(l,mid,ls);change_flag(mid+1,r,rs);}updata(x,ls,rs);}int tree_query(int l,int r,int x){ int ls=x<<1,rs=ls|1; if(l==t[x].l&&r==t[x].r){ if(t[x].flag) return -t[x].min;else return t[x].max; }int mid=(t[x].l+t[x].r)>>1;int ans;if(t[x].flag) {t[ls].flag^=1;t[rs].flag^=1;t[x].flag=0;}if(mid>=r) ans=tree_query(l,r,ls);else if(mid<l) ans=tree_query(l,r,rs);else ans=max(tree_query(l,mid,ls),tree_query(mid+1,r,rs));updata(x,ls,rs);return ans;}void flag(int x,int y){if(x==y) return;if(dep[x]>dep[y]) swap(x,y);if(topfa[x]==topfa[y]) {change_flag(id[x]+1,id[y],1);return;}if(dep[topfa[y]]>dep[topfa[x]]) swap(x,y);change_flag(id[topfa[x]],id[x],1);flag(fa[topfa[x]],y);}int query(int x,int y){if(x==y)  return -1234567890;if(dep[x]>dep[y]) swap(x,y);if(topfa[x]==topfa[y]) return tree_query(id[x]+1,id[y],1);if(dep[topfa[x]]<dep[topfa[y]]) swap(x,y);return max(tree_query(id[topfa[x]],id[x],1),query(fa[topfa[x]],y));}int main(){int T,n;scanf("%d",&T);while(T--){nu=0;cnt=0;memset(head,-1,sizeof(head));scanf("%d",&n);for(int i=1;i<n;i++){scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].v);add(a[i].x,a[i].y);}dfs1(1,0,1);dfs2(1,1);for(int i=1;i<n;i++){if(fa[a[i].y]==a[i].x) swap(a[i].x,a[i].y);num[id[a[i].x]]=a[i].v;}build(1,n,1);char s[10];int x,y;while(1){scanf("%s",&s);if(s[0]=='D') break;scanf("%d%d",&x,&y);if(s[0]=='C'){change(id[a[x].x],y,1);}else if(s[0]=='N'){flag(x,y);}else printf("%d\n",query(x,y));//cout << s[0] << endl;}}}