比赛 20140425 评测结果 AAWWWWWW
题目名称 大话西游 最终得分 25
用户昵称 GDFRWMY 运行时间 0.880 s
代码语言 C++ 内存使用 34.65 MiB
提交时间 2014-04-25 12:46:20
显示代码纯文本
#include<fstream>
#include<algorithm>
#include<vector>
using namespace std;
ifstream fin("westward.in");
ofstream fout("westward.out");
int ll[500000],rr[500000],lc[500000],rc[500000];
long long minw[500000],maxw[500000];
int dfn[500000],qwe[500000],pos[500000],deep[500000];
int edges[500000],edgex[500000];
int n,q,tot,tol,w[500000],u,v;
vector<int> c[500000];
void build(int root,int l,int r){
    ll[root]=l;
	rr[root]=r;
	if (l<r){
		tot++;
		lc[root]=tot;
		build(tot,l,(l+r)/2);
		tot++;
		rc[root]=tot;
		build(tot,(l+r)/2+1,r);
		if (lc[root]!=-1){
			if (minw[lc[root]]>minw[rc[root]])
			minw[root]=minw[rc[root]]; else minw[root]=minw[lc[root]]; 
		if (maxw[lc[root]]>maxw[rc[root]])
			maxw[root]=maxw[lc[root]]; else maxw[root]=maxw[rc[root]]; 
		//fout<<minw[lc[root]]<<' '<<minw[rc[root]]<<endl;
		}
	}
	else
	{lc[root]=-1;
	rc[root]=-1;
	minw[root]=maxw[root]=w[qwe[l]];
	//fout<<maxw[root]<<endl;
	}
	
	
}
void change(int root,int x,int p){
	
	if (root==-1) return;
		//fout<<ll[root]<<' '<<rr[root]<<x<<p<<endl;
	if (ll[root]>x||rr[root]<x) return;

	if (ll[root]==x&&rr[root]==x){
		minw[root]=p;
		maxw[root]=p;
	}
	else{
		change(lc[root],x,p);
		change(rc[root],x,p);
		if (lc[root]!=-1){
		if (minw[lc[root]]>minw[rc[root]])
			minw[root]=minw[rc[root]]; else minw[root]=minw[lc[root]]; 
		if (maxw[lc[root]]>maxw[rc[root]])
			maxw[root]=maxw[lc[root]]; else maxw[root]=maxw[lc[root]]; 
		}
	}
	//fout<<maxw[root]<<endl;
	
}

pair<long long,long long> gets (int root,int x,int y){
	if(root==-1) return make_pair(0,99999999);
	
	if(ll[root]>y||rr[root]<x) return make_pair(0,99999999);
    if(ll[root]>=x&&rr[root]<=y) return make_pair(maxw[root],minw[root]);

 pair <long long,long long> x1,x2;
	x1=gets(lc[root],x,y);
x2=gets(rc[root],x,y);
	//fout<<x1.second<<' '<<x2.second<<endl;

return make_pair(max(x1.first,x2.first),min(x1.second,x2.second));
}
long long query(int x){
int	u=edges[x],v=edgex[x];

pair <long long,long long>  w1,t1,t2,w2;
w1=gets(1,dfn[v],dfn[v]+pos[v]-1);
t1=gets(1,1,dfn[v]-1);
t2=gets(1,dfn[v]+pos[v],n);
w2=make_pair(max(t1.first,t2.first),min(t1.second,t2.second));
//fout<<dfn[v]<<' ' <<pos[v]-1<<endl;
return w1.first*w1.second+w2.first*w2.second;
}
	
void dfs(int x,int fa){
	tol++;
	dfn[x]=tol;
	qwe[tol]=x;
	//fout<<qwe[tol]<<endl;
	for(int i=0;i<c[x].size();i++)
		if(c[x][i]!=fa){
			deep[c[x][i]]=deep[x]+1;
			dfs(c[x][i],x);
			pos[x]+=pos[c[x][i]];
			//fout<<deep[c[x][i]]<<endl;
		}
		pos[x]++;
}
int main(){
	fin>>n>>q;
	for(int i=1;i<=n;i++)
		fin>>w[i];
	for(int i=1;i<n;i++){
		fin>>u>>v;
		edges[i]=u;
		edgex[i]=v;
		c[u].push_back(v);
		c[v].push_back(u);
	}
	dfs(1,0);
	
	tot=1;
	build(1,1,n);
	int z;
	for(int i=1;i<n;i++)
		if (deep[edges[i]]>deep[edgex[i]]){
			z=edges[i];
			edges[i]=edgex[i];
			edgex[i]=z;
		}
			
	for(int i=1;i<=q;i++){
		string s;
		int x,r;
			fin>>s;
            if(s[0]=='C'){
			   fin>>x>>r;
               change(1,x,r);
            }
            else if(s[0]=='Q'){
                    fin>>x;
                    fout<<query(x)<<endl;
                 }
    }
	return 0;
}