Artifact be7ddcd5afb0c754c6de63475d75862a764198b2:
0000: 23 69 6e 63 6c 75 64 65 20 3c 69 6f 73 74 72 65 #include <iostre
0010: 61 6d 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 73 am>..#include <s
0020: 73 74 72 65 61 6d 3e 0d 0a 23 69 6e 63 6c 75 64 stream>..#includ
0030: 65 20 3c 69 6f 6d 61 6e 69 70 3e 0d 0a 23 69 6e e <iomanip>..#in
0040: 63 6c 75 64 65 20 3c 76 65 63 74 6f 72 3e 0d 0a clude <vector>..
0050: 23 69 6e 63 6c 75 64 65 20 3c 73 74 72 69 6e 67 #include <string
0060: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 6d 61 70 >..#include <map
0070: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 73 65 74 >..#include <set
0080: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 61 6c 67 >..#include <alg
0090: 6f 72 69 74 68 6d 3e 0d 0a 23 69 6e 63 6c 75 64 orithm>..#includ
00a0: 65 20 3c 6e 75 6d 65 72 69 63 3e 0d 0a 23 69 6e e <numeric>..#in
00b0: 63 6c 75 64 65 20 3c 69 74 65 72 61 74 6f 72 3e clude <iterator>
00c0: 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 66 75 6e 63 ..#include <func
00d0: 74 69 6f 6e 61 6c 3e 0d 0a 23 69 6e 63 6c 75 64 tional>..#includ
00e0: 65 20 3c 63 6f 6d 70 6c 65 78 3e 0d 0a 23 69 6e e <complex>..#in
00f0: 63 6c 75 64 65 20 3c 71 75 65 75 65 3e 0d 0a 23 clude <queue>..#
0100: 69 6e 63 6c 75 64 65 20 3c 73 74 61 63 6b 3e 0d include <stack>.
0110: 0a 23 69 6e 63 6c 75 64 65 20 3c 63 6d 61 74 68 .#include <cmath
0120: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 63 61 73 >..#include <cas
0130: 73 65 72 74 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 sert>..#include
0140: 3c 63 73 74 72 69 6e 67 3e 0d 0a 75 73 69 6e 67 <cstring>..using
0150: 20 6e 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d namespace std;.
0160: 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f .typedef long lo
0170: 6e 67 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 ng LL;..typedef
0180: 63 6f 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 complex<double>
0190: 43 4d 50 3b 0d 0a 0d 0a 73 74 61 74 69 63 20 63 CMP;....static c
01a0: 6f 6e 73 74 20 4c 4c 20 4d 4f 44 56 41 4c 20 3d onst LL MODVAL =
01b0: 20 31 30 30 30 30 30 30 30 30 37 3b 0d 0a 0d 0a 1000000007;....
01c0: 63 6c 61 73 73 20 50 75 79 6f 50 75 79 6f 20 7b class PuyoPuyo {
01d0: 20 70 75 62 6c 69 63 3a 0d 0a 09 69 6e 74 20 74 public:...int t
01e0: 68 65 43 6f 75 6e 74 28 69 6e 74 20 4c 2c 20 69 heCount(int L, i
01f0: 6e 74 20 4e 29 20 0d 0a 09 7b 0d 0a 09 09 2f 2f nt N) ...{....//
0200: 20 41 34 20 3a 3a 3d 20 55 2a 0d 0a 09 09 2f 2f A4 ::= U*....//
0210: 20 55 20 20 3a 3a 3d 20 78 20 41 33 20 78 20 41 U ::= x A3 x A
0220: 33 20 78 20 41 33 20 78 20 20 20 28 55 3a 3a 3d 3 x A3 x (U::=
0230: 55 5b 30 5d 2c 20 20 55 5b 69 5d 3a 3a 3d 78 20 U[0], U[i]::=x
0240: 41 33 20 55 5b 69 2b 31 5d 2c 20 20 55 5b 4c 2d A3 U[i+1], U[L-
0250: 31 5d 3a 3a 3d 78 29 0d 0a 0d 0a 09 09 76 65 63 1]::=x)......vec
0260: 74 6f 72 3c 4c 4c 3e 20 20 20 20 20 20 20 20 20 tor<LL>
0270: 20 20 41 34 28 4e 2b 31 29 2c 20 41 33 28 4e 2b A4(N+1), A3(N+
0280: 31 29 3b 0d 0a 09 09 76 65 63 74 6f 72 3c 20 76 1);....vector< v
0290: 65 63 74 6f 72 3c 4c 4c 3e 20 3e 20 55 28 4c 2c ector<LL> > U(L,
02a0: 20 76 65 63 74 6f 72 3c 4c 4c 3e 28 4e 2b 31 29 vector<LL>(N+1)
02b0: 29 3b 0d 0a 0d 0a 09 09 41 34 5b 30 5d 20 3d 20 );......A4[0] =
02c0: 41 33 5b 30 5d 20 3d 20 55 5b 4c 2d 31 5d 5b 31 A3[0] = U[L-1][1
02d0: 5d 20 3d 20 31 3b 0d 0a 09 09 66 6f 72 28 69 6e ] = 1;....for(in
02e0: 74 20 6e 3d 31 3b 20 6e 3c 3d 4e 3b 20 2b 2b 6e t n=1; n<=N; ++n
02f0: 29 20 7b 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 ) {.....for(int
0300: 6b 3d 4c 2d 32 3b 20 6b 3e 3d 30 3b 20 2d 2d 6b k=L-2; k>=0; --k
0310: 29 0d 0a 09 09 09 09 66 6f 72 28 69 6e 74 20 7a )......for(int z
0320: 3d 30 3b 20 6e 2d 7a 2d 31 3e 3d 31 3b 20 7a 2b =0; n-z-1>=1; z+
0330: 3d 4c 29 0d 0a 09 09 09 09 09 55 5b 6b 5d 5b 6e =L).......U[k][n
0340: 5d 20 3d 20 28 55 5b 6b 5d 5b 6e 5d 20 2b 20 41 ] = (U[k][n] + A
0350: 33 5b 7a 5d 2a 55 5b 6b 2b 31 5d 5b 6e 2d 7a 2d 3[z]*U[k+1][n-z-
0360: 31 5d 29 20 25 20 4d 4f 44 56 41 4c 3b 0d 0a 09 1]) % MODVAL;...
0370: 09 09 66 6f 72 28 69 6e 74 20 69 3d 4c 3b 20 69 ..for(int i=L; i
0380: 3c 3d 6e 3b 20 69 2b 3d 4c 29 0d 0a 09 09 09 09 <=n; i+=L)......
0390: 41 34 5b 6e 5d 20 3d 20 28 41 34 5b 6e 5d 20 2b A4[n] = (A4[n] +
03a0: 20 34 2a 55 5b 30 5d 5b 69 5d 2a 41 34 5b 6e 2d 4*U[0][i]*A4[n-
03b0: 69 5d 29 20 25 20 4d 4f 44 56 41 4c 3b 0d 0a 09 i]) % MODVAL;...
03c0: 09 09 66 6f 72 28 69 6e 74 20 69 3d 4c 3b 20 69 ..for(int i=L; i
03d0: 3c 3d 6e 3b 20 69 2b 3d 4c 29 0d 0a 09 09 09 09 <=n; i+=L)......
03e0: 41 33 5b 6e 5d 20 3d 20 28 41 33 5b 6e 5d 20 2b A3[n] = (A3[n] +
03f0: 20 33 2a 55 5b 30 5d 5b 69 5d 2a 41 33 5b 6e 2d 3*U[0][i]*A3[n-
0400: 69 5d 29 20 25 20 4d 4f 44 56 41 4c 3b 0d 0a 09 i]) % MODVAL;...
0410: 09 7d 0d 0a 09 09 72 65 74 75 72 6e 20 69 6e 74 .}....return int
0420: 28 41 34 5b 4e 5d 29 3b 0d 0a 09 7d 0d 0a 7d 3b (A4[N]);...}..};
0430: 0d 0a 0d 0a 2f 2f 20 42 45 47 49 4e 20 43 55 54 ....// BEGIN CUT
0440: 20 48 45 52 45 0d 0a 23 69 6e 63 6c 75 64 65 20 HERE..#include
0450: 3c 63 74 69 6d 65 3e 0d 0a 64 6f 75 62 6c 65 20 <ctime>..double
0460: 73 74 61 72 74 5f 74 69 6d 65 3b 20 73 74 72 69 start_time; stri
0470: 6e 67 20 74 69 6d 65 72 28 29 0d 0a 20 7b 20 6f ng timer().. { o
0480: 73 74 72 69 6e 67 73 74 72 65 61 6d 20 6f 73 3b stringstream os;
0490: 20 6f 73 20 3c 3c 20 22 20 28 22 20 3c 3c 20 69 os << " (" << i
04a0: 6e 74 28 28 63 6c 6f 63 6b 28 29 2d 73 74 61 72 nt((clock()-star
04b0: 74 5f 74 69 6d 65 29 2f 43 4c 4f 43 4b 53 5f 50 t_time)/CLOCKS_P
04c0: 45 52 5f 53 45 43 2a 31 30 30 30 29 20 3c 3c 20 ER_SEC*1000) <<
04d0: 22 20 6d 73 65 63 29 22 3b 20 72 65 74 75 72 6e " msec)"; return
04e0: 20 6f 73 2e 73 74 72 28 29 3b 20 7d 0d 0a 74 65 os.str(); }..te
04f0: 6d 70 6c 61 74 65 3c 74 79 70 65 6e 61 6d 65 20 mplate<typename
0500: 54 3e 20 6f 73 74 72 65 61 6d 26 20 6f 70 65 72 T> ostream& oper
0510: 61 74 6f 72 3c 3c 28 6f 73 74 72 65 61 6d 26 20 ator<<(ostream&
0520: 6f 73 2c 20 63 6f 6e 73 74 20 76 65 63 74 6f 72 os, const vector
0530: 3c 54 3e 26 20 76 29 0d 0a 20 7b 20 6f 73 20 3c <T>& v).. { os <
0540: 3c 20 22 7b 20 22 3b 0d 0a 20 20 20 66 6f 72 28 < "{ ";.. for(
0550: 74 79 70 65 6e 61 6d 65 20 76 65 63 74 6f 72 3c typename vector<
0560: 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 61 74 T>::const_iterat
0570: 6f 72 20 69 74 3d 76 2e 62 65 67 69 6e 28 29 3b or it=v.begin();
0580: 20 69 74 21 3d 76 2e 65 6e 64 28 29 3b 20 2b 2b it!=v.end(); ++
0590: 69 74 29 0d 0a 20 20 20 6f 73 20 3c 3c 20 27 5c it).. os << '\
05a0: 22 27 20 3c 3c 20 2a 69 74 20 3c 3c 20 27 5c 22 "' << *it << '\"
05b0: 27 20 3c 3c 20 28 69 74 2b 31 3d 3d 76 2e 65 6e ' << (it+1==v.en
05c0: 64 28 29 20 3f 20 22 22 20 3a 20 22 2c 20 22 29 d() ? "" : ", ")
05d0: 3b 20 6f 73 20 3c 3c 20 22 20 7d 22 3b 20 72 65 ; os << " }"; re
05e0: 74 75 72 6e 20 6f 73 3b 20 7d 0d 0a 76 6f 69 64 turn os; }..void
05f0: 20 76 65 72 69 66 79 5f 63 61 73 65 28 63 6f 6e verify_case(con
0600: 73 74 20 69 6e 74 26 20 45 78 70 65 63 74 65 64 st int& Expected
0610: 2c 20 63 6f 6e 73 74 20 69 6e 74 26 20 52 65 63 , const int& Rec
0620: 65 69 76 65 64 29 20 7b 0d 0a 20 62 6f 6f 6c 20 eived) {.. bool
0630: 6f 6b 20 3d 20 28 45 78 70 65 63 74 65 64 20 3d ok = (Expected =
0640: 3d 20 52 65 63 65 69 76 65 64 29 3b 0d 0a 20 69 = Received);.. i
0650: 66 28 6f 6b 29 20 63 65 72 72 20 3c 3c 20 22 50 f(ok) cerr << "P
0660: 41 53 53 45 44 22 20 3c 3c 20 74 69 6d 65 72 28 ASSED" << timer(
0670: 29 20 3c 3c 20 65 6e 64 6c 3b 20 20 65 6c 73 65 ) << endl; else
0680: 20 7b 20 63 65 72 72 20 3c 3c 20 22 46 41 49 4c { cerr << "FAIL
0690: 45 44 22 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c ED" << timer() <
06a0: 3c 20 65 6e 64 6c 3b 0d 0a 20 63 65 72 72 20 3c < endl;.. cerr <
06b0: 3c 20 22 5c 74 6f 3a 20 5c 22 22 20 3c 3c 20 45 < "\to: \"" << E
06c0: 78 70 65 63 74 65 64 20 3c 3c 20 27 5c 22 27 20 xpected << '\"'
06d0: 3c 3c 20 65 6e 64 6c 20 3c 3c 20 22 5c 74 78 3a << endl << "\tx:
06e0: 20 5c 22 22 20 3c 3c 20 52 65 63 65 69 76 65 64 \"" << Received
06f0: 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c << '\"' << endl
0700: 3b 20 7d 20 7d 0d 0a 23 64 65 66 69 6e 65 20 43 ; } }..#define C
0710: 41 53 45 28 4e 29 20 7b 63 65 72 72 20 3c 3c 20 ASE(N) {cerr <<
0720: 22 54 65 73 74 20 43 61 73 65 20 23 22 20 3c 3c "Test Case #" <<
0730: 20 4e 20 3c 3c 20 22 2e 2e 2e 22 20 3c 3c 20 66 N << "..." << f
0740: 6c 75 73 68 3b 20 73 74 61 72 74 5f 74 69 6d 65 lush; start_time
0750: 3d 63 6c 6f 63 6b 28 29 3b 0d 0a 23 64 65 66 69 =clock();..#defi
0760: 6e 65 20 45 4e 44 09 20 76 65 72 69 66 79 5f 63 ne END. verify_c
0770: 61 73 65 28 5f 2c 20 50 75 79 6f 50 75 79 6f 28 ase(_, PuyoPuyo(
0780: 29 2e 74 68 65 43 6f 75 6e 74 28 4c 2c 20 4e 29 ).theCount(L, N)
0790: 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 7b );}..int main(){
07a0: 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 69 6e ....CASE(0)...in
07b0: 74 20 4c 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 t L = 2; ...int
07c0: 4e 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 5f 20 N = 2; ...int _
07d0: 3d 20 34 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 = 4; ..END..CASE
07e0: 28 31 29 0d 0a 09 69 6e 74 20 4c 20 3d 20 32 3b (1)...int L = 2;
07f0: 20 0d 0a 09 69 6e 74 20 4e 20 3d 20 34 3b 20 0d ...int N = 4; .
0800: 0a 09 69 6e 74 20 5f 20 3d 20 32 38 3b 20 0d 0a ..int _ = 28; ..
0810: 45 4e 44 0d 0a 43 41 53 45 28 32 29 0d 0a 09 69 END..CASE(2)...i
0820: 6e 74 20 4c 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 nt L = 2; ...int
0830: 20 4e 20 3d 20 35 38 3b 20 0d 0a 09 69 6e 74 20 N = 58; ...int
0840: 5f 20 3d 20 38 36 38 32 39 34 36 32 30 3b 20 0d _ = 868294620; .
0850: 0a 45 4e 44 0d 0a 43 41 53 45 28 33 29 0d 0a 09 .END..CASE(3)...
0860: 69 6e 74 20 4c 20 3d 20 34 3b 20 0d 0a 09 69 6e int L = 4; ...in
0870: 74 20 4e 20 3d 20 38 34 3b 20 0d 0a 09 69 6e 74 t N = 84; ...int
0880: 20 5f 20 3d 20 36 32 31 38 37 31 31 35 31 3b 20 _ = 621871151;
0890: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 34 29 0d 0a ..END..CASE(4)..
08a0: 09 69 6e 74 20 4c 20 3d 20 35 3b 20 0d 0a 09 69 .int L = 5; ...i
08b0: 6e 74 20 4e 20 3d 20 38 3b 20 0d 0a 09 69 6e 74 nt N = 8; ...int
08c0: 20 5f 20 3d 20 30 3b 20 0d 0a 45 4e 44 0d 0a 43 _ = 0; ..END..C
08d0: 41 53 45 28 35 29 0d 0a 09 69 6e 74 20 4c 20 3d ASE(5)...int L =
08e0: 20 32 3b 20 0d 0a 09 69 6e 74 20 4e 20 3d 20 31 2; ...int N = 1
08f0: 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 30 3b 20 ; ...int _ = 0;
0900: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 36 29 0d 0a ..END..CASE(6)..
0910: 09 69 6e 74 20 4c 20 3d 20 32 3b 20 0d 0a 09 69 .int L = 2; ...i
0920: 6e 74 20 4e 20 3d 20 31 30 30 30 3b 20 0d 0a 09 nt N = 1000; ...
0930: 69 6e 74 20 5f 20 3d 20 2d 31 3b 20 0d 0a 45 4e int _ = -1; ..EN
0940: 44 0d 0a 43 41 53 45 28 37 29 0d 0a 09 69 6e 74 D..CASE(7)...int
0950: 20 4c 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 4e L = 2; ...int N
0960: 20 3d 20 36 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d = 6; ...int _ =
0970: 20 32 33 32 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 232; ..END..CAS
0980: 45 28 38 29 0d 0a 09 69 6e 74 20 4c 20 3d 20 32 E(8)...int L = 2
0990: 3b 20 0d 0a 09 69 6e 74 20 4e 20 3d 20 38 3b 20 ; ...int N = 8;
09a0: 0d 0a 09 69 6e 74 20 5f 20 3d 20 32 30 39 32 3b ...int _ = 2092;
09b0: 20 0d 0a 45 4e 44 0d 0a 7d 0d 0a 2f 2f 20 45 4e ..END..}..// EN
09c0: 44 20 43 55 54 20 48 45 52 45 0d 0a D CUT HERE..