Submission #335435

#	Time	Username	Problem	Language	Result	Execution time	Memory
335435		3funnn	Zalmoxis (BOI18_zalmoxis)	C++17	100 / 100	720 ms	40432 KiB

zalmoxis

#include <bits/stdc++.h>
#define mp make_pair
#define pii pair <int, int>
 
using namespace std;
 
int need;
 
void dfs(int val){
    if (need == 0 || val == 0){
        cout << val << " ";
        return;
    }
    --need;
    dfs(val - 1);
    dfs(val - 1);
}
 
int main()
{
    int N, K;
    cin >> N >> K;
    vector <pii> a(N, mp(0, 0));
    for (int i = 0;i < N;++i)
        cin >> a[i].first;
    for (int p = 0;p < 30;++p){
        vector <pii> b;
        for (int i = 0;i < a.size();++i){
            if (a[i].first <= p){
                int j = i, cnt = 0;
                while (j < a.size() && a[j].first <= p){
                    cnt += (1 << a[j].first);
                    b.push_back(a[j]);
                    ++j;
                }
                if ((cnt >> p) & 1)
                    b.push_back(mp(p, 1));
                i = j - 1;
            }
            else
                b.push_back(a[i]);
        }
        a = b;
    }
    need = N + K - a.size();
    for (int i = 0;i < a.size();++i){
        if (a[i].second == 0)
            cout << a[i].first << " ";
        else
            dfs(a[i].first);
    }
    return 0;
}

Compilation message (stderr)

zalmoxis.cpp: In function 'int main()':
zalmoxis.cpp:28:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |         for (int i = 0;i < a.size();++i){
      |                        ~~^~~~~~~~~~
zalmoxis.cpp:31:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |                 while (j < a.size() && a[j].first <= p){
      |                        ~~^~~~~~~~~~
zalmoxis.cpp:46:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |     for (int i = 0;i < a.size();++i){
      |                    ~~^~~~~~~~~~

• Subtask 1: 30 / 30
• Subtask 2: 70 / 70