【BZOJ 3714】[PA2014]Kuglarz 最小生成树

解法有点。。。迷

要知道每一个杯子的情况，有球为1，没有球为0,sum为前缀和，即了解每一个sum[i]-sum[i-1]，显然如果我们知道了sum[k]-sum[i]和sum[j]-sum[k]那么sum[j]-sum[i]也就是已知的了，而且其中任选两个都能推出第三个，也就是说每一询问i,j我们就将i,j-1加入一个集合中，那么集合中的任意两个点都是可以通过其他点推出来的，又要求最后权值最小，就是最小生成树了。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define maxn 2021
#define LL long long
using namespace std;
struct edge{
	int a,b;LL c;
	bool operator<(const edge& y)const{return c<y.c;}
}e[maxn*maxn];
int f[maxn],n,cnt;LL ans;
int find(int x){return f[x]==x ? x : f[x]=find(f[x]);}

int main(){
	scanf("%d",&n);LL x;
	for(int i=1;i<=n;i++){
		f[i]=i;
		for(int j=i;j<=n;j++){
			scanf("%lld",&x);e[++cnt]=(edge){i-1,j,x};
		}
	}sort(e+1,e+1+cnt);
	for(int dx,dy,i=1;i<=cnt;i++){
		dx=find(e[i].a),dy=find(e[i].b);
		if(dx!=dy){
			f[dx]=dy;ans+=e[i].c;
		}
	}printf("%lld",ans);
	return 0;
}