POJ 3237 Tree(ツリー分割+セグメントツリー)
14433 ワード
Descriptionの1本の木にはn個のノードがあり、n-1個のエッジがあり、各エッジに1個の重み値があり、3種類の操作CHANGE i v:i番目のエッジの重み値をv NEGATE a bに変更する:aからbまでのパス上のすべてのエッジの重み値を逆(x->-x)QUERY a bにする:aからbまでのパス上のすべてのエッジの重み値の最大値Inputを問い合わせる第1行の整数tは用例グループ数を表す.各組の使用例の第1行の1つの整数nは点数を表し、その後n-1行の各行の3つの整数a b cはa点とb点の間にcの重み値を持つ辺を表し、その後数行の各行に1回の操作を行い、DONEでこの組の使用例入力Outputがクエリ毎に終了する.クエリー結果Sample Input 1 3 1 1 2 2 2 3 QUERY 1 2 CHANGE 1 3 QUERY 1 2 DONE Sample Output 1 3 Solutionツリー上のパスの問題を出力し、まずツリーチェーンを分割し、次にセグメントツリーを構築します.その後、線分ツリーの単点修正、区間更新、区間クエリーの問題が発生しましたCode
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 111111
#define INF 0x3f3f3f3f
struct Edge
{
int to,next;
}E[2*maxn];
struct Tree
{
int left,right,Min,Max,flag;//flag
}T[4*maxn];
int t,n,Q,e[maxn][3],head[maxn],cnt,idx,size[maxn],fa[maxn],son[maxn],dep[maxn],top[maxn],id[maxn];
void init()
{
cnt=idx=0;
memset(head,-1,sizeof(head));
dep[1]=fa[1]=size[0]=0;
memset(son,0,sizeof(son));
}
void add(int u,int v)
{
E[cnt].to=v;
E[cnt].next=head[u];
head[u]=cnt++;
}
void dfs1(int u)
{
size[u]=1;
for(int i=head[u];~i;i=E[i].next)
{
int v=E[i].to;
if(v!=fa[u])
{
fa[v]=u;
dep[v]=dep[u]+1;
dfs1(v);
size[u]+=size[v];
if(size[son[u]]<size[v]) son[u]=v;
}
}
}
void dfs2(int u,int topu)
{
top[u]=topu;
id[u]=++idx;
if(son[u]) dfs2(son[u],top[u]);//
for(int i=head[u];~i;i=E[i].next)
{
int v=E[i].to;
if(v!=fa[u]&&v!=son[u]) dfs2(v,v);
}
}
void build(int l,int r,int t)
{
T[t].left=l;
T[t].right=r;
T[t].Min=T[t].Max=T[t].flag=0;
if(l==r) return ;
int mid=(l+r)>>1;
build(l,mid,2*t);
build(mid+1,r,2*t+1);
}
void push_up(int t)
{
T[t].Min=min(T[2*t].Min,T[2*t+1].Min);
T[t].Max=max(T[2*t].Max,T[2*t+1].Max);
}
void push_down(int t)
{
if(T[t].left==T[t].right)return ;
if(T[t].flag)
{
int Min,Max;
Min=T[2*t].Min,Max=T[2*t].Max;
T[2*t].Max=-Min,T[2*t].Min=-Max;
Min=T[2*t+1].Min,Max=T[2*t+1].Max;
T[2*t+1].Max=-Min,T[2*t+1].Min=-Max;
T[2*t].flag^=1,T[2*t+1].flag^=1;
T[t].flag=0;
}
}
void update(int x,int v,int t)// x v
{
if(T[t].left==x&&T[t].right==x)
{
T[t].Min=T[t].Max=v;
T[t].flag=0;
return ;
}
push_down(t);
int mid=(T[t].left+T[t].right)>>1;
if(x<=mid) update(x,v,2*t);
else update(x,v,2*t+1);
push_up(t);
}
int query(int l,int r,int t)// [l,r]
{
if(T[t].left==l&&T[t].right==r)
return T[t].Max;
push_down(t);
int mid=(T[t].left+T[t].right)>>1;
if(r<=mid) return query(l,r,2*t);
else if(l>mid) return query(l,r,2*t+1);
else return max(query(l,mid,2*t),query(mid+1,r,2*t+1));
push_up(t);
}
void reverse(int l,int r,int t)// [l,r]
{
if(T[t].left==l&&T[t].right==r)
{
int Min,Max;
Min=T[t].Min,Max=T[t].Max;
T[t].Max=-Min,T[t].Min=-Max;
T[t].flag^=1;
return ;
}
push_down(t);
int mid=(T[t].left+T[t].right)>>1;
if(r<=mid) reverse(l,r,2*t);
else if(l>mid) reverse(l,r,2*t+1);
else
{
reverse(l,mid,2*t);
reverse(mid+1,r,2*t+1);
}
push_up(t);
}
int Get_max(int u,int v)// u->v
{
int top1=top[u],top2=top[v],ans=-INF;
while(top1!=top2)
{
if(dep[top1]<dep[top2])
{
swap(top1,top2);
swap(u,v);
}
ans=max(ans,query(id[top1],id[u],1));
u=fa[top1];
top1=top[u];
}
if(u==v)return ans;
if(dep[u]>dep[v])swap(u,v);
return max(ans,query(id[son[u]],id[v],1));
}
void Negate(int u,int v)// u->v
{
int top1=top[u],top2=top[v];
while(top1!=top2)
{
if(dep[top1]<dep[top2])
{
swap(top1,top2);
swap(u,v);
}
reverse(id[top1],id[u],1);
u=fa[top1];
top1=top[u];
}
if(u==v)return ;
if(dep[u]>dep[v])swap(u,v);
return reverse(id[son[u]],id[v],1);
}
int main()
{
scanf("%d",&t);
while(t--)
{
init();//
scanf("%d",&n);
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&e[i][0],&e[i][1],&e[i][2]);
//
add(e[i][0],e[i][1]);
add(e[i][1],e[i][0]);
}
dfs1(1);// dep,size,fa,son
dfs2(1,1);// top,id
build(1,idx,1);//
for(int i=1;i<n;i++)
{
if(dep[e[i][0]]>dep[e[i][1]])// , id[i] i
swap(e[i][0],e[i][1]);
update(id[e[i][1]],e[i][2],1);
}
char op[11];
while(~scanf("%s",op),op[0]!='D')
{
int u,v;
scanf("%d%d",&u,&v);
if(op[0]=='C')//
update(id[e[u][1]],v,1);
else if(op[0]=='N')//
Negate(u,v);
else//
printf("%d
",Get_max(u,v));
}
}
return 0;
}