Submission #254606

#	Time	Username	Problem	Language	Result	Execution time	Memory
254606		MarcoMeijer	Toys (CEOI18_toy)	C++14	100 / 100	2522 ms	86788 KiB

toy

#include <bits/stdc++.h>
using namespace std;

// macros
typedef long long ll;
typedef long double ld;
typedef pair<int, int> ii;
typedef pair<ll, ll> lll;
typedef tuple<int, int, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<ll> vll;
typedef vector<lll> vlll;
#define REP(a,b,c) for(int a=int(b); a<int(c); a++)
#define RE(a,c) REP(a,0,c)
#define RE1(a,c) REP(a,1,c+1)
#define REI(a,b,c) REP(a,b,c+1)
#define REV(a,b,c) for(int a=int(c-1); a>=int(b); a--)
#define FOR(a,b) for(auto& a : b)
#define all(a) a.begin(), a.end()
#define INF 1e9
#define EPS 1e-9
#define pb push_back
#define popb pop_back
#define fi first
#define se second
#define sz size()

// input
template<class T> void IN(T& x) {cin >> x;}
template<class H, class... T> void IN(H& h, T&... t) {IN(h); IN(t...); }

// output
template<class T1, class T2> void OUT(const pair<T1,T2>& x);
template<class T> void OUT(const T& x) {cout << x;}
template<class H, class... T> void OUT(const H& h, const T&... t) {OUT(h); OUT(t...); }
template<class T1, class T2> void OUT(const pair<T1,T2>& x) {OUT(x.fi," ",x.se);}
template<class... T> void OUTL(const T&... t) {OUT(t..., "\n"); }

//===================//
//  Added libraries  //
//===================//

//===================//
//end added libraries//
//===================//

void program();
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    program();
}


//===================//
//   begin program   //
//===================//

const int MX = 5e5;
int n, m;

vi f;
map<int, set<int>> mem;
set<int> visited;

vi getFactors(ll x) {
    vi ans;
    for(ll y=2;y*y<=x;y++) {
        if((x % y) == 0) {
            ans.pb(y);
        }
    }
    return ans;
}
set<int>& getAns(int x) {
    if(visited.count(x)) return mem[x];
    visited.insert(x);
    vi f = getFactors(x);
    set<int>& ans = mem[x];
    ans.insert(x-1);
    for(int i:f) {
        set<int>& oans = getAns(x/i);
        for(int o:oans) ans.insert(o+i-1);
    }
    return ans;
}

void program() {
    IN(n);

    set<int>& ans = getAns(n);
    OUTL(ans.sz);
    bool f=1;
    FOR(i,ans) OUT(f?"":" ",i), f=0; OUTL();
}

Compilation message (stderr)

toy.cpp: In function 'void program()':
toy.cpp:20:18: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
 #define FOR(a,b) for(auto& a : b)
                  ^
toy.cpp:97:5: note: in expansion of macro 'FOR'
     FOR(i,ans) OUT(f?"":" ",i), f=0; OUTL();
     ^~~
toy.cpp:97:38: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
     FOR(i,ans) OUT(f?"":" ",i), f=0; OUTL();
                                      ^~~~

• Subtask 1: 19 / 19
• Subtask 2: 20 / 20
• Subtask 3: 20 / 20
• Subtask 4: 20 / 20
• Subtask 5: 21 / 21