Codeforces 1380 G. Circular Dungeon —— 贪心

This way

题意：

现在有一个长度为n的有向圈，i只能往i+1移动，n往1移动。
有n个箱子要放到圈上的每个位置，现在有两种箱子，
1.拿到价值c[i]并且继续走
2.游戏结束
这个人会随机出生在某个点，然后进行游戏。
你要确定k个二号箱子，n-k个1号箱子，并且将其放到圈上，使得，这个人拿到的钱的期望最小。
问你k=1~n时的答案

题解：

水题啊，有点配不上这个分数，我们肯定是贪心地将大的箱子变成2号，然后k个置为2号之后，将接下来k个数放到这些箱子的前面一个位置，一一对应，这样子能取得的钱最少，那么sort一下做一遍就行了。这是调和级数所以时间复杂度是nlog

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const int N=3e5+5;
const ll mod=998244353;
ll sum[N],a[N];
bool cmp(int x,int y){return x>y;}
ll qpow(ll a,ll b){ll ans=1;for(;b;b>>=1,a=a*a%mod)if(b&1)ans=ans*a%mod;return ans;}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++)
        sum[i]=(sum[i-1]+a[i])%mod;
    ll inv=qpow(n,mod-2);
    for(int k=1;k<=n;k++){
        ll ans=0,v=0;
        for(int i=1;i<=n;i+=k,v++)
            ans=(ans+(sum[min(n,i+k-1)]-sum[i-1]+mod)*v)%mod;
        printf("%lld%c",ans*inv%mod," \n"[k==n]);
    }
    return 0;
}