3306 水果姐逛水果街Ⅲ
时间限制: 2 s
空间限制: 256000 KB
题目等级 : 大师 Master
题目描述 Description
连续两天都没有被cgh难倒,水果姐心情很不错,第三天又来逛
水果街。
今天cgh早早就来到了水果街等水果姐,因为他带来了帮手cys。cgh的魔法是把水果街变成树结构,而cys的魔法是瞬间改变某家店的水果价格。一边问一边改,绝不给水果姐思考的时间!
同样还是n家水果店,编号为1~n,每家店能买水果也能卖水果,并且同一家店卖与买的价格一样。
cgh和cys一共有m个操作。
操作的格式为
0 x y 或 1 x y
如果操作类型是0,表示cys把第x家店的价格改为y
如果操作类型是1,则表示要求水果姐从第x家店出发到第y家店,途中只能选一家店买一个水果,然后选一家店(可以是同一家店,但不能往回走)卖出去。并且输出最多可以赚多少钱。
这题是给天牛做的。
输入描述 Input Description
第一行n,表示有n家店
下来n个正整数,表示每家店一个苹果的价格。
下来n-1行,每行两个整数x,y,表示第x家店和第y家店有一条边。
下来一个整数m,表示下来有m个操作。
下来有m行,每行三个整数,表示一次操作。
输出描述 Output Description
每行对应一个1类型的操作,输出一个整数,表示面对cgh的每次询问,水果姐最多可以赚到多少钱。
样例输入 Sample Input
4
16 5 1 15
1 2
1 3
2 4
3
1 3 4
0 3 17
1 4 3
样例输出 Sample Output
15
12
数据范围及提示 Data Size & Hint
0<=苹果的价格<=10^8
0<n<=2*10^5
0<m<=10^5
题解:
因为有修改,所以写了树链剖分。查询的时候和写Ⅱ的时候有异曲同工之妙。。一样恶心。.
Code:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 200100
#define root 1,1,tot
#define lchild rt<<1,l,m
#define rchild rt<<1|1,m+1,r
using namespace std;
struct Edge{
int v,next;
}edge[2*N];
struct Tree{
int maxn,minn,cq,ch;
}tree[4*N];
int n,q,num=0,tot=0;
int head[N],fa[N],de[N],dv[N],sz[N],son[N],w[N],top[N];
bool vis[N]={0};
int in(){
int x=0; char ch=getchar();
while (ch<'0' || ch>'9') ch=getchar();
while (ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}
void add(int u,int v){
edge[++num].v=v; edge[num].next=head[u]; head[u]=num;
}
void dfs1(int x){
sz[x]=1; vis[x]=true;
for (int i=head[x]; i; i=edge[i].next){
int v=edge[i].v;
if (vis[v]) continue;
fa[v]=x; de[v]=de[x]+1;
dfs1(v); sz[x]+=sz[v];
if (sz[son[x]]<sz[v]) son[x]=v;
}
}
void dfs2(int x,int start){
w[x]=++tot; top[x]=start;
if (!son[x]) return; dfs2(son[x],start);
for (int i=head[x]; i; i=edge[i].next)
if (edge[i].v!=fa[x] && edge[i].v!=son[x])
dfs2(edge[i].v,edge[i].v);
}
void push_up(int rt){
tree[rt].maxn=max(tree[rt<<1].maxn,tree[rt<<1|1].maxn);
tree[rt].minn=min(tree[rt<<1].minn,tree[rt<<1|1].minn);
tree[rt].cq=max(max(tree[rt<<1].cq,tree[rt<<1|1].cq),tree[rt<<1|1].maxn-tree[rt<<1].minn);
tree[rt].ch=max(max(tree[rt<<1].ch,tree[rt<<1|1].ch),tree[rt<<1].maxn-tree[rt<<1|1].minn);
}
void update(int rt,int l,int r,int x,int p){
if (l==r){
tree[rt].maxn=tree[rt].minn=p;
tree[rt].cq=tree[rt].ch=0;
return;
}
int m=(l+r)>>1;
if (x<=m) update(lchild,x,p);
else update(rchild,x,p);
push_up(rt);
}
Tree find(int rt,int l,int r,int ll,int rr){
if (ll<=l && r<=rr) return (Tree)tree[rt];
int m=(l+r)>>1; Tree s1,s2,s3;
s1=s2=s3=(Tree){0,0x7fffffff,0,0};
if (ll<=m) s1=find(lchild,ll,rr);
if (rr>m) s2=find(rchild,ll,rr);
s3.maxn=max(s1.maxn,s2.maxn);
s3.minn=min(s1.minn,s2.minn);
s3.cq=max(max(s1.cq,s2.cq),s2.maxn-s1.minn);
s3.ch=max(max(s1.ch,s2.ch),s1.maxn-s2.minn);
return (Tree)s3;
}
void build(){
for (int i=1; i<=n; i++)
update(root,w[i],dv[i]);
}
int query(int x,int y){
int f1=top[x],f2=top[y],sx=0,sy=0,maxx=0,maxy=0,minx=0x7fffffff,miny=0x7fffffff;
Tree ss=(Tree){0,0x7fffffff,0,0};
while (f1!=f2){
if (de[f1]>de[f2]){
ss=find(root,w[f1],w[x]);
sx=max(sx,max(ss.ch,ss.maxn-minx));
maxx=max(maxx,ss.maxn); minx=min(minx,ss.minn);
x=fa[f1]; f1=top[x];
}
else {
ss=find(root,w[f2],w[y]);
sy=max(sy,max(ss.cq,maxy-ss.minn));
maxy=max(maxy,ss.maxn); miny=min(miny,ss.minn);
y=fa[f2]; f2=top[y];
}
}
if (de[x]>de[y]){
ss=find(root,w[y],w[x]);
sx=max(sx,max(ss.ch,ss.maxn-minx));
maxx=max(maxx,ss.maxn); minx=min(minx,ss.minn);
}
else {
ss=find(root,w[x],w[y]);
sy=max(sy,max(ss.cq,maxy-ss.minn));
maxy=max(maxy,ss.maxn); miny=min(miny,ss.minn);
}
return max(max(sx,sy),maxy-minx);
}
int main(){
n=in();
for (int i=1; i<=n; i++) dv[i]=in();
for (int i=1; i<n; i++){
int u=in(),v=in();
add(u,v),add(v,u);
}
dfs1(1); dfs2(1,1); build();
q=in();
while (q--){
int opt=in(),x=in(),y=in();
if (!opt) update(root,w[x],y);
else printf("%d\n",query(x,y));
}
return 0;
}