【贪心+推式子计算贡献】ABC216 E

最新推荐文章于 2024-09-18 20:33:32 发布

lamentropetion

最新推荐文章于 2024-09-18 20:33:32 发布

阅读量90

点赞数

分类专栏：贪心文章标签：算法

本文链接：https://blog.csdn.net/weixin_62528401/article/details/131577318

版权

贪心专栏收录该内容

86 篇文章 0 订阅

订阅专栏

文章介绍了如何解决E-AmusementPark问题，首先强调了贡献与ai值的关系，建议对ai进行排序，然后从大到小遍历计算贡献。贡献可以通过几何模型或手动推导得出。接着提供了一个C++代码示例，展示了如何根据剩余贡献预算动态规划计算答案。最后，代码中包含了贡献计算的具体逻辑和边界处理。

摘要由CSDN通过智能技术生成

E - Amusement Park (atcoder.jp)

题意：

思路：

首先，优先访问ai大的贡献一定大

因此可以考虑把ai从小到大排个序，然后从尾部遍历

然后去计算贡献

贡献比较难推，但是手搓一下还是可以推出来的

先想出几何模型，再去手推

如果不会几何模型，直接去手推找规律也可

Code：

#include <bits/stdc++.h>
using namespace std;
#define int long long
int a[100005];
signed main()
{
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    sort(a+1,a+1+n);
    long long ans=0;
    for(int i=n;i>=1;i--)
    {
        int num = n-i+1;
        int cost = num*(a[i]-a[i-1]);
        if(cost<=k)
        {
            k-=cost;
            ans+=num * (a[i]+a[i-1]+1) * (a[i]-a[i-1]) / 2;
        }
        else
        {
            int t1 = k/num;
            int t2 = k%num;
            ans += num * (a[i] + a[i]-t1+1) * t1 / 2;
            ans += (a[i]-t1)*t2;
            break;
        }
    }
    printf("%lld\n",ans);
}