Submission #17424101

---

Source Code Expand

#include<bits/stdc++.h>
#define pb push_back
using namespace std;
typedef vector<int> vec;
int now,n,k,a[105],b[105],c[1005],pos[105];
bool check()
{
    int mx=a[1],mn=a[1];
    for(int i=2;i<=k;i++)mx=max(mx,a[i]),mn=min(mn,a[i]);
    if(mx>=2*mn+1)return 0;
    return 1;
}
vec get(int len)
{
    for(int i=1;i<=k;i++)b[i]=a[i],pos[i]=0;
    for(int i=now-k+len+1;i<=now;i++)pos[c[i]]=i;
    for(int i=1;i<=k;i++)
        if(!pos[i])
        {
            b[i]--;
            if(b[i]<0)return vec(1,k+1);
        }
    int mn=b[1],mx=b[1];
    for(int i=2;i<=k;i++)mn=min(b[i],mn),mx=max(b[i],mx);
    if(mn*2+1<mx)return vec(1,k+1);
    if(mn*2>=mx)
    {
        vec res;
        for(int i=1;i<=k;i++)if(!pos[i])res.pb(i);
        return res;
    }
    else
    {
        vec x(0),y(0),z(0);
        for(int i=1;i<=k;i++)if((!pos[i])&&b[i]==mx)x.pb(i);
        for(int i=1;i<=k;i++)if((!pos[i])&&b[i]==mn)x.pb(i);
        for(int i=1;i<=k;i++)if((!pos[i])&&b[i]!=mx&&b[i]!=mn)y.pb(i);
        int tmp=0;
        for(int i=0;i<x.size();z.pb(x[i]),i++)
            while(tmp<y.size()&&y[tmp]<x[i])z.pb(y[tmp++]);
        while(tmp<y.size())z.pb(y[tmp++]);
        int t1=0,t2=0;
        for(int i=now-k+len+1;i<=now;i++)
        {
            if(b[c[i]]==mx&&(!t1))t1=i;
            if(b[c[i]]==mn)t2=i;
        }
        for(int i=0;i<z.size();i++)
        {
            if(b[z[i]]==mx&&(!t1))t1=i+now+1;
            if(b[z[i]]==mn)t2=i+now+1;
        }
        if(t1<t2)return z;
        return vec(1,k+1);
    }
}
void extend()
{
    vec res(1,k+1);
    for(int i=1;i<=min(n-now,k);i++)if(now-k+i+1>0)res=min(res,get(i));
    for(auto i:res)printf("%d ",i),c[++now]=i,a[i]--;
}
int main()
{
    scanf("%d",&k);
    for(int i=1;i<=k;i++)scanf("%d",&a[i]),n+=a[i];
    if(!check()){puts("-1");return 0;}
    while(now<n)extend();
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘vec get(int)’:
./Main.cpp:39:22: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<int>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   39 |         for(int i=0;i<x.size();z.pb(x[i]),i++)
      |                     ~^~~~~~~~~
./Main.cpp:40:22: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<int>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   40 |             while(tmp<y.size()&&y[tmp]<x[i])z.pb(y[tmp++]);
      |                   ~~~^~~~~~~~~
./Main.cpp:41:18: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<int>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   41 |         while(tmp<y.size())z.pb(y[tmp++]);
      |               ~~~^~~~~~~~~
./Main.cpp:48:22: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<int>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   48 |         for(int i=0;i<z.size();i++)
      |                     ~^~~~~~~~~
./Main.cpp: In function ‘int main()’:
./Main.cpp:65:10: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   65 |     scanf("%d",&k);
      |     ~~~~~^~~~~~~~~
./Main.cpp:66:31: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   66 |     for(int i=1;i<=k;i++)scanf("%d",&a[i]),n+=a[i];
      |                          ~~~~~^~~~~~~~~~~~

Judge Result