上海理工大学第二届“联想杯”全国程序设计邀请赛-I-贪心+优先队列

2023大厂真题提交网址(含题解):

www.CodeFun2000.com（http://101.43.147.120/）

最近我们一直在将收集到的机试真题制作数据并搬运到自己的OJ上，供大家免费练习,体会真题难度。现在OJ已录入50+道2023年最新大厂真题，同时在不断的更新。同时，可以关注"塔子哥学算法"公众号获得每道题的题解。

题目大意:

给你一个长度为$n$的二进制字符串,他的价值为每一个连续1的段的$C_n^2$,你每次可以将一个1变成0，问你最少的操作次数使得总价值$\le k$.

$n\le 1e5,k\le 1e10$.

题目思路：

对于连续的一段，一次操作，我们肯定是对半切最优。若干次操作，一定是均分最优。那么我们可以用一个优先队列{维护每一个段，他被操作过几次},大根堆.

这里我们需要计算,一个长度为$n$的段,操作$k$次之后的价值.

目标肯定是让每个子段越平均越好，则每一段长度为$\lfloor \frac{n}{k+1}\rfloor$.剩下$rest=n\%(k+1)$个1.分配到每个段上。

即，$n\%(k+1)$个长为$\lfloor \frac{n}{k+1}\rfloor +1$的段，$n-n\%(k+1)$个长为$\lfloor \frac{n}{k+1}\rfloor$的段

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vll vector<ll>
#define fi first
#define se second
const int maxn = 1e6 + 5;
const int mod = 1e9 + 7;
ll calc (ll x){
    return x * (x + 1) / 2;
}
ll a[maxn];
struct Node
{
    ll len , cnt , d;
    bool operator < (const Node & a) const{
        return d < a.d;
    }
};
ll cal (ll n , ll k)
{
    n -= k;
    ll len = n / (k + 1);
    ll rest = n - len * (k + 1);
    ll ans = rest * calc(len + 1) + (k + 1 - rest) * calc(len);
    return ans;
}
priority_queue<Node> q;
int main()
{
    ios::sync_with_stdio(false);
    ll n , k; cin >> n >> k;
    string a; cin >> a;
    a += '0';
    int cnt = 0;
    ll now = 0;
    for (auto g : a){
        if (g == '0'){
            if (cnt){
                q.push({cnt , 0 , cal(cnt , 0) - cal(cnt , 1)});
                now += cal(cnt , 0);
                cnt = 0;
            }
        }
        else {
            cnt++;
        }
    }
    int ans = 0;
    while (now > k){
        Node g = q.top();
        q.pop();
        now -= cal(g.len , g.cnt++);
        now += cal(g.len , g.cnt);
        ans++;
        if (g.len == g.cnt) continue;
        g.d = cal(g.len , g.cnt) - cal(g.len , g.cnt + 1);
        q.push(g);
    }
    cout << ans << endl;
    return 0;
}