76db68d634 2018-07-28 kinaba: #include <iostream> 76db68d634 2018-07-28 kinaba: #include <sstream> 76db68d634 2018-07-28 kinaba: #include <iomanip> 76db68d634 2018-07-28 kinaba: #include <vector> 76db68d634 2018-07-28 kinaba: #include <string> 76db68d634 2018-07-28 kinaba: #include <map> 76db68d634 2018-07-28 kinaba: #include <set> 76db68d634 2018-07-28 kinaba: #include <algorithm> 76db68d634 2018-07-28 kinaba: #include <numeric> 76db68d634 2018-07-28 kinaba: #include <iterator> 76db68d634 2018-07-28 kinaba: #include <functional> 76db68d634 2018-07-28 kinaba: #include <complex> 76db68d634 2018-07-28 kinaba: #include <queue> 76db68d634 2018-07-28 kinaba: #include <stack> 76db68d634 2018-07-28 kinaba: #include <cmath> 76db68d634 2018-07-28 kinaba: #include <cassert> 76db68d634 2018-07-28 kinaba: #include <tuple> 76db68d634 2018-07-28 kinaba: using namespace std; 76db68d634 2018-07-28 kinaba: typedef long long LL; 76db68d634 2018-07-28 kinaba: typedef complex<double> CMP; 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: static const unsigned MODVAL = 1000000007; 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: template<typename T> 76db68d634 2018-07-28 kinaba: struct DP2 76db68d634 2018-07-28 kinaba: { 76db68d634 2018-07-28 kinaba: const int N1, N2; 76db68d634 2018-07-28 kinaba: vector<T> data; 76db68d634 2018-07-28 kinaba: DP2(int N1, int N2, const T& t = T()) 76db68d634 2018-07-28 kinaba: : N1(N1), N2(N2), data(N1*N2, t) { 76db68d634 2018-07-28 kinaba: assert(data.size() * sizeof(T)<(1 << 28)); 76db68d634 2018-07-28 kinaba: } 76db68d634 2018-07-28 kinaba: T& operator()(int i1, int i2) 76db68d634 2018-07-28 kinaba: { 76db68d634 2018-07-28 kinaba: return data[(i1*N2) + i2]; 76db68d634 2018-07-28 kinaba: } 76db68d634 2018-07-28 kinaba: void swap(DP2& rhs) 76db68d634 2018-07-28 kinaba: { 76db68d634 2018-07-28 kinaba: data.swap(rhs.data); 76db68d634 2018-07-28 kinaba: } 76db68d634 2018-07-28 kinaba: }; 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: class SumPyramid { public: 76db68d634 2018-07-28 kinaba: int countPyramids(int levels, int top) 76db68d634 2018-07-28 kinaba: { 76db68d634 2018-07-28 kinaba: vector<int> C(1, 1); 76db68d634 2018-07-28 kinaba: for (int _ = 1; _ < levels; ++_) { 76db68d634 2018-07-28 kinaba: vector<int> CC(C.size() + 1); 76db68d634 2018-07-28 kinaba: for (size_t i = 0; i < CC.size(); ++i) 76db68d634 2018-07-28 kinaba: CC[i] = min(top + 1, (i ? C[i - 1] : 0) + (i<C.size() ? C[i] : 0)); 76db68d634 2018-07-28 kinaba: C = CC; 76db68d634 2018-07-28 kinaba: } 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: DP2<int> memo(levels, top+1, -1); 76db68d634 2018-07-28 kinaba: return rec(levels - 1, 0, top, C, memo); 76db68d634 2018-07-28 kinaba: } 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: int rec(int n, int k, int sum, const vector<int>& C, DP2<int>& memo) 76db68d634 2018-07-28 kinaba: { 76db68d634 2018-07-28 kinaba: // #{x,y,..,z} where nCk x + nC{k+1} y + ... + nCn z == sum 76db68d634 2018-07-28 kinaba: if (memo(k, sum) != -1) 76db68d634 2018-07-28 kinaba: return memo(k, sum); 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: if (k == n) 76db68d634 2018-07-28 kinaba: return memo(k, sum) = 1; 76db68d634 2018-07-28 kinaba: int ans = 0; 76db68d634 2018-07-28 kinaba: for (int v = 0; v*C[k] <= sum; ++v) 76db68d634 2018-07-28 kinaba: ans = (ans + rec(n, k+1, sum - v*C[k], C, memo)) % MODVAL; 76db68d634 2018-07-28 kinaba: return memo(k, sum) = ans; 76db68d634 2018-07-28 kinaba: } 76db68d634 2018-07-28 kinaba: }; 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: // BEGIN CUT HERE 76db68d634 2018-07-28 kinaba: #include <ctime> 76db68d634 2018-07-28 kinaba: double start_time; string timer() 76db68d634 2018-07-28 kinaba: { ostringstream os; os << " (" << int((clock()-start_time)/CLOCKS_PER_SEC*1000) << " msec)"; return os.str(); } 76db68d634 2018-07-28 kinaba: template<typename T> ostream& operator<<(ostream& os, const vector<T>& v) 76db68d634 2018-07-28 kinaba: { os << "{ "; 76db68d634 2018-07-28 kinaba: for(typename vector<T>::const_iterator it=v.begin(); it!=v.end(); ++it) 76db68d634 2018-07-28 kinaba: os << '\"' << *it << '\"' << (it+1==v.end() ? "" : ", "); os << " }"; return os; } 76db68d634 2018-07-28 kinaba: void verify_case(const int& Expected, const int& Received) { 76db68d634 2018-07-28 kinaba: bool ok = (Expected == Received); 76db68d634 2018-07-28 kinaba: if(ok) cerr << "PASSED" << timer() << endl; else { cerr << "FAILED" << timer() << endl; 76db68d634 2018-07-28 kinaba: cerr << "\to: \"" << Expected << '\"' << endl << "\tx: \"" << Received << '\"' << endl; } } 76db68d634 2018-07-28 kinaba: #define CASE(N) {cerr << "Test Case #" << N << "..." << flush; start_time=clock(); 76db68d634 2018-07-28 kinaba: #define END verify_case(_, SumPyramid().countPyramids(levels, top));} 76db68d634 2018-07-28 kinaba: int main(){ 76db68d634 2018-07-28 kinaba: 76db68d634 2018-07-28 kinaba: CASE(0) 76db68d634 2018-07-28 kinaba: int levels = 1; 76db68d634 2018-07-28 kinaba: int top = 47; 76db68d634 2018-07-28 kinaba: int _ = 1; 76db68d634 2018-07-28 kinaba: END 76db68d634 2018-07-28 kinaba: CASE(1) 76db68d634 2018-07-28 kinaba: int levels = 2; 76db68d634 2018-07-28 kinaba: int top = 10; 76db68d634 2018-07-28 kinaba: int _ = 11; 76db68d634 2018-07-28 kinaba: END 76db68d634 2018-07-28 kinaba: CASE(2) 76db68d634 2018-07-28 kinaba: int levels = 3; 76db68d634 2018-07-28 kinaba: int top = 2; 76db68d634 2018-07-28 kinaba: int _ = 4; 76db68d634 2018-07-28 kinaba: END 76db68d634 2018-07-28 kinaba: CASE(3) 76db68d634 2018-07-28 kinaba: int levels = 5; 76db68d634 2018-07-28 kinaba: int top = 7; 76db68d634 2018-07-28 kinaba: int _ = 18; 76db68d634 2018-07-28 kinaba: END 76db68d634 2018-07-28 kinaba: CASE(4) 76db68d634 2018-07-28 kinaba: int levels = 1000; 76db68d634 2018-07-28 kinaba: int top = 1000; 76db68d634 2018-07-28 kinaba: int _ = -1; 76db68d634 2018-07-28 kinaba: END 76db68d634 2018-07-28 kinaba: /* 76db68d634 2018-07-28 kinaba: CASE(5) 76db68d634 2018-07-28 kinaba: int levels = ; 76db68d634 2018-07-28 kinaba: int top = ; 76db68d634 2018-07-28 kinaba: int _ = ; 76db68d634 2018-07-28 kinaba: END 76db68d634 2018-07-28 kinaba: */ 76db68d634 2018-07-28 kinaba: } 76db68d634 2018-07-28 kinaba: // END CUT HERE