Submission #108105

#	Time	Username	Problem	Language	Result	Execution time	Memory
108105		brcode	Zalmoxis (BOI18_zalmoxis)	C++14	100 / 100	985 ms	54564 KiB

zalmoxis

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
vector<int> v1;
vector<int> ans;
set<int> s1;
int idx;
void solve(int curr,int level){
    if(v1[idx] == level){
        ans.push_back(curr);
        idx++;
        return;
    }
    solve(2*curr,level-1);
    if(idx == v1.size()||level<=v1[idx]){
        s1.insert(2*curr+1);
        return;
    }
    solve(2*curr+1,level-1);
}
void getans(int curr,int level){
    if(binary_search(ans.begin(),ans.end(),curr)){
        cout<<level<<" ";
        return;
    }
    getans(2*curr,level-1);
    getans(2*curr+1,level-1);
}
int main(){
    int n,k;
    cin>>n>>k;
    for(int i=0;i<n;i++){
        int x;
        cin>>x;
        v1.push_back(x);
    }
    solve(1,30);
    while(s1.size()<k){
        auto hold = *s1.begin();
        s1.erase(hold);
        s1.insert((hold)*2);
        s1.insert((hold)*2 +1);
    }
    for(auto i:s1){
        ans.push_back(i);
    }
    sort(ans.begin(),ans.end());
    getans(1,30);
}

Compilation message (stderr)

zalmoxis.cpp: In function 'void solve(int, int)':
zalmoxis.cpp:17:12: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(idx == v1.size()||level<=v1[idx]){
        ~~~~^~~~~~~~~~~~
zalmoxis.cpp: In function 'int main()':
zalmoxis.cpp:40:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     while(s1.size()<k){
           ~~~~~~~~~^~

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