Submission #121271

#	Time	Username	Problem	Language	Result	Execution time	Memory
121271		BThero	Fibonacci representations (CEOI18_fib)	C++17	20 / 100	4048 ms	1524 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;

int dp[2][100005];

ll fibo[100005];

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;
}

void solve() {
    scanf("%d", &n);
    int ans = 1;

    set<int> S;
    S.insert(0);

    fibo[1] = 1;
    fibo[2] = 2;

    for (int i = 3; i <= 50; ++i) {
        fibo[i] = fibo[i - 1] + fibo[i - 2];
    }

    ll sum = 0;

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

        if (n <= 15) {
            sum += fibo[x];            
            S.clear();

            for (ll tmp = sum; tmp > 0;) {
                int p = upper_bound(fibo + 1, fibo + 50 + 1, tmp) - fibo - 1;
                tmp -= fibo[p];
                S.insert(p);
            }
        }
        else { 
            S.insert(x);
        }
              
        dp[0][0] = 1;
        int prev = 0, ptr = 0;

        for (int cur : S) {
            ++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() {
    int tt = 1;

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

    return 0;
}

Compilation message (stderr)

fib.cpp: In function 'void solve()':
fib.cpp:64:9: warning: unused variable 'ans' [-Wunused-variable]
     int ans = 1;
         ^~~
fib.cpp:63:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
     ~~~~~^~~~~~~~~~
fib.cpp:80: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: 0 / 20
• Subtask 3: 15 / 15
• Subtask 4: 0 / 10
• Subtask 5: 0 / 15
• Subtask 6: 0 / 35