Submission #78916

#	Time	Username	Problem	Language	Result	Execution time	Memory
78916		aminra	Fibonacci representations (CEOI18_fib)	C++14	15 / 100	2256 ms	716 KiB

fib

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
const int MOD = (int)1e9 + 7;
const int MAXN = (int)107;
const int infint = (int)1e9;
ll n, a[MAXN], dp1[MAXN], dp2[MAXN], fib[MAXN], dp[MAXN][MAXN], cp[MAXN], pc[MAXN];
vector<int> v;
ll get()
{
	dp1[0] = 1, dp2[0] = 0;
	for (int i = 1; i < v.size(); i++)
		dp1[i] = (dp1[i - 1] + dp2[i - 1]) % MOD, dp2[i] = dp1[i - 1] * ((v[i] - v[i - 1] - 1) / 2) % MOD + dp2[i - 1] * ((v[i] - v[i - 1]) / 2) % MOD, dp2[i] %= MOD;
	return (dp1[v.size() - 1] + dp2[v.size() - 1]) % MOD;
}
vector<ll> zeck(ll p)
{
	vector<ll> ans;
	for (int i = 18; i >= 1; i--)
		if(p >= fib[i])
		{
			p -= fib[i];
			ans.push_back(i);
		}
	return ans;
}
ll get(ll p)
{
	vector<ll> ans = zeck(p);
	reverse(ans.begin(), ans.end());
	for (int i = 0; i < ans.size(); i++)
		ans[i]--;
	memset(cp, 0, sizeof cp);
	memset(pc, 0, sizeof pc);
	cp[0] = dp[ans[0]][0];
	pc[0] = dp[ans[0]][0] - 1;
	for (int i = 1; i < ans.size(); i++)
	{
		cp[i] = cp[i - 1] * dp[ans[i]][ans[i - 1] + 1];
		cp[i] += (dp[ans[i]][ans[i - 1]] - dp[ans[i]][ans[i - 1] + 1]) * pc[i - 1];
		pc[i] = cp[i - 1] * (dp[ans[i]][ans[i - 1] + 1] - 1);
		pc[i] += (dp[ans[i]][ans[i - 1]] - dp[ans[i]][ans[i - 1] + 1]) * pc[i - 1];
	}
	return cp[ans.size() - 1];
}
void A()
{
	fib[1] = 1;
	fib[2] = 2;
	for (int i = 3; i <= 18; i++)
		fib[i] = fib[i - 1] + fib[i - 2];
	ll cur = 0;
	dp[0][0] = 1;
	for (int i = 1; i <= 18; i++)
		for (int j = 0; j <= i; j++)
		{
			dp[i][j] = 1;
			if(j <= i - 2)
				dp[i][j] += dp[i - 2][j];
		}
	for (int i = 0; i < n; i++)
	{
		cur += fib[a[i]];
		cout << get(cur) << " ";
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> a[i];
	ll mx = 0;
	for (int i = 0; i < n; i++)
		mx = max(mx, a[i]);
	
	if(n <= 15 && mx <= 15)
	{
		A();
		return 0;
	}
	v.push_back(0);
	for (int i = 0; i < n; i++)
	{
		v.push_back(a[i]);
		sort(v.begin(), v.end());
		cout << get() << " ";
	}
}

Compilation message (stderr)

fib.cpp: In function 'll get()':
fib.cpp:13:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 1; i < v.size(); i++)
                  ~~^~~~~~~~~~
fib.cpp: In function 'll get(ll)':
fib.cpp:32:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < ans.size(); i++)
                  ~~^~~~~~~~~~~~
fib.cpp:38:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 1; i < ans.size(); i++)
                  ~~^~~~~~~~~~~~

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