atcoder 2019.1.27晚 E.Weights on Vertices and Edges

https://atcoder.jp/contests/nikkei2019-qual/tasks/nikkei2019_qual_e

题目给出n个点m条边，问你至少删掉多少条边，能够使得最后的所有连通块中，一个连通块中的点权之和要大于这个连通块中每一条边的值。

考场上已经接近正解了，按边权排序，从小到大考虑是否加进答案里去，并查集维护一下点权和，求出最多能加进答案的边，输出m-ans，但是不知道这个怎么判断。

题解的思路对这个判断很巧妙，对于一个合法连通块中的边，因为从小到大枚举的边，并查集中维护一个当前连通块中还有多少边没有加进答案，如果此时点权和已经比当前边大了，那么说明该连通块剩下的边全部合法，于是把当前连通块中没加入答案的边全部加入答案。

在submissions中还发现另外一种写法，用非路径压缩的最短路写，每次把小的连通块的根连上大的连通块的根，这样分叉就很多，也许能保证找根的深度不超过logn？(没想的很清楚)

从头开始跑一遍最小生成树，记录并查集上连接的顺序情况，而且不清空之前的值，再从后往前删掉那些非法的边，按照记录还原点权和，连通块大小等信息，最后剩下的边就是最后连通性地状况，再扫一遍所有边，在不改变连通性地情况下，看哪些边在答案中。

前一种正解52ms，后一种56ms，非压缩路径的并查集也这么快的吗？

#include<bits/stdc++.h>
#define maxl 100010

using namespace std;

int n,m,ans;
int a[maxl];
struct ed
{
	int u,v,val;
}e[maxl];
int f[maxl],cnt[maxl];
long long sum[maxl];

inline bool cmp(const ed &x,const ed &y)
{
	return x.val<y.val;
}

inline void prework()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		f[i]=i;cnt[i]=0;sum[i]=a[i];
	}
	for(int i=1;i<=m;i++)
		scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].val);
	sort(e+1,e+1+m,cmp);
}

inline int find(int x)
{
	if(f[x]!=x)
	{
		f[x]=find(f[x]);
		cnt[f[x]]+=cnt[x];
		sum[f[x]]+=cnt[x];
		sum[x]=cnt[x]=0;
	}
	return f[x];
}

inline void mainwork()
{
	int x,y,xx,yy;
	for(int i=1;i<=m;i++)
	{
		x=e[i].u;y=e[i].v;
		xx=find(x);yy=find(y);
		if(xx!=yy)
		{
			cnt[xx]+=cnt[yy]+1;
			sum[xx]+=sum[yy];
			f[yy]=xx;
			sum[yy]=0;cnt[yy]=0;
			if(sum[xx]>=e[i].val)
			{
				ans+=cnt[xx];
				cnt[xx]=0;
			}
		}
		else
		{
			cnt[xx]++;
			if(sum[xx]>=e[i].val)
			{
				ans+=cnt[xx];
				cnt[xx]=0; 
			}
		}
	}
}

inline void print()
{
	printf("%d",m-ans);
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#define ll long long
using namespace std;
const int N=2e5;
struct edge{int fr,to,w;}a[N],sta[N];
int n,m,val[N],fa[N];
int siz[N],cnt,top,ans;
ll sum[N];
int cmp(const edge&A,const edge&B) {return A.w<B.w;}
int find(int x) {return fa[x]==x?x:find(fa[x]);}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&val[i]);
		fa[i]=i,siz[i]=1,sum[i]=val[i];
	}
	for(int i=1,x,y,w;i<=m;i++)
	{
		scanf("%d%d%d",&x,&y,&w);
		a[++cnt]=(edge){x,y,w};
	}
	sort(a+1,a+cnt+1,cmp);

	for(int i=1;i<=cnt;i++)
	{
		int x=a[i].fr,y=a[i].to;
		x=find(x);y=find(y);
		if(x==y) continue;
		if(siz[x]>siz[y]) swap(x,y);
		sta[++top]=(edge){x,y,a[i].w};
		fa[x]=y;siz[y]+=siz[x];sum[y]+=sum[x];
	}
	for(int i=top;i>=1;i--)
	{
		int x=sta[i].fr,y=sta[i].to;
		ll v=sum[find(y)],val=sta[i].w;
		if(v<val) fa[x]=x,siz[y]-=siz[x],sum[y]-=sum[x];
	}
	for(int i=1;i<=m;i++)
	{
		int x=a[i].fr,y=a[i].to;
		if(find(x)!=find(y)) ans++;
		else if(a[i].w>sum[find(x)]) ans++;
	}
	cout<<ans<<endl;
}