bzoj3714: [PA2014]Kuglarz

这题真是妙啊，只要知道所有所有[1,i]的和就能确定最后的值，当我们知道sum[i-1]，询问一次[i,j]，就知道了sum[j],这样我们可以用最小生成树来解决问题。对于[i,j],连一条从i-1到j的边。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
    char ch=getchar();int f=0,x=1;
    while(ch<'0'||ch>'9'){if(ch=='-') x=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){f=(f<<1)+(f<<3)+ch-'0';ch=getchar();}
    return f*x;
}
struct data
{
    int x;
    int y;
    int val;
}a[4000005];
inline bool cmp(const data &x,const data &y)
{
    return x.val<y.val;
}

int fa[4005],tot;long long ans;
int find(int x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int main()
{
    int n=read();
    for(int i=0;i<=n;i++) fa[i]=i;
    for(int i=1;i<=n;i++)
    {
        for(int j=i;j<=n;j++)
        {
            a[++tot].x=i-1;
            a[tot].y=j;
            a[tot].val=read();
        }
    }
    sort(a+1,a+tot+1,cmp);
    for(int i=1;i<=tot;i++)
    {
        int fx=find(a[i].x),fy=find(a[i].y);
        if(fx!=fy)
        {
            tot++;
            ans+=a[i].val;
            fa[fx]=fy;
            if(tot==n) break;
        }
    }
    cout<<ans;
}