1. 程式人生 > >洛谷P1505 [國家集訓隊]旅遊

洛谷P1505 [國家集訓隊]旅遊

題目描述

\(Ray\) 樂忠於旅遊,這次他來到了\(T\) 城。\(T\) 城是一個水上城市,一共有 \(N\) 個景點,有些景點之間會用一座橋連線。為了方便遊客到達每個景點但又為了節約成本,\(T\) 城的任意兩個景點之間有且只有一條路徑。換句話說, \(T\) 城中只有\(N − 1\) 座橋。

\(Ray\) 發現,有些橋上可以看到美麗的景色,讓人心情愉悅,但有些橋狹窄泥濘,令人煩躁。於是,他給每座橋定義一個愉悅度\(w\),也就是說,\(Ray\) 經過這座橋會增加\(w\) 的愉悅度,這或許是正的也可能是負的。有時,\(Ray\) 看待同一座橋的心情也會發生改變。

現在,\(Ray\)

想讓你幫他計算從\(u\) 景點到\(v\) 景點能獲得的總愉悅度。有時,他還想知道某段路上最美麗的橋所提供的最大愉悅度,或是某段路上最糟糕的一座橋提供的最低愉悅度。

輸入輸出格式

輸入格式:

輸入的第一行包含一個整數\(N\),表示\(T\) 城中的景點個數。景點編號為 \(0...N − 1\)

接下來\(N − 1\) 行,每行三個整數\(u\)\(v\)\(w\),表示有一條u 到\(v\),使 \(Ray\) 愉悅度增加\(w\) 的橋。橋的編號為\(1...N − 1\)\(|w| \leq 1000\)。 輸入的第\(N + 1\) 行包含一個整數\(M\),表示\(Ray\)

的運算元目。

接下來有\(M\) 行,每行描述了一個操作,操作有如下五種形式:

\(C\) \(i\) \(w\),表示\(Ray\) 對於經過第\(i\) 座橋的愉悅度變成了\(w\)

\(N\) \(u\) \(v\),表示\(Ray\) 對於經過景點\(u\)\(v\) 的路徑上的每一座橋的愉悅度都變成原來的相反數。

\(SUM\) \(u\) \(v\),表示詢問從景點\(u\)\(v\) 所獲得的總愉悅度。

\(MAX\) \(u\) \(v\),表示詢問從景點\(u\)\(v\) 的路徑上的所有橋中某一座橋所提供的最大愉悅度。

\(MIN\) \(u\)

\(v\),表示詢問從景點\(u\)\(v\) 的路徑上的所有橋中某一座橋所提供的最小愉悅度。

測試資料保證,任意時刻,\(Ray\) 對於經過每一座橋的愉悅度的絕對值小於等於\(1000\)

輸出格式:

對於每一個詢問(操作\(S\)\(MAX\)\(MIN\)),輸出答案。

輸入輸出樣例

輸入樣例#1:

3
0 1 1
1 2 2
8
SUM 0 2
MAX 0 2
N 0 1
SUM 0 2
MIN 0 2
C 1 3
SUM 0 2
MAX 0 2

輸出樣例#1:

3
2
1
-1
5
3

說明

很容易的基礎題哦>.<

思路:如果說這道題是點權的話,那麼就是一道樹鏈剖分的板子題,但是,這道題是邊權,那怎麼辦呢?可以發現,每個兒子只有一個父親,那麼我們就可以用這個兒子的點權來代替它與它父親之間的邊權,然後用樹鏈剖分+線段樹維護最大值,最小值和區間和即可。

程式碼:

