Hex Artifact Content
Not logged in

Artifact 2dc990563f003d056269eaf3f69bce652e119874:


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 74 75 70 6c 65 3e 0d 0a 75 73 69 6e 67 20 6e  <tuple>..using n
0150: 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d 0a 74  amespace std;..t
0160: 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67  ypedef long long
0170: 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 63 6f   LL;..typedef co
0180: 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 43 4d  mplex<double> CM
0190: 50 3b 0d 0a 0d 0a 73 74 61 74 69 63 20 63 6f 6e  P;....static con
01a0: 73 74 20 75 6e 73 69 67 6e 65 64 20 4d 4f 44 56  st unsigned MODV
01b0: 41 4c 20 3d 20 31 30 30 30 30 30 30 30 30 37 3b  AL = 1000000007;
01c0: 0d 0a 73 74 72 75 63 74 20 6d 69 6e 74 0d 0a 7b  ..struct mint..{
01d0: 0d 0a 09 75 6e 73 69 67 6e 65 64 20 76 61 6c 3b  ...unsigned val;
01e0: 0d 0a 09 6d 69 6e 74 28 29 3a 76 61 6c 28 30 29  ...mint():val(0)
01f0: 7b 7d 0d 0a 09 6d 69 6e 74 28 69 6e 74 20 20 20  {}...mint(int   
0200: 20 20 20 78 29 3a 76 61 6c 28 78 25 4d 4f 44 56     x):val(x%MODV
0210: 41 4c 29 20 7b 7d 0d 0a 09 6d 69 6e 74 28 75 6e  AL) {}...mint(un
0220: 73 69 67 6e 65 64 20 78 29 3a 76 61 6c 28 78 25  signed x):val(x%
0230: 4d 4f 44 56 41 4c 29 20 7b 7d 0d 0a 09 6d 69 6e  MODVAL) {}...min
0240: 74 28 4c 4c 20 20 20 20 20 20 20 78 29 3a 76 61  t(LL       x):va
0250: 6c 28 78 25 4d 4f 44 56 41 4c 29 20 7b 7d 0d 0a  l(x%MODVAL) {}..
0260: 7d 3b 0d 0a 6d 69 6e 74 26 20 6f 70 65 72 61 74  };..mint& operat
0270: 6f 72 2b 3d 28 6d 69 6e 74 26 20 78 2c 20 6d 69  or+=(mint& x, mi
0280: 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78  nt y) { return x
0290: 20 3d 20 78 2e 76 61 6c 2b 79 2e 76 61 6c 3b 20   = x.val+y.val; 
02a0: 7d 0d 0a 6d 69 6e 74 20 6f 70 65 72 61 74 6f 72  }..mint operator
02b0: 2b 28 6d 69 6e 74 20 78 2c 20 6d 69 6e 74 20 79  +(mint x, mint y
02c0: 29 20 7b 20 72 65 74 75 72 6e 20 78 2b 3d 79 3b  ) { return x+=y;
02d0: 20 7d 0d 0a 0d 0a 63 6c 61 73 73 20 41 6c 74 65   }....class Alte
02e0: 72 6e 61 74 69 76 65 50 69 6c 65 73 20 7b 20 70  rnativePiles { p
02f0: 75 62 6c 69 63 3a 0d 0a 09 69 6e 74 20 63 6f 75  ublic:...int cou
0300: 6e 74 28 73 74 72 69 6e 67 20 43 2c 20 69 6e 74  nt(string C, int
0310: 20 4d 29 0d 0a 09 7b 0d 0a 09 09 76 65 63 74 6f   M)...{....vecto
0320: 72 3c 6d 69 6e 74 3e 20 64 70 28 28 4d 2b 31 29  r<mint> dp((M+1)
0330: 2a 4d 29 3b 0d 0a 09 09 64 70 5b 30 5d 20 3d 20  *M);....dp[0] = 
0340: 31 3b 0d 0a 0d 0a 09 09 66 6f 72 28 63 68 61 72  1;......for(char
0350: 20 63 20 3a 20 43 29 0d 0a 09 09 7b 0d 0a 09 09   c : C)....{....
0360: 09 76 65 63 74 6f 72 3c 6d 69 6e 74 3e 20 64 70  .vector<mint> dp
0370: 32 28 28 4d 2b 31 29 2a 4d 29 3b 0d 0a 09 09 09  2((M+1)*M);.....
0380: 66 6f 72 28 69 6e 74 20 6d 3d 30 3b 20 6d 3c 3d  for(int m=0; m<=
0390: 4d 3b 20 2b 2b 6d 29 0d 0a 09 09 09 66 6f 72 28  M; ++m).....for(
03a0: 69 6e 74 20 72 3d 30 3b 20 72 3c 4d 3b 20 2b 2b  int r=0; r<M; ++
03b0: 72 29 20 7b 0d 0a 09 09 09 09 69 6e 74 20 6d 72  r) {......int mr
03c0: 20 3d 20 6d 2a 4d 2b 72 3b 0d 0a 09 09 09 09 69   = m*M+r;......i
03d0: 66 28 28 63 3d 3d 27 52 27 20 7c 7c 20 63 3d 3d  f((c=='R' || c==
03e0: 27 57 27 29 20 26 26 20 6d 2b 31 3c 3d 4d 29 0d  'W') && m+1<=M).
03f0: 0a 09 09 09 09 09 64 70 32 5b 28 6d 2b 31 29 2a  ......dp2[(m+1)*
0400: 4d 2b 28 72 2b 31 29 25 4d 5d 20 2b 3d 20 64 70  M+(r+1)%M] += dp
0410: 5b 6d 72 5d 3b 0d 0a 09 09 09 09 69 66 28 28 63  [mr];......if((c
0420: 3d 3d 27 47 27 20 7c 7c 20 63 3d 3d 27 57 27 29  =='G' || c=='W')
0430: 20 26 26 20 6d 2d 31 3e 3d 30 29 0d 0a 09 09 09   && m-1>=0).....
0440: 09 09 64 70 32 5b 28 6d 2d 31 29 2a 4d 2b 72 5d  ..dp2[(m-1)*M+r]
0450: 20 2b 3d 20 64 70 5b 6d 72 5d 3b 0d 0a 09 09 09   += dp[mr];.....
0460: 09 69 66 28 63 3d 3d 27 42 27 20 7c 7c 20 63 3d  .if(c=='B' || c=
0470: 3d 27 57 27 29 0d 0a 09 09 09 09 09 64 70 32 5b  ='W').......dp2[
0480: 6d 2a 4d 2b 72 5d 20 2b 3d 20 64 70 5b 6d 72 5d  m*M+r] += dp[mr]
0490: 3b 0d 0a 09 09 09 7d 0d 0a 09 09 09 64 70 2e 73  ;.....}.....dp.s
04a0: 77 61 70 28 64 70 32 29 3b 0d 0a 09 09 7d 0d 0a  wap(dp2);....}..
04b0: 09 09 72 65 74 75 72 6e 20 64 70 5b 30 5d 2e 76  ..return dp[0].v
04c0: 61 6c 3b 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f  al;...}..};..../
04d0: 2f 20 42 45 47 49 4e 20 43 55 54 20 48 45 52 45  / BEGIN CUT HERE
04e0: 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 63 74 69 6d  ..#include <ctim
04f0: 65 3e 0d 0a 64 6f 75 62 6c 65 20 73 74 61 72 74  e>..double start
0500: 5f 74 69 6d 65 3b 20 73 74 72 69 6e 67 20 74 69  _time; string ti
0510: 6d 65 72 28 29 0d 0a 20 7b 20 6f 73 74 72 69 6e  mer().. { ostrin
0520: 67 73 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c  gstream os; os <
0530: 3c 20 22 20 28 22 20 3c 3c 20 69 6e 74 28 28 63  < " (" << int((c
0540: 6c 6f 63 6b 28 29 2d 73 74 61 72 74 5f 74 69 6d  lock()-start_tim
0550: 65 29 2f 43 4c 4f 43 4b 53 5f 50 45 52 5f 53 45  e)/CLOCKS_PER_SE
0560: 43 2a 31 30 30 30 29 20 3c 3c 20 22 20 6d 73 65  C*1000) << " mse
0570: 63 29 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73  c)"; return os.s
0580: 74 72 28 29 3b 20 7d 0d 0a 74 65 6d 70 6c 61 74  tr(); }..templat
0590: 65 3c 74 79 70 65 6e 61 6d 65 20 54 3e 20 6f 73  e<typename T> os
05a0: 74 72 65 61 6d 26 20 6f 70 65 72 61 74 6f 72 3c  tream& operator<
05b0: 3c 28 6f 73 74 72 65 61 6d 26 20 6f 73 2c 20 63  <(ostream& os, c
05c0: 6f 6e 73 74 20 76 65 63 74 6f 72 3c 54 3e 26 20  onst vector<T>& 
05d0: 76 29 0d 0a 20 7b 20 6f 73 20 3c 3c 20 22 7b 20  v).. { os << "{ 
05e0: 22 3b 0d 0a 20 20 20 66 6f 72 28 74 79 70 65 6e  ";..   for(typen
05f0: 61 6d 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a 63  ame vector<T>::c
0600: 6f 6e 73 74 5f 69 74 65 72 61 74 6f 72 20 69 74  onst_iterator it
0610: 3d 76 2e 62 65 67 69 6e 28 29 3b 20 69 74 21 3d  =v.begin(); it!=
0620: 76 2e 65 6e 64 28 29 3b 20 2b 2b 69 74 29 0d 0a  v.end(); ++it)..
0630: 20 20 20 6f 73 20 3c 3c 20 27 5c 22 27 20 3c 3c     os << '\"' <<
0640: 20 2a 69 74 20 3c 3c 20 27 5c 22 27 20 3c 3c 20   *it << '\"' << 
0650: 28 69 74 2b 31 3d 3d 76 2e 65 6e 64 28 29 20 3f  (it+1==v.end() ?
0660: 20 22 22 20 3a 20 22 2c 20 22 29 3b 20 6f 73 20   "" : ", "); os 
0670: 3c 3c 20 22 20 7d 22 3b 20 72 65 74 75 72 6e 20  << " }"; return 
0680: 6f 73 3b 20 7d 0d 0a 76 6f 69 64 20 76 65 72 69  os; }..void veri
0690: 66 79 5f 63 61 73 65 28 63 6f 6e 73 74 20 69 6e  fy_case(const in
06a0: 74 26 20 45 78 70 65 63 74 65 64 2c 20 63 6f 6e  t& Expected, con
06b0: 73 74 20 69 6e 74 26 20 52 65 63 65 69 76 65 64  st int& Received
06c0: 29 20 7b 0d 0a 20 62 6f 6f 6c 20 6f 6b 20 3d 20  ) {.. bool ok = 
06d0: 28 45 78 70 65 63 74 65 64 20 3d 3d 20 52 65 63  (Expected == Rec
06e0: 65 69 76 65 64 29 3b 0d 0a 20 69 66 28 6f 6b 29  eived);.. if(ok)
06f0: 20 63 65 72 72 20 3c 3c 20 22 50 41 53 53 45 44   cerr << "PASSED
0700: 22 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20  " << timer() << 
0710: 65 6e 64 6c 3b 20 20 65 6c 73 65 20 7b 20 63 65  endl;  else { ce
0720: 72 72 20 3c 3c 20 22 46 41 49 4c 45 44 22 20 3c  rr << "FAILED" <
0730: 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64  < timer() << end
0740: 6c 3b 0d 0a 20 63 65 72 72 20 3c 3c 20 22 5c 74  l;.. cerr << "\t
0750: 6f 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74  o: \"" << Expect
0760: 65 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e  ed << '\"' << en
0770: 64 6c 20 3c 3c 20 22 5c 74 78 3a 20 5c 22 22 20  dl << "\tx: \"" 
0780: 3c 3c 20 52 65 63 65 69 76 65 64 20 3c 3c 20 27  << Received << '
0790: 5c 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d  \"' << endl; } }
07a0: 0d 0a 23 64 65 66 69 6e 65 20 43 41 53 45 28 4e  ..#define CASE(N
07b0: 29 20 7b 63 65 72 72 20 3c 3c 20 22 54 65 73 74  ) {cerr << "Test
07c0: 20 43 61 73 65 20 23 22 20 3c 3c 20 4e 20 3c 3c   Case #" << N <<
07d0: 20 22 2e 2e 2e 22 20 3c 3c 20 66 6c 75 73 68 3b   "..." << flush;
07e0: 20 73 74 61 72 74 5f 74 69 6d 65 3d 63 6c 6f 63   start_time=cloc
07f0: 6b 28 29 3b 0d 0a 23 64 65 66 69 6e 65 20 45 4e  k();..#define EN
0800: 44 09 20 76 65 72 69 66 79 5f 63 61 73 65 28 5f  D. verify_case(_
0810: 2c 20 41 6c 74 65 72 6e 61 74 69 76 65 50 69 6c  , AlternativePil
0820: 65 73 28 29 2e 63 6f 75 6e 74 28 43 2c 20 4d 29  es().count(C, M)
0830: 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 7b  );}..int main(){
0840: 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 73 74  ....CASE(0)...st
0850: 72 69 6e 67 20 43 20 3d 20 22 57 52 47 57 57 52  ring C = "WRGWWR
0860: 47 57 22 3b 20 0d 0a 09 69 6e 74 20 4d 20 3d 20  GW"; ...int M = 
0870: 32 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 33 3b  2; ...int _ = 3;
0880: 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 31 29 0d   ..END..CASE(1).
0890: 0a 09 73 74 72 69 6e 67 20 43 20 3d 20 22 52 52  ..string C = "RR
08a0: 47 47 22 3b 20 0d 0a 09 69 6e 74 20 4d 20 3d 20  GG"; ...int M = 
08b0: 31 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 30 3b  1; ...int _ = 0;
08c0: 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 32 29 0d   ..END..CASE(2).
08d0: 0a 09 73 74 72 69 6e 67 20 43 20 3d 20 22 42 42  ..string C = "BB
08e0: 42 42 22 3b 20 0d 0a 09 69 6e 74 20 4d 20 3d 20  BB"; ...int M = 
08f0: 35 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 31 3b  5; ...int _ = 1;
0900: 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 33 29 0d   ..END..CASE(3).
0910: 0a 09 73 74 72 69 6e 67 20 43 20 3d 20 22 57 57  ..string C = "WW
0920: 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57  WWWWWWWWWWWWWWWW
0930: 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57  WWWWWWWWWWWWWWWW
0940: 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57  WWWWWWWWWWWWWWWW
0950: 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57  WWWWWWWWWWWWWWWW
0960: 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57  WWWWWWWWWWWWWWWW
0970: 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57 57  WWWWWWWWWWWWWWWW
0980: 57 57 22 3b 20 0d 0a 09 69 6e 74 20 4d 20 3d 20  WW"; ...int M = 
0990: 35 30 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 32  50; ...int _ = 2
09a0: 36 35 34 37 30 34 33 35 3b 20 0d 0a 45 4e 44 0d  65470435; ..END.
09b0: 0a 43 41 53 45 28 34 29 0d 0a 09 73 74 72 69 6e  .CASE(4)...strin
09c0: 67 20 43 20 3d 20 22 57 52 57 52 47 57 57 47 57  g C = "WRWRGWWGW
09d0: 57 57 52 57 42 57 52 57 47 57 57 52 57 42 57 57  WWRWBWRWGWWRWBWW
09e0: 52 47 57 42 57 47 52 47 57 57 47 57 47 52 57 47  RGWBWGRGWWGWGRWG
09f0: 52 57 42 52 57 42 57 22 3b 20 0d 0a 09 69 6e 74  RWBRWBW"; ...int
0a00: 20 4d 20 3d 20 37 3b 20 0d 0a 09 69 6e 74 20 5f   M = 7; ...int _
0a10: 20 3d 20 37 34 30 30 33 34 38 3b 20 0d 0a 45 4e   = 7400348; ..EN
0a20: 44 0d 0a 2f 2a 0d 0a 43 41 53 45 28 36 29 0d 0a  D../*..CASE(6)..
0a30: 09 73 74 72 69 6e 67 20 43 20 3d 20 3b 20 0d 0a  .string C = ; ..
0a40: 09 69 6e 74 20 4d 20 3d 20 3b 20 0d 0a 09 69 6e  .int M = ; ...in
0a50: 74 20 5f 20 3d 20 3b 20 0d 0a 45 4e 44 0d 0a 2a  t _ = ; ..END..*
0a60: 2f 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 20 43 55 54  /..}..// END CUT
0a70: 20 48 45 52 45 0d 0a                              HERE..