i++ data 代碼 == 範圍 cst string input 樹狀

【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】極地旅行社 離線+樹鏈剖分+樹狀數組