e87d5512af 2011-06-03 kinaba: #include <iostream> e87d5512af 2011-06-03 kinaba: #include <sstream> e87d5512af 2011-06-03 kinaba: #include <iomanip> e87d5512af 2011-06-03 kinaba: #include <vector> e87d5512af 2011-06-03 kinaba: #include <string> e87d5512af 2011-06-03 kinaba: #include <map> e87d5512af 2011-06-03 kinaba: #include <set> e87d5512af 2011-06-03 kinaba: #include <algorithm> e87d5512af 2011-06-03 kinaba: #include <numeric> e87d5512af 2011-06-03 kinaba: #include <iterator> e87d5512af 2011-06-03 kinaba: #include <functional> e87d5512af 2011-06-03 kinaba: #include <complex> e87d5512af 2011-06-03 kinaba: #include <queue> e87d5512af 2011-06-03 kinaba: #include <stack> e87d5512af 2011-06-03 kinaba: #include <cmath> e87d5512af 2011-06-03 kinaba: #include <cassert> e87d5512af 2011-06-03 kinaba: #include <cstring> e87d5512af 2011-06-03 kinaba: using namespace std; e87d5512af 2011-06-03 kinaba: typedef long long LL; e87d5512af 2011-06-03 kinaba: typedef complex<double> CMP; e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: static const int MODVAL = 1000000007; // must be prime for op/ e87d5512af 2011-06-03 kinaba: struct mint e87d5512af 2011-06-03 kinaba: { e87d5512af 2011-06-03 kinaba: int val; e87d5512af 2011-06-03 kinaba: mint():val(0){} e87d5512af 2011-06-03 kinaba: mint(int x):val(x%MODVAL) {} e87d5512af 2011-06-03 kinaba: mint(LL x):val(x%MODVAL) {} e87d5512af 2011-06-03 kinaba: }; e87d5512af 2011-06-03 kinaba: mint operator+(mint x, mint y) { return x.val+y.val; } e87d5512af 2011-06-03 kinaba: mint operator*(mint x, mint y) { return LL(x.val)*y.val; } e87d5512af 2011-06-03 kinaba: mint POW(mint x, int e) { e87d5512af 2011-06-03 kinaba: mint v = 1; e87d5512af 2011-06-03 kinaba: for(;e;x=x*x,e>>=1) e87d5512af 2011-06-03 kinaba: if(e&1) e87d5512af 2011-06-03 kinaba: v=v*x; e87d5512af 2011-06-03 kinaba: return v; e87d5512af 2011-06-03 kinaba: } e87d5512af 2011-06-03 kinaba: mint operator/(mint x, mint y) { return x * POW(y, MODVAL-2); } e87d5512af 2011-06-03 kinaba: vector<mint> FAC_(1,1); e87d5512af 2011-06-03 kinaba: void FAC_INIT(int n) { while( FAC_.size()<=n ) FAC_.push_back( FAC_.back()*int(FAC_.size()) ); } e87d5512af 2011-06-03 kinaba: mint FAC(int x) { return FAC_[x]; } e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: class CubeBuilding { public: e87d5512af 2011-06-03 kinaba: int getCount(int R, int G, int B, int N) e87d5512af 2011-06-03 kinaba: { e87d5512af 2011-06-03 kinaba: FAC_INIT(25); e87d5512af 2011-06-03 kinaba: memo2.assign(26*26*26*26, -1); e87d5512af 2011-06-03 kinaba: memo2all.assign(26*26*26*26, -1); e87d5512af 2011-06-03 kinaba: memo3.assign(26*26*26*26, -1); e87d5512af 2011-06-03 kinaba: return (dim3(R,G,B,N,N)+dim3(G,B,R,N,N)+dim3(B,G,R,N,N)).val; e87d5512af 2011-06-03 kinaba: } e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: vector<int> memo3; e87d5512af 2011-06-03 kinaba: mint dim3(int R, int G, int B, int n, int N) e87d5512af 2011-06-03 kinaba: { e87d5512af 2011-06-03 kinaba: if( n == 0 ) e87d5512af 2011-06-03 kinaba: return (R+G+B==0 ? 1 : 0); e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: if(G>B) swap(G,B); e87d5512af 2011-06-03 kinaba: int key = ((R*26+G)*26+B)*26+n; e87d5512af 2011-06-03 kinaba: if( memo3[key] >= 0 ) e87d5512af 2011-06-03 kinaba: return memo3[key]; e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: mint total = 0; e87d5512af 2011-06-03 kinaba: for(int r=0; r<=R; ++r) e87d5512af 2011-06-03 kinaba: for(int g=0; g<=G; ++g) e87d5512af 2011-06-03 kinaba: for(int b=0; b<=B; ++b) e87d5512af 2011-06-03 kinaba: total = total + dim2all(r,g,b,N)*dim3(R-r,G-g,B-b,n-1,N); e87d5512af 2011-06-03 kinaba: memo3[key] = total.val; e87d5512af 2011-06-03 kinaba: return total; e87d5512af 2011-06-03 kinaba: } e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: vector<int> memo2; e87d5512af 2011-06-03 kinaba: mint dim2(int R, int G, int B, int n) e87d5512af 2011-06-03 kinaba: { e87d5512af 2011-06-03 kinaba: if( n == 0 ) return (R+G+B==0 ? 1 : 0); e87d5512af 2011-06-03 kinaba: if( R == 0 ) return (G+B==0 ? 1 : 0); e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: if(G>B) swap(G,B); e87d5512af 2011-06-03 kinaba: int key = ((R*26+G)*26+B)*26+n; e87d5512af 2011-06-03 kinaba: if( memo2[key] >= 0 ) e87d5512af 2011-06-03 kinaba: return memo2[key]; e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: mint total = 0; e87d5512af 2011-06-03 kinaba: for(int r=1; r<=R; ++r) e87d5512af 2011-06-03 kinaba: for(int g=0; g<=G; ++g) { e87d5512af 2011-06-03 kinaba: int b = n-r-g; e87d5512af 2011-06-03 kinaba: if( 0<=b && b<=B ) e87d5512af 2011-06-03 kinaba: total = total + dim2all(R-r,G-g,B-b,n) * (FAC(n-1)/FAC(r-1)/FAC(g)/FAC(b)); e87d5512af 2011-06-03 kinaba: } e87d5512af 2011-06-03 kinaba: memo2[key] = total.val; e87d5512af 2011-06-03 kinaba: return total; e87d5512af 2011-06-03 kinaba: } e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: vector<int> memo2all; e87d5512af 2011-06-03 kinaba: mint dim2all(int R, int G, int B, int n) e87d5512af 2011-06-03 kinaba: { e87d5512af 2011-06-03 kinaba: int key = ((R*26+G)*26+B)*26+n; e87d5512af 2011-06-03 kinaba: if( memo2all[key] >= 0 ) e87d5512af 2011-06-03 kinaba: return memo2all[key]; e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: mint total = 0; e87d5512af 2011-06-03 kinaba: for(int m=0; m<=n; ++m) e87d5512af 2011-06-03 kinaba: total = total + dim2(R,G,B,m)*(FAC(n)/FAC(m)/FAC(n-m)); e87d5512af 2011-06-03 kinaba: memo2all[key] = total.val; e87d5512af 2011-06-03 kinaba: return total; e87d5512af 2011-06-03 kinaba: } e87d5512af 2011-06-03 kinaba: }; e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: // BEGIN CUT HERE e87d5512af 2011-06-03 kinaba: #include <ctime> e87d5512af 2011-06-03 kinaba: double start_time; string timer() e87d5512af 2011-06-03 kinaba: { ostringstream os; os << " (" << int((clock()-start_time)/CLOCKS_PER_SEC*1000) << " msec)"; return os.str(); } e87d5512af 2011-06-03 kinaba: template<typename T> ostream& operator<<(ostream& os, const vector<T>& v) e87d5512af 2011-06-03 kinaba: { os << "{ "; e87d5512af 2011-06-03 kinaba: for(typename vector<T>::const_iterator it=v.begin(); it!=v.end(); ++it) e87d5512af 2011-06-03 kinaba: os << '\"' << *it << '\"' << (it+1==v.end() ? "" : ", "); os << " }"; return os; } e87d5512af 2011-06-03 kinaba: void verify_case(const int& Expected, const int& Received) { e87d5512af 2011-06-03 kinaba: bool ok = (Expected == Received); e87d5512af 2011-06-03 kinaba: if(ok) cerr << "PASSED" << timer() << endl; else { cerr << "FAILED" << timer() << endl; e87d5512af 2011-06-03 kinaba: cerr << "\to: \"" << Expected << '\"' << endl << "\tx: \"" << Received << '\"' << endl; } } e87d5512af 2011-06-03 kinaba: #define CASE(N) {cerr << "Test Case #" << N << "..." << flush; start_time=clock(); e87d5512af 2011-06-03 kinaba: #define END verify_case(_, CubeBuilding().getCount(R, G, B, N));} e87d5512af 2011-06-03 kinaba: int main(){ e87d5512af 2011-06-03 kinaba: e87d5512af 2011-06-03 kinaba: CASE(0) e87d5512af 2011-06-03 kinaba: int R = 1; e87d5512af 2011-06-03 kinaba: int G = 0; e87d5512af 2011-06-03 kinaba: int B = 1; e87d5512af 2011-06-03 kinaba: int N = 2; e87d5512af 2011-06-03 kinaba: int _ = 4; e87d5512af 2011-06-03 kinaba: END e87d5512af 2011-06-03 kinaba: CASE(1) e87d5512af 2011-06-03 kinaba: int R = 1; e87d5512af 2011-06-03 kinaba: int G = 1; e87d5512af 2011-06-03 kinaba: int B = 2; e87d5512af 2011-06-03 kinaba: int N = 1; e87d5512af 2011-06-03 kinaba: int _ = 0; e87d5512af 2011-06-03 kinaba: END e87d5512af 2011-06-03 kinaba: CASE(2) e87d5512af 2011-06-03 kinaba: int R = 2; e87d5512af 2011-06-03 kinaba: int G = 2; e87d5512af 2011-06-03 kinaba: int B = 1; e87d5512af 2011-06-03 kinaba: int N = 3; e87d5512af 2011-06-03 kinaba: int _ = 162; e87d5512af 2011-06-03 kinaba: END e87d5512af 2011-06-03 kinaba: CASE(3) e87d5512af 2011-06-03 kinaba: int R = 0; e87d5512af 2011-06-03 kinaba: int G = 0; e87d5512af 2011-06-03 kinaba: int B = 10; e87d5512af 2011-06-03 kinaba: int N = 12; e87d5512af 2011-06-03 kinaba: int _ = 372185933; e87d5512af 2011-06-03 kinaba: END e87d5512af 2011-06-03 kinaba: CASE(4) e87d5512af 2011-06-03 kinaba: int R = 25; e87d5512af 2011-06-03 kinaba: int G = 25; e87d5512af 2011-06-03 kinaba: int B = 25; e87d5512af 2011-06-03 kinaba: int N = 25; e87d5512af 2011-06-03 kinaba: int _ = -1; e87d5512af 2011-06-03 kinaba: END e87d5512af 2011-06-03 kinaba: /* e87d5512af 2011-06-03 kinaba: CASE(5) e87d5512af 2011-06-03 kinaba: int R = ; e87d5512af 2011-06-03 kinaba: int G = ; e87d5512af 2011-06-03 kinaba: int B = ; e87d5512af 2011-06-03 kinaba: int N = ; e87d5512af 2011-06-03 kinaba: int _ = ; e87d5512af 2011-06-03 kinaba: END e87d5512af 2011-06-03 kinaba: */ e87d5512af 2011-06-03 kinaba: } e87d5512af 2011-06-03 kinaba: // END CUT HERE