Jzoj5245 Competing Souls

温馨提示:本文附带bgm

某日，竞赛班的学生来到了一家糖果店。
店里卖着M袋糖果，第i袋糖果里装有i颗糖，价格为i。
有N个学生对这些糖果产生了兴趣，于是迅速站成一排，且将他们编号为1到N。其中第i个学生带着a[i]¥。每一轮，他们按顺序买糖果（每一轮每个人只会买一袋）。由于体内的竞争之魂与超乎常人的不服输精神，当前学生买的这袋糖果一定会比上一个人多（当然，第一次可以随便买）。如果第N个人买了糖果，那么就到第1个人开始下一轮。
然而，钱和糖果都有限，总是要停下来的。可以在任意时刻停止购买糖果，但是最后一次必须是第N个人购买。
现在他们想知道，最终所有人购买糖果数之和最大可以是多少。（当然可以一袋都不买~）
我们发现有一个限制，一定要最后是第n个人购买
所以我们可以考虑将其视为一个矩阵
首先枚举这个矩阵的行数i，让后依次填上1~i*n
让后开始考虑将几行的值改变
例如这样：n=3,m=10时
1 2 3
4 5 6
7 8 9
变成
1 2 3
4 5 7
8 9 10
那么我们来考虑如何变换
首先显然，将两行都+1和将一行+2是等价的，所以我们每次都将一整行直接加到最大（除非不够钱）
首先计算出，每一列减去已经填好得数后得到的最小的那个，记为dx
让后根据这个数来计算最多有多少行可以被整体加到最大
显然这个行数h=min(i,dx/(m-i*n))
剩下的，在第n-h行中，每一个数都尽可能加大，除非不够（参考上面那个例子的第二行）
这样整体复杂度为O(n×mn)=O(m)

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define LL long long
using namespace std;
int n,m;
LL s[500010],w[500010],A,dx,S,h;
int _18520(){
    scanf("%d%d",&n,&m); A=0;
    for(int i=1;i<=n;++i) scanf("%lld",s+i);
    for(int i=1,z=m/n;i<=z;++i){
        dx=1ll<<60; S=(n*i+1ll)*(LL)n*i>>1;
        for(int j=1;j<=n;++j){
            s[j]-=i*n-n+j;
            dx=min(dx,s[j]);
        }
        if(dx<0) break;
        h=min((LL)i,dx/(m-i*n));
        S+=n*h*(m-i*n); dx=h*(m-i*n);
        for(int j=1;j<=n;++j) w[j]=s[j]-dx;
        dx=m-i*n;
        for(int j=i-h;j>=i-h-1&&j;--j)
            for(int k=n;k;--k){
                dx=min(w[k],dx);
                S+=dx; w[k]-=dx;
                if(dx==0) break;
            }
        A=max(A,S);
    }
    printf("%lld
",A);
}
int main(){
    freopen("compete.in","r",stdin);
    freopen("compete.out","w",stdout); 
    int T; for(scanf("%d",&T);T--;_18520());
}