Submission #121325

#	Time	Username	Problem	Language	Result	Execution time	Memory
121325		BThero	Fibonacci representations (CEOI18_fib)	C++17	50 / 100	43 ms	496 KiB

fib

// Why am I so dumb? :c
// chrono::system_clock::now().time_since_epoch().count()

//#pragma GCC optimize("Ofast")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")

#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>

#define pb push_back
#define mp make_pair

#define all(x) (x).begin(), (x).end()

#define fi first
#define se second

using namespace std;
using namespace __gnu_pbds;

typedef long long ll;   
template<typename T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

const int MOD = (int)1e9 + 7;

map<int, int> cnt;

int dp[2][205];

int n;

int addMod(int a, int b, int m = MOD) {
    a += b;

    if (m <= a) {
        a -= m;
    }

    return a;
}

int mulMod(int a, int b, int m = MOD) {
    return a * 1ll * b % m;
}

int binPow(int a, int b) {
    int ret = 1;

    while (b > 0) {
        if (b & 1) {
            ret = mulMod(ret, a);
        }

        a = mulMod(a, a);
        b >>= 1;
    }

    return ret;
}

int getVal(int x) {
    auto it = cnt.find(x);

    if (it == cnt.end()) {
        return 0;
    }

    return it -> se;
}

void rec() {
    int pos = -1;

    for (auto it : cnt) {
        if (it.se == 2) {
            pos = it.fi;
        }
    }    

    if (pos == -1) {
        return;            
    }

    if (pos == 1) {
        if (getVal(2) == 0) {
            cnt[2] = 1;
            cnt[1] = 0;
        }
        else {
            --cnt[1];
            --cnt[2];
            ++cnt[3];
            rec();
        }
    }
    else if (pos == 2) {
        --cnt[2];
        cnt[1] += 2;
        --cnt[1];
        --cnt[2];
        ++cnt[3];
        rec();        
    }
    else {
        --cnt[pos];
        ++cnt[pos - 1];
        ++cnt[pos - 2];
        --cnt[pos - 1];
        --cnt[pos];
        ++cnt[pos + 1];
        rec();
    }
}

void norm() {
    vector<int> tmp;    

    for (int step = 1; step <= 100; ++step) {
        for (auto it2 = cnt.rbegin(); it2 != cnt.rend(); ++it2) {
            auto it = *it2;

            if (it.se == 1 && getVal(it.fi + 1) == 1) {
                cnt[it.fi] = 0;
                cnt[it.fi + 1] = 0;
                cnt[it.fi + 2] = 1;

            }        
        }        
    }    
}

void solve() {
    scanf("%d", &n);
    
    for (int i = 1; i <= n; ++i) {
        int x;
        scanf("%d", &x);

        ++cnt[x];
        rec();
        norm();          

        dp[0][0] = 1;
        int prev = 0, ptr = 0;
                              
        for (auto it : cnt) {
            if (it.se == 0) {
                continue;
            }

            int cur = it.fi;

            ++ptr;
            dp[0][ptr] = addMod(dp[0][ptr - 1], dp[1][ptr - 1]);
            dp[1][ptr] = mulMod(dp[0][ptr - 1], (cur - prev - 1) / 2);
            dp[1][ptr] = addMod(dp[1][ptr], mulMod(dp[1][ptr - 1], (cur - prev) / 2));
            prev = cur;                    
        }                                                                                  

        printf("%d\n", addMod(dp[0][ptr], dp[1][ptr]));
    }
}

int main() {
    //freopen("fast.txt", "w", stdout);
    int tt = 1;

    while (tt--) {
        solve();
    }

    return 0;
}

Compilation message (stderr)

fib.cpp: In function 'void solve()':
fib.cpp:134:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
     ~~~~~^~~~~~~~~~
fib.cpp:138:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &x);
         ~~~~~^~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 20 / 20
• Subtask 3: 15 / 15
• Subtask 4: 10 / 10
• Subtask 5: 0 / 15
• Subtask 6: 0 / 35