ACwing 3664. 数组补全 在线转离线 贪心

题目

链接

题解思路

一开始我的想法是将每一步的中位数求出，一步一步的加数，在线处理，中位数大于y就加1，小于y就加y，这是基本的贪心，但是这样做复杂度貌似挺高的。

闫总将数分为大于y和小于y的部分，大于y的部分一定大于等于n/2+1,小于y部分一定小于等于n/2，n为奇数。
当断出的数组大于数y的数没超过n/2+1，就让他只加到n/2+1，这样可以保证答案最小，

超过n/2+1时取大，这时在中位数的左边还有大于等于y的数，小于y的数就要变少了。当变少的数比 断出的数组 小于数y的数 少时 ， 说明数组无解，此时最小的情况都满足不了。

接着求出最优即最小的总和，特判，然后输出数即可。

AC代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
int main ()
{
    int n,k,p,x,y,sum = 0,le = 0 ,re = 0 ;
    cin>>n>>k>>p>>x>>y;
    for (int i = 0 ; i < k ; i++ )
    {
        int t;
        cin>>t;
        if (t < y )
            le++;
        else
            re++;
        sum += t;
    }
    int rr = max(n/2 + 1 , re) , l = n - rr;
    if ( le > l )
    {
        cout<<"-1\n";
        return 0;
    }
    sum += l - le;
    sum += (rr - re)*y;
    if (sum <= x )
    {
        for ( int i = 1 ; i <=  l - le ; i++ )
            cout<<"1 ";
        for ( int i = 1 ; i <= rr - re ; i++ )
            cout<<y<<" ";
        cout<<"\n";
    }else
        cout<<"-1\n";
    return 0;
}