首页 > 代码库 > 【BZOJ2843】极地旅行社 离线+树链剖分+树状数组

【BZOJ2843】极地旅行社 离线+树链剖分+树状数组

2024-10-27 09:10:02   206人阅读

【BZOJ2843】极地旅行社

Description

不久之前，Mirko建立了一个旅行社，名叫“极地之梦”。这家旅行社在北极附近购买了N座冰岛，并且提供观光服务。当地最受欢迎的当然是帝企鹅了，这些小家伙经常成群结队的游走在各个冰岛之间。Mirko的旅行社遭受一次重大打击，以至于观光游轮已经不划算了。旅行社将在冰岛之间建造大桥，并用观光巴士来运载游客。Mirko希望开发一个电脑程序来管理这些大桥的建造过程，以免有不可预料的错误发生。这些冰岛从1到N标号。一开始时这些
岛屿没有大桥连接，并且所有岛上的帝企鹅数量都是知道的。每座岛上的企鹅数量虽然会有所改变，但是始终在[0, 1000]之间。你的程序需要处理以下三种命令：
1."bridge A B"——在A与B之间建立一座大桥（A与B是不同的岛屿）。由于经费限制，这项命令被接受，当且仅当A与B不联通。若这项命令被接受，你的程序需要输出"yes"，之后会建造这座大桥。否则，你的程序需要输出"no"。
2."penguins A X"——根据可靠消息，岛屿A此时的帝企鹅数量变为X。这项命令只是用来提供信息的，你的程序不需要回应。
3."excursion A B"——一个旅行团希望从A出发到B。若A与B连通，你的程序需要输出这个旅行团一路上所能看到的帝企鹅数量（包括起点A与终点B），若不联通，你的程序需要输出"impossible"。

Input

第一行一个正整数N，表示冰岛的数量。
第二行N个范围[0, 1000]的整数，为每座岛屿初始的帝企鹅数量。
第三行一个正整数M，表示命令的数量。接下来M行即命令，为题目描述所示。
1<=N<=30000,1<=M<=100000

Output

对于每个bridge命令与excursion命令，输出一行，为题目描述所示。

Sample Input

5
4 2 4 5 6
10
excursion 1 1
excursion 1 2
bridge 1 2
excursion 1 2
bridge 3 4
bridge 3 5
excursion 4 5
bridge 1 3
excursion 2 4
excursion 2 5

Sample Output

4
impossible
yes
6
yes
yes
15
yes
15
16

题解：网上大部分人都是用LCT写的，感觉用LCT代码应该会比树剖短不少~但是树剖常数还是小啊~

先离线读入整个图，用并查集判一下连那些边，然后无脑树剖。

#include <cstdio>#include <iostream>#include <cstring>using namespace std;const int maxn=30010;const int maxm=100010;int n,m,cnt;int to[maxn<<1],next[maxn<<1],v[maxn],f[maxn],head[maxn],s[maxn],dep[maxn],fa[maxn],top[maxn];int op[maxm],qa[maxm],qb[maxm],siz[maxn],son[maxn],p[maxn];int rd(){	int ret=0,f=1;	char gc=getchar();	while(gc<‘0‘||gc>‘9‘)	{if(gc==‘-‘)f=-f;	gc=getchar();}	while(gc>=‘0‘&&gc<=‘9‘)	ret=ret*10+gc-‘0‘,gc=getchar();	return ret*f;}char str[10];int find(int x){	return (f[x]==x)?x:(f[x]=find(f[x]));}void add(int a,int b){	to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;}void dfs1(int x){	siz[x]=1;	for(int i=head[x];i!=-1;i=next[i])	if(!dep[to[i]])	{		dep[to[i]]=dep[x]+1,fa[to[i]]=x,dfs1(to[i]),siz[x]+=siz[to[i]];		if(siz[to[i]]>siz[son[x]])	son[x]=to[i];	}}void dfs2(int x,int tp){	top[x]=tp,p[x]=++p[0];	if(son[x])	dfs2(son[x],tp);	for(int i=head[x];i!=-1;i=next[i])	if(to[i]!=fa[x]&&to[i]!=son[x])	dfs2(to[i],to[i]);}void updata(int x,int val){	for(int i=x;i<=n;i+=i&-i)	s[i]+=val;}int query(int x){	int i,ret=0;	for(i=x;i;i-=i&-i)	ret+=s[i];	return ret;}int ask(int x,int y){	int ret=0;	while(top[x]!=top[y])	{		if(dep[top[x]]<dep[top[y]])	swap(x,y);		ret+=query(p[x])-query(p[top[x]]-1),x=fa[top[x]];	}	if(dep[x]>dep[y])	swap(x,y);	ret+=query(p[y])-query(p[x]-1);	return ret;}int main(){	n=rd();	int i;	for(i=1;i<=n;i++)	v[i]=rd(),f[i]=i;	m=rd();	memset(head,-1,sizeof(head));	for(i=1;i<=m;i++)	{		scanf("%s",str),qa[i]=rd(),qb[i]=rd();		if(str[0]==‘b‘)		{			op[i]=1;			if(find(qa[i])==find(qb[i]))	op[i]=-1;			else	f[f[qa[i]]]=f[qb[i]],add(qa[i],qb[i]),add(qb[i],qa[i]);		}		if(str[0]==‘p‘)	op[i]=2;		if(str[0]==‘e‘)		{			op[i]=3;			if(find(qa[i])!=find(qb[i]))	op[i]=-3;		}	}	for(i=1;i<=n;i++)	if(!dep[i])	dep[i]=1,dfs1(i),dfs2(i,i);	for(i=1;i<=n;i++)	updata(p[i],v[i]);	for(i=1;i<=m;i++)	{		if(op[i]==-1)	printf("no\n");		if(op[i]==-3)	printf("impossible\n");		if(op[i]==1)	printf("yes\n");		if(op[i]==2)	updata(p[qa[i]],qb[i]-v[qa[i]]),v[qa[i]]=qb[i];		if(op[i]==3)	printf("%d\n",ask(qa[i],qb[i]));	}	return 0;}

【BZOJ2843】极地旅行社 离线+树链剖分+树状数组


联系
我们