제출 #482483

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
482483		HolyCrusaD3R	Gift (IZhO18_nicegift)	C++17	30 / 100	1340 ms	524292 KiB

nicegift

#include<bits/stdc++.h>
#define int long long
#define f first
#define s second
#define pb push_back
#define p push
using namespace std;

int n,k,q,tmp,sum=0,m=INT_MIN;
vector<int> v,pv;
vector<vector<pair<int,int>>> matrix;
vector<vector<int>> ans;

signed main(){

    //freopen("input.in", "r", stdin);
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin>>n>>k;
    q=n;
    while(q--){
        cin>>tmp;
        sum+=tmp;
        m=max(m,tmp);
        v.pb(tmp);
    }
    if(float(sum/2)<float(m) || sum%k!=0){
        cout<<-1;
        return 0;
    }
    //pv = v;
    const int flag = sum/k;
    sum = 0;
    int i = 0, j = 0;
    stack<int> s;
    //cout<<"t"<<endl;
    while(i!=v.size()){
        //cout<<"t1"<<endl;
        matrix.pb({});
        if(!s.empty()){
            matrix[j].pb({s.top(),i-1});
            s.pop();
        }
        while(sum<flag && i!=v.size()){
            //cout<<"t2"<<endl;
            matrix[j].pb({v[i],i});
            //cout<<"t3"<<endl;
            sum+=v[i];
            i++;
        }
        if(sum-flag != 0){
            //cout<<"sum - "<<sum<<endl;
            //cout<<"matrix last before - "<<matrix[j][matrix[j].size()-1]<<endl;
            matrix[j][matrix[j].size()-1].f+=(flag - sum);
            s.p(sum-flag);
            //cout<<flag<<endl;
            //cout<<"matrix last - "<<matrix[j][matrix[j].size()-1]<<endl;
            //cout<<"s.top() - "<<s.top()<<endl;
        }
        j++;
        sum=sum-flag;
    }

    /*
    for(int i=0;i<matrix.size();i++){
        for(int j=0;j<matrix[i].size();j++){
            cout<<matrix[i][j].f<<" "<<matrix[i][j].s<<endl;
        }
        cout<<endl;
    }
    */
    int z=0;
    while(true){
        sum = 0;
        m = INT_MAX;
        for(int i=0;i<matrix.size();i++){
            //cout<<matrix[i][matrix[i].size()-1].f<<" ";
            if(matrix[i].size()>0){
                m=min(matrix[i][matrix[i].size()-1].f,m);
            }else{
                sum++;
            }
        }
        if(sum==k){
            break;
        }else{
            sum=0;
        }
        //cout<<endl<<m;
        while(!s.empty()){
            s.pop();
        }
        for(int i=0;i<matrix.size();i++){
            s.p(matrix[i][matrix[i].size()-1].s+1);
            matrix[i][matrix[i].size()-1].f-=m;
            if(matrix[i][matrix[i].size()-1].f == 0){
                matrix[i].pop_back();
            }
        }
        ans.pb({});
        ans[z].pb(m);
        while(!s.empty()){
            ans[z].pb(s.top());
            s.pop();
        }
        z++;
    }

    cout<<ans.size()<<endl;
    for(int i=0;i<ans.size();i++){
        for(int j=0;j<ans[i].size();j++){
            cout<<ans[i][j]<<" ";
        }
        cout<<endl;
    }


    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

nicegift.cpp: In function 'int main()':
nicegift.cpp:39:12: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |     while(i!=v.size()){
      |           ~^~~~~~~~~~
nicegift.cpp:46:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |         while(sum<flag && i!=v.size()){
      |                           ~^~~~~~~~~~
nicegift.cpp:78:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::vector<std::pair<long long int, long long int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   78 |         for(int i=0;i<matrix.size();i++){
      |                     ~^~~~~~~~~~~~~~
nicegift.cpp:95:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::vector<std::pair<long long int, long long int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   95 |         for(int i=0;i<matrix.size();i++){
      |                     ~^~~~~~~~~~~~~~
nicegift.cpp:112:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::vector<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  112 |     for(int i=0;i<ans.size();i++){
      |                 ~^~~~~~~~~~~
nicegift.cpp:113:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  113 |         for(int j=0;j<ans[i].size();j++){
      |                     ~^~~~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 11 / 11
• Subtask 3: 12 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 51