Submission #331467

#	Time	Username	Problem	Language	Result	Execution time	Memory
331467		Sho10	Devil's Share (RMI19_devil)	C++14	27 / 100	1576 ms	1408 KiB

devil

#include <bits/stdc++.h> //Andrei Alexandru a.k.a Sho10
#define ll long long
#define double long double
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#define all(a) (a).begin(), (a).end()
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define pi pair
#define rc(s) return cout<<s,0
#define endl '\n'
#define mod 1000007
#define PI 3.14159265359
#define MAXN 100005
#define INF 1000000005
#define LINF 1000000000000000005ll
#define CODE_START  ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
int t,cnt[10],k,sz=0;
string ans="";
string s="";
string mn="";
void solve(int nr,int pos){
s+=(nr+'0');
cnt[nr]--;
if(pos==sz){
string mx="";
for(int i=1;i<=k;i++)
{
    mx+='1';
}
for(int i=0;i<s.size()-k+1;i++)
{
    string x=s.substr(i,k);
    mx=max(mx,x);
}
if(mx<mn){
    ans=s;
    mn=mx;
}
cnt[nr]++;
s.pop_back();
return;
}
for(int i=1;i<=4;i++)
{
    if(cnt[i]){
        solve(i,pos+1);
    }
}
cnt[nr]++;
s.pop_back();
return;
}
int32_t main(){
CODE_START;
cin>>t;
while(t--){
    cin>>k;
    if(k==2){
 for(ll i=1; i<=9; i++){
        cin>>cnt[i];
    }
    ll mx=0;
    for(ll i=1;i<=9;i++){
    if(cnt[i]>0){mx=i;}
    }

    string ans="";
    ans.pb('0'+mx);
    cnt[mx]--;
    mx=0;
    for(ll i=9; i>=1; i--){
        if(cnt[i]>0){
                mx=i;
        break;
        }
    }

    ll sum=0;
    for(ll i=1;i<mx;i++){
            if(sum>=cnt[mx]){
                break;
    }
     sum+=cnt[i];
     }
    if(sum>=cnt[mx]){
        for(ll i=1;(i<=mx)&&(cnt[mx]>0);i++){
            while((cnt[i]>0)&&(cnt[mx]>0)){
                    cout<<mx;
            cnt[mx]--;
            if(cnt[i]>0)
                {cout<<i;
            cnt[i]--;
            }
            }
        }
    }
 for(ll i=9;i>=1;i--){
        for(ll j=1;j<=cnt[i];j++){
            cout<<i;
}
    }
    cout<<ans<<endl;
}else {
    mn="";
    for(int i=1;i<=k;i++)
    {
        mn+='9';
    }
    sz=0;
    for(int i=1;i<=9;i++)
    {
        cin>>cnt[i];
        sz+=cnt[i];
    }
    ll pos=4;
    ll sum=0;
    for(ll i=4;i>=1;i--)
    {
     sum+=cnt[i];
     if(sum>=k){
        solve(i,1);
        break;
     }
    }
 cout<<ans<<endl;
}
}
}

Compilation message (stderr)

devil.cpp: In function 'void solve(int, int)':
devil.cpp:34:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 | for(int i=0;i<s.size()-k+1;i++)
      |             ~^~~~~~~~~~~~~
devil.cpp: In function 'int32_t main()':
devil.cpp:119:8: warning: unused variable 'pos' [-Wunused-variable]
  119 |     ll pos=4;
      |        ^~~

• Subtask 1: 13 / 13
• Subtask 2: 14 / 14
• Subtask 3: 0 / 29
• Subtask 4: 0 / 44