#include<cstdio>
#include<algorithm>
#include<cctype>
#define maxn 200007
#define ls rt<<1
#define rs rt<<1|1
using namespace std;
const int inf=0x7fffffff;
int n,m,num,head[maxn],a[maxn],size[maxn],d[maxn],top[maxn];
int cnt,sum[maxn<<2],lazy[maxn<<2],maxx[maxn<<2],minn[maxn<<2];
int fa[maxn],id[maxn],zrj[maxn],son[maxn];
char s1[8];
inline int qread() {
  char c=getchar();int num=0,f=1;
  for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
  for(;isdigit(c);c=getchar()) num=num*10+c-'0';
  return num*f;
}
struct node {
  int v,w,nxt;
}e[maxn<<1];
inline void ct(int u, int v, int w) {
  e[++num].v=v;
  e[num].w=w;
  e[num].nxt=head[u];
  head[u]=num;
}
void dfs1(int u) {
  size[u]=1;
  for(int i=head[u];i;i=e[i].nxt) {
    int v=e[i].v;
    if(v!=fa[u]) {
      d[v]=d[u]+1;
      fa[v]=u;
      zrj[v]=e[i].w;
      dfs1(v);
      size[u]+=size[v];
      if(size[v]>size[son[u]]) son[u]=v;
    }
  }
}
void dfs2(int u, int t) {
  id[u]=++cnt;
  top[u]=t;
  a[cnt]=zrj[u];
  if(son[u]) dfs2(son[u],t);
  for(int i=head[u];i;i=e[i].nxt) {
    int v=e[i].v;
    if(v!=fa[u]&&v!=son[u]) dfs2(v,v);
  }
}
inline void pushup(int rt) {
  sum[rt]=sum[ls]+sum[rs];
  maxx[rt]=max(maxx[ls],maxx[rs]);
  minn[rt]=min(minn[ls],minn[rs]);
}
inline void pushdown(int rt) {
  if(lazy[rt]) {
    sum[ls]=-sum[ls],lazy[ls]^=1;
    sum[rs]=-sum[rs],lazy[rs]^=1;
    int t1=maxx[ls],t2=maxx[rs],s1=minn[ls],s2=minn[rs];
    maxx[ls]=-s1,maxx[rs]=-s2,minn[ls]=-t1,minn[rs]=-t2;
    lazy[rt]=0;
  }
}
void build(int rt, int l, int r) {
  if(l==r) {
    sum[rt]=maxx[rt]=minn[rt]=a[l];
    return;
  }
  int mid=(l+r)>>1;
  build(ls,l,mid);
  build(rs,mid+1,r);
  pushup(rt);
}
void add(int rt, int l, int r, int L, int val) {
  if(l==r) {
    sum[rt]=maxx[rt]=minn[rt]=val;
    return;
  }
  pushdown(rt);
  int mid=(l+r)>>1;
  if(L<=mid) add(ls,l,mid,L,val);
  else add(rs,mid+1,r,L,val);
  pushup(rt);
}
void modify(int rt, int l, int r, int L, int R) {
  if(L>r||R<l) return;
  if(L<=l&&r<=R) {
    sum[rt]=-sum[rt],lazy[rt]^=1;
    int t=maxx[rt],s=minn[rt];
    maxx[rt]=-s,minn[rt]=-t;
    return;
  } 
  int mid=(l+r)>>1;
  pushdown(rt);
  modify(ls,l,mid,L,R),modify(rs,mid+1,r,L,R);
  pushup(rt);
}
int csum(int rt, int l, int r, int L, int R) {
  if(L>r||R<l) return 0;
  if(L<=l&&r<=R) return sum[rt];
  int mid=(l+r)>>1;
  pushdown(rt);
  return csum(ls,l,mid,L,R)+csum(rs,mid+1,r,L,R);
}
int cmax(int rt, int l, int r, int L, int R) {
  if(L>r||R<l) return -inf;
  if(L<=l&&r<=R) return maxx[rt];
  int mid=(l+r)>>1,ans=-inf;
  pushdown(rt);
  if(L<=mid) ans=max(ans,cmax(ls,l,mid,L,R));
  if(R>mid) ans=max(ans,cmax(rs,mid+1,r,L,R));
  return ans;
}
int cmin(int rt, int l, int r, int L, int R) {
  if(L>r||R<l) return inf;
  if(L<=l&&r<=R) return minn[rt];
  int mid=(l+r)>>1,ans=inf;
  pushdown(rt);
  if(L<=mid) ans=min(ans,cmin(ls,l,mid,L,R));
  if(R>mid) ans=min(ans,cmin(rs,mid+1,r,L,R));
  return ans;
}
void cal(int x, int y) {
  int fx=top[x],fy=top[y];
  while(fx!=fy) {
    if(d[fx]<d[fy]) swap(x,y),swap(fx,fy);
    modify(1,1,cnt,id[fx],id[x]);
    x=fa[fx],fx=top[x];
  }
  if(id[x]>id[y]) swap(x,y);
  modify(1,1,cnt,id[x]+1,id[y]);
}
int query_max(int x, int y) {
  int fx=top[x],fy=top[y],ans=-inf;
  while(fx!=fy) {
    if(d[fx]<d[fy]) swap(x,y),swap(fx,fy);
    ans=max(ans,cmax(1,1,cnt,id[fx],id[x]));
    x=fa[fx],fx=top[x];
  }
  if(id[x]>id[y]) swap(x,y);
  ans=max(ans,cmax(1,1,cnt,id[x]+1,id[y]));
  return ans;
}
int query_min(int x, int y) {
  int fx=top[x],fy=top[y],ans=inf;
  while(fx!=fy) {
    if(d[fx]<d[fy]) swap(x,y),swap(fx,fy);
    ans=min(ans,cmin(1,1,cnt,id[fx],id[x]));
    x=fa[fx],fx=top[x];
  }
  if(id[x]>id[y]) swap(x,y);
  ans=min(ans,cmin(1,1,cnt,id[x]+1,id[y]));
  return ans;
}
int query_sum(int x, int y) {
  int fx=top[x],fy=top[y],ans=0;
  while(fx!=fy) {
    if(d[fx]<d[fy]) swap(x,y),swap(fx,fy);
    ans+=csum(1,1,cnt,id[fx],id[x]);
    x=fa[fx],fx=top[x];
  }
  if(id[x]>id[y]) swap(x,y);
  ans+=csum(1,1,cnt,id[x]+1,id[y]);
  return ans;
}
int main() {
  n=qread();
  for(int i=1,u,v,w;i<n;++i) {
    u=qread()+1,v=qread()+1,w=qread();
    ct(u,v,w);ct(v,u,w);
  }
  dfs1(1);dfs2(1,1);build(1,1,n);
  m=qread();
  for(int i=1,x,y;i<=m;++i) {
    scanf("%s",s1);x=qread()+1,y=qread()+1;
    if(s1[0]=='C') add(1,1,n,id[x],y-1);
    if(s1[0]=='N') cal(x,y);
    if(s1[0]=='S') printf("%d\n",query_sum(x,y));
    if(s1[1]=='I') printf("%d\n",query_min(x,y));
    if(s1[1]=='A') printf("%d\n",query_max(x,y));
  }
  return 0;
}