#include <bits//stdc++.h> using namespace std; #include "messy.h" mt19937 rg(time(nullptr)); std::vector<int> restore_permutation(int n, int w, int r) { string a=string(n,'0'),cur=a; vector<int>ind(n),ans(n); for(int i=0;i<n;++i){ a[i]='1'; ind[i]=i; add_element(a); } int id,it,cnt=0; compile_set(); vector<bool>vis(n); for(int i=0;i<n;++i){ for(int j=0;j<n;++j)vis[j]=0; for(;;){ id=rg()%ind.size(); if(vis[it=ind[id]])continue; vis[it]=1; cur[it]='1'; cnt++; assert(cnt<=r); if(check_element(cur)){ ans[i]=it; ind.erase(begin(ind)+id); break; } cur[it]='0'; } } return ans; }

messy.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
messy_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~