您的位置:首页 > 其它

test 10 problem C:最近公共祖先(线段树+乱搞)

2016-11-16 16:55 375 查看




题解:线段树+乱搞。

我们考虑将点x改为黑点的贡献,对于x子树中的点他们都可以取到x的值,利用dfs序,用线段树更新区间答案即可。对于x到根路径上的点,他们多了一个儿子有黑点,那除去这个儿子中点,其他的点都可以取到他的权值,所以也可以实现区间更新。当一个点有不少于2儿子同时拥有黑点的时候,他的整棵子树都可以拥有它的权值,直接整个子树更新即可,因这个点上面的点已经记录过当前这个儿子的贡献了,也就是对于他的父亲目前拥有的黑点属于同一个儿子,没有再更新的必要了,所以直接退出即可,否则向上跳,直到根为止。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 200003
using namespace std;
int n,m,tot,sz,vis
;
int tr[N*4],val
,dfsn
,pos
,belong
,l
,r
,delta[N*4],cover[N*4],delta1[N*4];
int point
,v
,next
,fa
,deep
,size
,son
,base
,sum
;
void add(int x,int y)
{
tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y;
tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x;
//cout<<x<<" "<<y<<endl;
}
void dfs(int x,int f)
{
deep[x]=deep[f]+1;
size[x]=1;
for (int i=point[x];i;i=next[i])
if (v[i]!=f) {
sum[x]++;
fa[v[i]]=x;
dfs(v[i],x);
size[x]+=size[v[i]];
if (size[son[x]]<size[v[i]]) son[x]=v[i];
}
}
void dfs1(int x,int chain)
{
belong[x]=chain; dfsn[++sz]=x; pos[x]=sz; l[x]=r[x]=sz;
if (!son[x]) return ;
dfs1(son[x],chain);
for (int i=point[x];i;i=next[i])
if (v[i]!=son[x]&&v[i]!=fa[x]) dfs1(v[i],v[i]);
r[x]=sz;
}
void pushdown(int now)
{
if (delta[now]!=-1) {
tr[now<<1]=max(tr[now<<1],delta[now]); tr[now<<1|1]=max(tr[now<<1|1],delta[now]);
delta[now<<1]=max(delta[now<<1],delta[now]); delta[now<<1|1]=max(delta[now<<1|1],delta[now]);
delta[now]=-1;
}
}
void qjchange(int now,int l,int r,int ll,int rr,int v)
{
if (ll>rr||!ll) return;
if(ll<=l&&r<=rr) {
tr[now]=max(tr[now],v);
delta[now]=max(delta[now],v);
return;
}
pushdown(now);
int mid=(l+r)/2;
if (ll<=mid) qjchange(now<<1,l,mid,ll,rr,v);
if (rr>mid) qjchange(now<<1|1,mid+1,r,ll,rr,v);
}
int segpoint(int now,int l,int r,int x)
{
if (l==r)  return tr[now];
pushdown(now);
int mid=(l+r)/2;
if (x<=mid) return segpoint(now<<1,l,mid,x);
else return segpoint(now<<1|1,mid+1,r,x);
}
int main()
{
freopen("lca3.in","r",stdin);
freopen("my.out","w",stdout);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&val[i]);
for (int i=1;i<n;i++) {
int x,y;  scanf("%d%d",&x,&y);
add(x,y);
}
dfs(1,0); dfs1(1,1);
//for (int i=1;i<=n;i++)
//cout<<fa[i]<<" ";
//cout<<endl;
memset(tr,-1,sizeof(tr));
memset(delta,-1,sizeof(delta));
for (int i=1;i<=m;i++) {
char s[10]; int x;
scanf("%s%d",s,&x);
if (s[0]=='Q')   printf("%d\n",segpoint(1,1,n,pos[x]));
else {
if (base[x]==1) {
qjchange(1,1,n,l[x],r[x],val[x]);
continue;
}
qjchange(1,1,n,l[x],r[x],val[x]); base[x]++;
while (fa[x]){
int f=fa[x];
if (base[f]==1) {
qjchange(1,1,n,l[f],r[f],val[f]);
break;
}
qjchange(1,1,n,l[f],l[x]-1,val[f]);
qjchange(1,1,n,r[x]+1,r[f],val[f]);
base[f]++;
x=fa[x];
}
}
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: