Torni

#include <iostream>
#include <algorithm>
#include <vector>
 
const int MOD = 1000000007;
const int MAX = 1000009;
 
long tapoja2(int n);
long tapoja12(int n);
 
long tapoja1(int n) {
	static int _t[MAX]; if (_t[n]) return _t[n];
	long summa = 0;
	summa += 1;
	for (int i = 1; i < n; ++i) {
		summa += tapoja1(n - i);
		summa %= MOD;
	}
	return _t[n] = summa;
}
 
long tapoja(int n) {
	static int _t[MAX]; if (_t[n]) return _t[n];
	long summa = 0;
	summa += tapoja2(n);
	summa += tapoja12(n);
	summa %= MOD;
	return _t[n] = summa;
}
 
long tapoja2(int n) {
	static int _t[MAX]; if (_t[n]) return _t[n];
	long summa = 0;
 
	// Täytetään 2xn-palalla loppuun.
	summa += 1;
	summa %= MOD;
	// Täytetään 2xi-palalla ja rekursio.
	for (int i = 1; i < n; ++i) {
		summa += tapoja(n - i);
		summa %= MOD;
	}
	return _t[n] = summa;
}
 
long tapoja12(int n) {
	static int _t[MAX]; if (_t[n]) return _t[n];
	long summa = 0;
	// Täytetään 1x*-paloilla korkeuteen i ja rekursio 2x*-paloihin.
	for (int i = 1; i < n; ++i) {
		long t1 = tapoja1(i);
		long tt1 = (t1 * t1) % MOD;
		summa += tt1 * tapoja2(n - i);
		summa %= MOD;
	}
	// Täytetään 1x*-paloilla loppuun.
	long t1 = tapoja1(n);
	summa += t1 * t1;
	summa %= MOD;
	return _t[n] = summa;
}
 
int main() {
	int t;
	std::cin >> t;
	for (int i = 0; i < t; ++i) {
		int n;
		std::cin >> n;
		std::cout << tapoja(n) << std::endl;
	}
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED