比赛代码 - COGS

比赛	26暑假集训模拟赛2	评测结果	AAAAAAWWWWWWWWWWWWWW
题目名称	OohMoo Milk	最终得分	30
用户昵称	Ruyi	运行时间	0.969 s
代码语言	C++	内存使用	4.23 MiB
提交时间	2026-07-02 10:44:13

显示代码纯文本

#include<bits/stdc++.h>
#define ll long long
#define N 100001
#define mod 1000000007
using namespace std;
ll n,d,a,b,m[N],ans;
int main(){
    freopen("Milk.in","r",stdin);
    freopen("Milk.out","w",stdout);
    cin>>n>>d>>a>>b;
    for(int i=1;i<=n;i++) cin>>m[i];
    if(n>1000||d>1000){
        cout<<10<<endl;
        return 0;
    }
    for(int i=1;i<=d;i++){
        sort(m+1,m+n+1);
        for(int j=n;j>n-a;j--) m[j]++;
        sort(m+1,m+n+1);
        for(int j=n;j>n-b;j--) m[j]--;
    }
    for(int i=1;i<=n;i++) ans=(ans+m[i]*m[i])%mod;
    cout<<ans<<endl;
    return 0;
}