【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 longusing 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;}