「日常训练」School Marks（Codeforces Round 301 Div.2 B）

Posted 2021-01-09 samhx

题意与分析（CodeForces 540B）

代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#define MP make_pair
#define PB push_back
#define fi first
#define se second
#define ZERO(x) memset((x), 0, sizeof(x))
#define ALL(x) (x).begin(),(x).end()
#define rep(i, a, b) for (int i = (a); i <= (b); ++i)
#define per(i, a, b) for (int i = (a); i >= (b); --i)
#define QUICKIO                      ios::sync_with_stdio(false);     cin.tie(0);                      cout.tie(0);
using namespace std;

int main()
{
    int n,k,p,x,y; cin>>n>>k>>p>>x>>y;
    int nowtot=0,belowycnt=0;
    int arr[1005]; rep(i,1,k) { cin>>arr[i]; nowtot+=arr[i]; if(arr[i]<y) belowycnt++; }
    int restmark=x-nowtot;
    if(nowtot+n-k>x || belowycnt>n/2) cout<<-1<<endl;
    else
    {
        int geycnt=k-belowycnt;
        vector<int> vec;
        rep(i,k+1,n)
        {
            if(belowycnt<n/2)
            {
                restmark--;
                if(restmark>=0)
                    { vec.PB(1); belowycnt++; }
            }
            else
            {
                restmark-=y;
                if(restmark>=0)
                    vec.PB(y);
            }
            if(restmark<=0) break;
        }
        if(vec.size()==n-k)
        {
            for(auto it:vec) cout<<it<<" ";
            cout<<endl;
        }
        else cout<<-1<<endl;
    }

    
    return 0;
}