bzoj3991 [SDOI2015]寻宝游戏
原题地址:http://www.lydsy.com/JudgeOnline/problem.php?id=3991
题意:
一棵n个节点的树,m个操作,每次取消或添加一个关键点,每次操作输出求包含当前这些关键点虚树的路径和的两倍。
数据范围
1<=N<=100000
1<=M<=100000
对于全部的数据,1<=z<=10^9
题解:
(今天的t3)
答案为按DFS序排序后相邻两个关键点的间距离和加上首尾两个点的距离。
于是乎把所有的关键点塞进set里,维护一个sum。
set中在其前一个的为pre,在其后一个的为nxt
删除一个点就加上pre与nxt之间的距离,减去其与pre,其与nxt的距离。
加入一个点就减去pre与nxt之间的距离,加上其与pre,其与nxt的距离。
求距离就简单地求下lca。
注意答案还要加上首尾两个点的距离,另外为了方便处理没有pre或没有nxt的情况,可以先往set中加入0和n+1便于特判。
(get了访问set中前/后一个元素的方法:int pre=* –S.find(dfn[x]); int nxt=* ++S.find(dfn[x]);)
(现在知道虚树为何,膜你抄全歌词达成☆)
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<set>
#define LL long long
using namespace std;
const int N=200005;
const int P=18;
set<int> S;
int n,m,head[N],to[2*N],nxt[2*N],num=0;
int dep[N],dfn[N],seq[N],inc=0,anc[N][P+5];
LL w[2*N],dis[N];
bool vis[N];
LL sum=0;
void build(int u,int v,LL ww)
{
num++;
to[num]=v;
nxt[num]=head[u];
w[num]=ww;
head[u]=num;
}
void dfs(int u,int f)
{
inc++;
anc[u][0]=f;
for(int i=1;i<P;i++)
anc[u][i]=anc[anc[u][i-1]][i-1];
seq[inc]=u; dfn[u]=inc;
dep[u]=dep[f]+1;
for(int i=head[u];i;i=nxt[i])
{
int v=to[i];
if(v==f) continue;
dis[v]=dis[u]+w[i];
dfs(v,u);
}
}
int getlca(int u,int v)
{
if(dep[u]<dep[v]) swap(u,v);
int d=dep[u]-dep[v];
for(int i=0;d;d>>=1,i++)
if(d&1) u=anc[u][i];
if(u==v) return u;
for(int i=P-1;i>=0;i--)
if(anc[u][i]!=anc[v][i])
{u=anc[u][i]; v=anc[v][i];}
return anc[u][0];
}
LL cul(int u,int v)
{
int lca=getlca(u,v);
return dis[u]+dis[v]-2*dis[lca];
}
int main()
{
memset(head,0,sizeof(head));
memset(vis,0,sizeof(vis));
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++)
{
int u,v; LL ww;
scanf("%d%d%lld",&u,&v,&ww);
build(u,v,ww); build(v,u,ww);
}
dfs(1,1);
S.insert(0); S.insert(n+1);
while(m--)
{
int x;
scanf("%d",&x);
if(!vis[x])
{
S.insert(dfn[x]);
int pre=*--S.find(dfn[x]);
int nxt=*++S.find(dfn[x]);
if(pre>=1&&nxt<=n)
sum-=cul(seq[pre],seq[nxt]);
if(pre>=1) sum+=cul(seq[pre],x);
if(nxt<=n) sum+=cul(x,seq[nxt]);
vis[x]=1;
}
else
{
int pre=*--S.find(dfn[x]);
int nxt=*++S.find(dfn[x]);
if(pre>=1&&nxt<=n)
sum+=cul(seq[pre],seq[nxt]);
if(pre>=1) sum-=cul(seq[pre],x);
if(nxt<=n) sum-=cul(x,seq[nxt]);
vis[x]=0;
S.erase(dfn[x]);
}
LL add;
int first=*++S.find(0); int last=*--S.find(n+1);
if(first>n||last<1||first==last) add=0;
else add=cul(seq[first],seq[last]);
printf("%lld\n",sum+add);
}
return 0;
}