http://www.elijahqi.win/archives/3341
Description
小B最近正在玩一個尋寶游戲，這個游戲的地圖中有N個村莊和N-1條道路，并且任何兩個村莊之間有且僅有一條路徑可達。游戲開始時，玩家可以任意選擇一個村莊，瞬間轉移到這個村莊，然后可以任意在地圖的道路上行走，若走到某個村莊中有寶物，則視為找到該村莊內的寶物，直到找到所有寶物并返回到最初轉移到的村莊為止。小B希望評測一下這個游戲的難度，因此他需要知道玩家找到所有寶物需要行走的最短路程。但是這個游戲中寶物經常變化，有時某個村莊中會突然出現寶物，有時某個村莊內的寶物會突然消失，因此小B需要不斷地更新數據，但是小B太懶了，不愿意自己計算，因此他向你求助。為了簡化問題，我們認為最開始時所有村莊內均沒有寶物

Input
第一行，兩個整數N、M，其中M為寶物的變動次數。

接下來的N-1行，每行三個整數x、y、z，表示村莊x、y之間有一條長度為z的道路。
接下來的M行，每行一個整數t，表示一個寶物變動的操作。若該操作前村莊t內沒有寶物，則操作后村莊內有寶物；若該操作前村莊t內有寶物，則操作后村莊內沒有寶物。

Output
M行，每行一個整數，其中第i行的整數表示第i次操作之后玩家找到所有寶物需要行走的最短路程。若只有一個村莊內有寶物，或者所有村莊內都沒有寶物，則輸出0。

Sample Input
4 5
1 2 30
2 3 50
2 4 60
2
3
4
2
1
Sample Output
0
100
220
220
280
HINT
1<=N<=100000

1<=M<=100000
對于全部的數據，1<=z<=10^9
Source
Round 1 感謝yts1999上傳

結論題 考慮最短距離應該就是兩兩之間距離*2即可

那么用set模擬虛樹建樹的過程 每次在set里面動態維護dfs序列 的大小關系那么新插入就相當于找一個dfs序列上與我相鄰的兩個點 即前驅后繼 那么計算答案的時候只需要 將 當前點到前驅后繼的答案加上減去前去到后繼的距離即可 刪除同理 為逆操作

#include<set> #include<cstdio> #include<cctype> #include<algorithm> #define ll long long using namespace std; inline char gc(){static char now[1<<16],*S,*T;if(T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}return *S++; } inline int read(){int x=0,f=1;char ch=gc();while(!isdigit(ch)) {if (ch=='-') f=-1;ch=gc();}while(isdigit(ch)) x=x*10+ch-'0',ch=gc();return x*f; } const int N=100100; const int inf=0x3f3f3f3f; struct node{int y,next,z; }data[N<<1]; int h[N],dep[N],dfn[N],n,m,Log[N],fa[N][20],cnt,num;ll dis[N],ans;bool mark[N]; struct cmp{inline bool operator()(const int &a,const int &b){return dfn[a]<dfn[b];} }; set<int,cmp> s; set<int,cmp>::iterator it1,it2; inline void dfs(int x){dfn[x]=++cnt;for (int i=h[x];i;i=data[i].next){int y=data[i].y,z=data[i].z;if(y==fa[x][0]) continue;dep[y]=dep[x]+1;dis[y]=dis[x]+data[i].z;fa[y][0]=x;for (int j=1;j<=Log[dep[y]];++j) fa[y][j]=fa[fa[y][j-1]][j-1];dfs(y);} } inline int lca(int x,int y){if(dep[x]<dep[y]) swap(x,y);int dis=dep[x]-dep[y];for (int i=0;i<=Log[dis];++i) if ((1<<i)&dis) x=fa[x][i];if (x==y) return x;for (int i=Log[dep[y]];~i;--i) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];return fa[x][0]; } inline ll calc(int x,int y){return dis[x]+dis[y]-(dis[lca(x,y)]<<1); } int main(){freopen("bzoj3991.in","r",stdin);n=read();m=read();Log[0]=-1;for (int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;for (int i=1;i<n;++i){int x=read(),y=read(),z=read();data[++num].y=y;data[num].next=h[x];h[x]=num;data[num].z=z;data[++num].y=x;data[num].next=h[y];h[y]=num;data[num].z=z;}dfs(1);cnt=0;for (int i=1;i<=m;++i){int t=read();if (mark[t]){--cnt;mark[t]=0;if (cnt<=1) {s.erase(t);puts("0");continue;}it1=s.lower_bound(t);it2=s.upper_bound(t);if (it1==s.begin()){it1=s.end();--it1;ans-=calc(t,*it1);ans-=calc(t,*it2);ans+=calc(*it1,*it2);s.erase(t);printf("%lld\n",ans);continue;}if (it2==s.end()){--it1;it2=s.begin();ans-=calc(t,*it1);ans-=calc(t,*it2);ans+=calc(*it1,*it2);s.erase(t);printf("%lld\n",ans);continue;}--it1;s.erase(t);ans-=calc(t,*it1);ans-=calc(t,*it2);ans+=calc(*it1,*it2);printf("%lld\n",ans);}else{++cnt;mark[t]=1;if (cnt<=1) {s.insert(t);puts("0");continue;}it1=s.lower_bound(t);it2=s.upper_bound(t);if (it1==s.begin()){it1=s.end();--it1;ans+=calc(t,*it1);ans+=calc(t,*it2);ans-=calc(*it1,*it2);s.insert(t);printf("%lld\n",ans);continue;}if (it2==s.end()){--it1;it2=s.begin();ans+=calc(t,*it1);ans+=calc(t,*it2);ans-=calc(*it1,*it2);s.insert(t);printf("%lld\n",ans);continue;}--it1;s.insert(t);ans+=calc(t,*it1);ans+=calc(t,*it2);ans-=calc(*it1,*it2);printf("%lld\n",ans);}}return 0; }

總結

以上是生活随笔為你收集整理的bzoj 3991 [SDOI2015]寻宝游戏的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。