COJ 0979 WZJ的数据结构(负二十一)

时间:2023-03-08 17:46:28
WZJ的数据结构(负二十一)
难度级别:C; 运行时间限制:5000ms; 运行空间限制:262144KB; 代码长度限制:2000000B
试题描述

请你实现一个数据结构,完成这样的功能:

给你一个N个点的图,初始状态无边。

每次加入一条双向边(u,v,w),若加入后没有构成一棵生成树,输出“Not Yet”,否则输出当前最小生成树的权值。

输入
第一行两个正整数N,M。表示有N个点M个操作。
接下来M行每行三个正整数u,v,w。
输出
每次加入一条双向边(u,v,w),若加入后没有构成一棵生成树,输出“Not Yet”,否则输出当前最小生成树的权值。
输入示例
4 6
1 2 10
2 3 10
3 4 10
2 2 1
1 3 2
2 4 3
输出示例
Not Yet
Not Yet
30
30
22
15
其他说明
1<=N<=100000
1<=M<=500000
1<=ui,vi<=N
1<=wi<=1000

题解:这才是真正的动态MST呀

 #include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#define PAU putchar(' ')
#define ENT putchar('\n')
#define CH for(int d=0;d<=1;d++) if(ch[d])
using namespace std;
const int maxn=+,maxm=+,inf=-1u>>;
struct node{
node*fa,*ch[],*mx;int x;bool rev;
node(){fa=ch[]=ch[]=NULL;mx=this;x=;rev=false;}
void revt(){swap(ch[],ch[]);rev^=;return;}
void update(){mx=this;CH{if(mx->x<ch[d]->mx->x)mx=ch[d]->mx;}return;}
void down(){if(rev){CH{ch[d]->revt();}rev=false;}return;}
}lct[maxn+maxm],*nodecnt;
int parent(node*x,node*&y){return (y=x->fa)?y->ch[]==x?:y->ch[]==x?:-:-;}
void rotate(node*x){
node*y,*z;int d1=parent(x,y),d2=parent(y,z);
if(y->ch[d1]=x->ch[d1^]) y->ch[d1]->fa=y;
y->fa=x;x->fa=z;x->ch[d1^]=y;
if(d2!=-) z->ch[d2]=x;
y->update();return;
}
void pushdown(node*x){
static node*s[maxn];int top=;
for(node*y;;x=y){
s[top++]=x;y=x->fa;
if(!y||(y->ch[]!=x&&y->ch[]!=x)) break;
} while(top--) s[top]->down();return;
}
node*splay(node*x){
pushdown(x);node*y,*z;int d1,d2;
while(true){
if((d1=parent(x,y))<) break;
if((d2=parent(y,z))<){rotate(x);break;}
if(d1==d2) rotate(y),rotate(x);
else rotate(x),rotate(x);
} x->update();return x;
}
node*access(node*x){
node*ret=NULL;
for(;x;x=x->fa) splay(x)->ch[]=ret,(ret=x)->update();
return ret;
}
void makeroot(int x){access(x+lct)->revt();return;}
void link(int x,int y){makeroot(x);splay(x+lct)->fa=lct+y;return;}
void cut(int x,int y){
makeroot(x);node*p=(access(y+lct),splay(y+lct));
p->ch[]=p->ch[]->fa=NULL;p->update();return;
}
node*findtop(int x){
node*t=(access(x+lct),splay(x+lct));while(t->ch[]) t->down(),t=t->ch[];return t;
}
node*query(int x,int y){
makeroot(x);return access(y+lct)->mx;
}
int n,m;
inline int read(){
int x=,sig=;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')sig=-;ch=getchar();}
while(isdigit(ch))x=*x+ch-'',ch=getchar();
return x*=sig;
}
inline void write(int x){
if(x==){putchar('');return;}if(x<)putchar('-'),x=-x;
int len=,buf[];while(x)buf[len++]=x%,x/=;
for(int i=len-;i>=;i--)putchar(buf[i]+'');return;
}
int s[maxm],t[maxm];
void init(){
n=read();m=read();int ans=,cnt=n;nodecnt=lct+n+;
for(int i=;i<=m;i++){
s[i]=read();t[i]=read();
node*q=nodecnt++;q->x=read();int k=q-lct;makeroot(k);
if(s[i]!=t[i]) {
if(findtop(s[i])!=findtop(t[i])){
link(s[i],k);link(t[i],k);ans+=q->x;cnt--;
}else{
node*p=query(s[i],t[i]);if(p->x>q->x) {
ans+=q->x-p->x;int id=p-lct-n;
cut(s[id],p-lct);cut(t[id],p-lct);
link(s[i],k);link(t[i],k);}
}
}
if(cnt==) write(ans),ENT;
else puts("Not Yet");
}
return;
}
void work(){
return;
}
void print(){
return;
}
int main(){init();work();print();return ;}