最小生成树

题目描述

结论

下面的m均指非树边有m条。
非树边的权值肯定赋值为路径上的最大值（根据环切原理这是下界）。
考虑按权值从小到大加入，每加入一条边就会连接两棵树。
那么一条非树边如果左端在其中一颗树中，右端在另一颗树中，则其权值和该树边一致。
由于是按照权值从小到大加入的，我们希望每次有尽量多的非树边权值确定。
那么容易发现，第i次一定会连接一个大小为i的树和一个单个的节点，也就是产生i-1条非树边。
定义S(i)=i*(i-1)/2。因此把问题改成：
$1<=a_i<=a_{i+1}$
$\sum_{j=1}^{n-1}aj=s$
$min{\sum_{j=1}^kaj*(j-1)+a_{k+1}*[m-S(k)]}$
当k+1没有达到n-1的时候，我们an-1怎么设都不影响答案，因此此时肯定是让a1~n-2全部为1最优秀，答案为m+s。
否则，我们设m-S(k)=t，然后讨论。
假设an-1=x，固定了x的话，前面怎么分配最优呢？
最大的坐标会有更大的系数，因此尽量使后面小，又不违反单调性质，很容易想到是平均分配，即能够找到一个分界点，前面都是y后面都是y+1。
我们可以找到最小的x，这个称为边界。
考虑一组 y y y……y y+1 y+1 …… y+1 x，其中y有z个。
假如它没有边界压力。此时x加1，会有一个y+1变成y，那么就贡献t-z。此时x减1，会有一个y变成y+1，那么就贡献z-t-1。
你容易发现它会往一边调整，因此前n-2个不是全部一样的话是不优的。
除了有边界压力的初始情况外，这个要判掉。
接下来一个考虑一组y y y ……y x，你可以给x加上n-2，让前面全部变成y-1，会带来一个常数影响。
那么也会有两种方向，最后变成1 1 1…… 1 s-(n-2)，或者变到有边界压力为止。
然后就解决了。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
ll n,m,s,x,y,z,w,t,ans;
int i,j,k,l,ca;
ll S(ll n){
    return n*(n-1)/2;
}
int main(){
    freopen("mst.in","r",stdin);freopen("mst.out","w",stdout);
    scanf("%d",&ca);
    while (ca--){
        scanf("%lld%lld%lld",&n,&m,&s);
        m-=(n-1);
        if (m<=S(n-2)) printf("%lld\n",m+s);
        else{
            t=m-S(n-2);
            x=s/(n-1);
            if (x*(n-1)<s) x++;
            y=(s-x)/(n-2);
            z=n-2-(s-x)%(n-2);
            ans=s+y*S(z)+(y+1)*(S(n-2)-S(z))+x*t;
            ans=min(ans,s+y*S(n-2)+(x+n-2-z)*t);
            ans=min(ans,s+S(n-2)+(s-(n-2))*t);
            printf("%lld\n",ans);
        }
    }
}