Description 捉迷藏
Jiajia和Wind是一对恩爱的夫妻,并且他们有很多孩子。某天,Jiajia、Wind和孩子们决定在家里玩
捉迷藏游戏。他们的家很大且构造很奇特,由N个屋子和N-1条双向走廊组成,这N-1条走廊的分布使得任意两个屋
子都互相可达。游戏是这样进行的,孩子们负责躲藏,Jiajia负责找,而Wind负责操纵这N个屋子的灯。在起初的
时候,所有的灯都没有被打开。每一次,孩子们只会躲藏在没有开灯的房间中,但是为了增加刺激性,孩子们会要
求打开某个房间的电灯或者关闭某个房间的电灯。为了评估某一次游戏的复杂性,Jiajia希望知道可能的最远的两
个孩子的距离(即最远的两个关灯房间的距离)。 我们将以如下形式定义每一种操作: C(hange) i 改变第i个房
间的照明状态,若原来打开,则关闭;若原来关闭,则打开。 G(ame) 开始一次游戏,查询最远的两个关灯房间的 距离。 Input
第一行包含一个整数N,表示房间的个数,房间将被编号为1,2,3…N的整数。接下来N-1行每行两个整数a, b,
表示房间a与房间b之间有一条走廊相连。接下来一行包含一个整数Q,表示操作次数。接着Q行,每行一个操作,如 上文所示。 Output
对于每一个操作Game,输出一个非负整数到hide.out,表示最远的两个关灯房间的距离。若只有一个房间是关
着灯的,输出0;若所有房间的灯都开着,输出-1。
边分治模板题,SPOJ上的卡不过去。
动态点分治做法见【这里】。
每次选择一条边,左右两边分别用一个堆来维护到这条边的最远距离。因为不想手写堆,所以自创了删点以后加到map里的方法。反正复杂度不会变坏。
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int oo=0x3f3f3f3f;
vector<int> t1[1000010];
int tem[1000010];
priority_queueint,int> > pq[1000010];
map<int,int> mp[1000010];
int fir[1000010],ne[1000010],to[1000010],w[1000010],val[1000010],du[1000010],
clr[1000010],vis[1000010],del[1000010],size[1000010],ls[1000010],rs[1000010],
pos[1000010],fa[1000010],
n,root,low,R,num,clo;
int rd()
{
int x=0;
char c=getchar();
while (c<'0'||c>'9') c=getchar();
while (c>='0'&&c<='9')
{
x=x*10+c-'0';
c=getchar();
}
return x;
}
char rdc()
{
char c=getchar();
while (c<'A'||c>'Z') c=getchar();
return c;
}
void add(int u,int v,int x)
{
num++;
ne[num*2]=fir[u];
fir[u]=num*2;
to[num*2]=v;
w[num*2]=x;
ne[num*2+1]=fir[v];
fir[v]=num*2+1;
to[num*2+1]=u;
w[num*2+1]=x;
}
void dfs2(int u,int fa,int siz)
{
int v;
size[u]=1;
for (int i=fir[u];i;i=ne[i])
if (!del[i>>1]&&!vis[i>>1]&&(v=to[i])!=fa)
{
dfs2(v,u,siz);
size[u]+=size[v];
if (root==-1||max(size[v],siz-size[v])>1,low=v;
}
}
void dfs3(int u,int fa,int dis,int with)
{
int v;
pq[with].push(make_pair(dis,u));
for (int i=fir[u];i;i=ne[i])
if (!del[i>>1]&&!vis[i>>1]&&(v=to[i])!=fa)
dfs3(v,u,dis+w[i],with);
}
void up(int u)
{
pair<int,int> p1,p2;
p1.second=p2.second=-1;
while (!pq[ls[u]].empty())
{
p1=pq[ls[u]].top();
if (clr[p1.second])
{
pq[ls[u]].pop();
mp[ls[u]][p1.second]=p1.first;
p1.second=-1;
}
else break;
}
while (!pq[rs[u]].empty())
{
p2=pq[rs[u]].top();
if (clr[p2.second])
{
pq[rs[u]].pop();
mp[rs[u]][p2.second]=p2.first;
p2.second=-1;
}
else break;
}
val[u]=max(val[ls[u]],val[rs[u]]);
if (p1.second!=-1&&p2.second!=-1) val[u]=max(val[u],p1.first+p2.first+w[u<<1]);
}
int dfs1(int u,int siz)
{
if (siz==1)
{
pos[u]=++clo;
pq[clo].push(make_pair(0,u));
if (clr[u]) val[clo]=-oo;
else val[clo]=0;
return clo;
}
int r1,x,y,sx,sy;
root=-1;
dfs2(u,-1,siz);
vis[r1=root]=1;
x=to[r1<<1];
y=to[r1<<1|1];
if (x==low) sx=size[x],sy=siz-size[x];
else sx=siz-size[y],sy=size[y];
ls[r1]=dfs1(x,sx);
fa[ls[r1]]=r1;
rs[r1]=dfs1(y,sy);
fa[rs[r1]]=r1;
dfs3(x,-1,0,ls[r1]);
dfs3(y,-1,0,rs[r1]);
up(r1);
vis[r1]=0;
return r1;
}
void modi(int u)
{
int p=pos[u];
clr[u]^=1;
if (!clr[u])
{
val[p]=0;
while (p!=R)
{
if (mp[p].count(u))
{
pq[p].push(make_pair(mp[p][u],u));
mp[p].erase(u);
}
up(fa[p]);
p=fa[p];
}
}
else
{
val[p]=-oo;
while (p!=R)
{
up(fa[p]);
p=fa[p];
}
}
}
int build(int l,int r)
{
if (l==r) return tem[l];
int mid=(l+r)/2,id=++n,x=build(l,mid),y=build(mid+1,r);
clr[id]=1;
if (l==mid) add(id,x,1);
else add(id,x,0);
if (mid+1==r) add(id,y,1);
else add(id,y,0);
return id;
}
void dfs4(int u,int fa)
{
int tot=0,x,y,v;
for (int i=0;iif (t1[u][i]!=fa)
tem[++tot]=t1[u][i];
if (tot>2)
{
int mid=(1+tot)/2;
x=build(1,mid);
y=build(mid+1,tot);
add(u,x,0);
if (tot==3) add(u,y,1);
else add(u,y,0);
}
else for (int i=1;i<=tot;i++) add(u,tem[i],1);
for (int i=0;iif ((v=t1[u][i])!=fa)
dfs4(v,u);
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
//printf("Elapsed time:%f secs.\n",(double)clock()/CLOCKS_PER_SEC);
int u,v,x,q;
char c;
n=rd();
for (int i=1;iprintf("Elapsed time:%f secs.\n",(double)clock()/CLOCKS_PER_SEC);
dfs4(1,-1);
//printf("Elapsed time:%f secs.\n",(double)clock()/CLOCKS_PER_SEC);
//for (int i=1;i<=n;i++) for (int j=fir[i];j;j=ne[j]) printf("(%d)%d->%d:%d\n",j/2,i,to[j],w[j]);
clo=num;
R=dfs1(1,n);
//printf("Elapsed time:%f secs.\n",(double)clock()/CLOCKS_PER_SEC);
q=rd();
while (q--)
{
c=rdc();
if (c=='G') printf("%d\n",val[R]);
else
{
u=rd();
modi(u);
}
}
//printf("Elapsed time:%f secs.\n",(double)clock()/CLOCKS_PER_SEC);
}