BZOJ-3531: [Sdoi2014]旅行

[文章目录]

Description

 S国有N个城市,编号从1到N。城市间用N-1条双向道路连接,满足
从一个城市出发可以到达其它所有城市。每个城市信仰不同的宗教,如飞天面条神教、隐形独角兽教、绝地教都是常见的信仰。为了方便,我们用不同的正整数代表各种宗教,  S国的居民常常旅行。旅行时他们总会走最短路,并且为了避免麻烦,只在信仰和他们相同的城市留宿。当然旅程的终点也是信仰与他相同的城市。S国政府为每个城市标定了不同的旅行评级,旅行者们常会记下途中(包括起点和终点)留宿过的城市的评级总和或最大值。
在S国的历史上常会发生以下几种事件:
”CC x c”:城市x的居民全体改信了c教;
”CW x w”:城市x的评级调整为w;
”QS x y”:一位旅行者从城市x出发,到城市y,并记下了途中留宿过的城市的评级总和;
”QM x y”:一位旅行者从城市x出发,到城市y,并记下了途中留宿过
的城市的评级最大值。
由于年代久远,旅行者记下的数字已经遗失了,但记录开始之前每座城市的信仰与评级,还有事件记录本身是完好的。请根据这些信息,还原旅行者记下的数字。    为了方便,我们认为事件之间的间隔足够长,以致在任意一次旅行中,所有城市的评级和信仰保持不变。


啊,1A,爽。
树剖+动态开点线段树。
发现需要开n个线段树,然后每个维护的节点数的总和是n,n^2会MLE,所以动态开点。类似链表,记录lson和rson的id(下标),然后找。区间长度的话照样是那么大,然后搜索的时候通过判断端点进入左右子树。所以说每个节点上面就有log(n)个节点为他服务,所以空间复杂度最大为nlog(n),加上开的根节点,就大概那么大吧。

在线段树上投影的tid[]在所有线段树上都是一样的。

插入点的时候发现不存在儿子就开一个。发现修改和插入可以是一个函数,爽(suang)。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define N 101000
#define MT 3501000
int w[N],c[N],n,m;
char st[10];
int head[N],nxt[N<<1],to[N<<1],cnt;
int root[N],tot;
int tid[N],tcnt,top[N],siz[N],son[N],fa[N],deep[N];

void add(int x,int y)
{
	to[++cnt]=y;
	nxt[cnt]=head[x];
	head[x]=cnt;
}

void dfs1(int x)
{
	int tmp=-1;siz[x]=1;
	for(int i=head[x];i;i=nxt[i])
		if(to[i]!=fa[x])
		{
			fa[to[i]]=x;deep[to[i]]=deep[x]+1;
			dfs1(to[i]);
			if(siz[to[i]]>tmp)
				tmp=siz[to[i]],son[x]=to[i];
			siz[x]+=siz[to[i]];
		}
}
struct seg
{
	int sum[MT],mx[MT],ls[MT],rs[MT],lch[MT],rch[MT];
	void pushup(int pos)
	{
		sum[pos]=sum[ls[pos]]+sum[rs[pos]];
		mx[pos]=max(mx[ls[pos]],mx[rs[pos]]);
	}
	void insert(int &pos,int l,int r,int id,int val)
	{
		if(!pos) pos=++tot;
		if(l==id&&r==id)
		{
			sum[pos]=mx[pos]=val;
			return ;
		}
		int mid=l+r>>1;
		if(id<=mid) insert(ls[pos],l,mid,id,val);
		else insert(rs[pos],mid+1,r,id,val);
		pushup(pos);
	}
	int query_sum(int pos,int l,int r,int x,int y)
	{
		if(!pos) return 0;
		if(x<=l&&y>=r) return sum[pos];
		int mid=l+r>>1;
		if(y<=mid) return query_sum(ls[pos],l,mid,x,y);
		else if(x>mid) return query_sum(rs[pos],mid+1,r,x,y);
		else return query_sum(ls[pos],l,mid,x,y)+query_sum(rs[pos],mid+1,r,x,y);
	}
	int query_mx(int pos,int l,int r,int x,int y)
	{
		if(!pos) return 0;
		if(x<=l&&y>=r) return mx[pos];
		int mid=l+r>>1;
		if(y<=mid) return query_mx(ls[pos],l,mid,x,y);
		else if(x>mid) return query_mx(rs[pos],mid+1,r,x,y);
		else return max(query_mx(ls[pos],l,mid,x,y),query_mx(rs[pos],mid+1,r,x,y));
	}
}T;
void dfs2(int x,int anc)
{
	top[x]=anc;tid[x]=++tcnt;
	T.insert(root[c[x]],1,n,tcnt,w[x]);
	if(son[x]) dfs2(son[x],anc);
	for(int i=head[x];i;i=nxt[i])
		if(to[i]!=fa[x]&&to[i]!=son[x])
			dfs2(to[i],to[i]);
}

void cc()
{
	int x,y; scanf("%d%d",&x,&y);
	T.insert(root[c[x]],1,n,tid[x],0);
	c[x]=y;
	T.insert(root[c[x]],1,n,tid[x],w[x]);
}

void cw()
{
	int x,y; scanf("%d%d",&x,&y);w[x]=y;
	T.insert(root[c[x]],1,n,tid[x],y);
}

void qs()
{
	int x,y;scanf("%d%d",&x,&y);int id=root[c[x]],ans=0;
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]]) swap(x,y);
		ans+=T.query_sum(id,1,n,tid[top[x]],tid[x]);
		x=fa[top[x]];
	}
	if(tid[x]>tid[y]) swap(x,y);
	ans+=T.query_sum(id,1,n,tid[x],tid[y]);
	printf("%d\n",ans);
}

void qm()
{
	int x,y;scanf("%d%d",&x,&y);int id=root[c[x]],ans=0;
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]]) swap(x,y);
		ans=max(ans,T.query_mx(id,1,n,tid[top[x]],tid[x]));
		x=fa[top[x]];
	}
	if(tid[x]>tid[y]) swap(x,y);
	ans=max(ans,T.query_mx(id,1,n,tid[x],tid[y]));
	printf("%d\n",ans);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d%d",&w[i],&c[i]);
	for(int i=1;i<=n;i++) root[i]=i,T.lch[i]=1,T.rch[i]=n; tot=n;
	for(int x,y,i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	}
	dfs1(1);dfs2(1,1);
	while(m--)
	{
		scanf("%s",st);
		switch(st[1])
		{
			case 'C': cc();break;
			case 'W': cw();break;
			case 'S': qs();break;
			case 'M': qm();break;
		}
	}
	return 0;
}

 

发表评论

邮箱地址不会被公开。 必填项已用*标注