LeetCode376周赛T4 100123. 执行操作使频率分数最大（双指针或者二分，中位数贪心）

最新推荐文章于 2024-07-12 17:06:20 发布

live4m

最新推荐文章于 2024-07-12 17:06:20 发布

阅读量27

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/weixin_44178736/article/details/135045473

版权

题意：

在这里插入图片描述

解法：

显然要先对数组排序.

要选的数一定是某一段连续区间，
对于选中的区间[l,r]，一定是将所有数变为[l,r]的中位数最优.
枚举r，指针维护代价<=k的最小左端点即可.

另一种解法是二分区间长度mid,
二分的check就是判断是否存在代价<=k,长度为mid的区间.

Code：

#define ll long long
class Solution {
public:
    int maxFrequencyScore(vector<int>& a, long long k) {
        int n=a.size();
        sort(a.begin(),a.end());
        a.insert(a.begin(),0);

        vector<ll>sum(n+1,0);
        for(int i=1;i<=n;i++)sum[i]=sum[i-1]+a[i];

        // a[l,r]变成a[(l+r)/2]的代价
        auto gets=[&](int l,int r){
            int i=(l+r)/2;
            int lc=i-l+1;
            int rc=r-i+1;
            ll ans=0;
            ans+=1ll*lc*a[i]-(sum[i]-sum[l-1]);
            ans+=(sum[r]-sum[i-1])-1ll*rc*a[i];
            return ans;
        };
        int ans=0;
        int l=1;
        for(int r=1;r<=n;r++){
            while(gets(l,r)>k)l++;
            ans=max(ans,r-l+1);
        }
        return ans;
    }
}