Hex Artifact Content
Not logged in

Artifact 9a8c214df95651f474733d47d7267fde5c40c5a0:


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 69 6e 74 20 4d 4f 44 56 41 4c 20 3d 20  st int MODVAL = 
01b0: 31 30 30 30 30 30 30 30 30 37 3b 0d 0a 63 6c 61  1000000007;..cla
01c0: 73 73 20 42 61 67 41 6e 64 43 61 72 64 73 20 7b  ss BagAndCards {
01d0: 20 70 75 62 6c 69 63 3a 0d 0a 09 69 6e 74 20 67   public:...int g
01e0: 65 74 48 61 73 68 28 69 6e 74 20 6e 2c 20 69 6e  etHash(int n, in
01f0: 74 20 6d 2c 20 69 6e 74 20 78 2c 20 69 6e 74 20  t m, int x, int 
0200: 61 2c 20 69 6e 74 20 62 2c 20 69 6e 74 20 63 2c  a, int b, int c,
0210: 20 73 74 72 69 6e 67 20 69 73 47 6f 6f 64 29 0d   string isGood).
0220: 0a 09 7b 0d 0a 09 09 76 65 63 74 6f 72 3c 76 65  ..{....vector<ve
0230: 63 74 6f 72 3c 69 6e 74 3e 3e 20 63 6f 75 6e 74  ctor<int>> count
0240: 28 6e 2c 20 76 65 63 74 6f 72 3c 69 6e 74 3e 28  (n, vector<int>(
0250: 6d 2c 20 30 29 29 3b 0d 0a 09 09 66 6f 72 28 69  m, 0));....for(i
0260: 6e 74 20 69 3d 30 3b 20 69 3c 6e 3b 20 2b 2b 69  nt i=0; i<n; ++i
0270: 29 0d 0a 09 09 66 6f 72 28 69 6e 74 20 6a 3d 30  )....for(int j=0
0280: 3b 20 6a 3c 6d 3b 20 2b 2b 6a 29 20 7b 0d 0a 09  ; j<m; ++j) {...
0290: 09 09 63 6f 75 6e 74 5b 69 5d 5b 6a 5d 20 3d 20  ..count[i][j] = 
02a0: 78 3b 0d 0a 09 09 09 78 20 3d 20 69 6e 74 28 28  x;.....x = int((
02b0: 28 4c 4c 28 78 29 20 2a 20 61 20 2b 20 62 29 20  (LL(x) * a + b) 
02c0: 5e 20 63 29 20 25 20 4d 4f 44 56 41 4c 29 3b 0d  ^ c) % MODVAL);.
02d0: 0a 09 09 7d 0d 0a 0d 0a 09 09 69 6e 74 20 74 6f  ...}......int to
02e0: 74 61 6c 5f 61 6e 73 20 3d 20 30 3b 0d 0a 09 09  tal_ans = 0;....
02f0: 66 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 3c 6e  for(int i=0; i<n
0300: 3b 20 2b 2b 69 29 0d 0a 09 09 7b 0d 0a 09 09 09  ; ++i)....{.....
0310: 76 65 63 74 6f 72 3c 4c 4c 3e 20 6d 75 6c 74 28  vector<LL> mult(
0320: 6d 29 3b 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20  m);.....for(int 
0330: 63 31 3d 30 3b 20 63 31 3c 6d 3b 20 2b 2b 63 31  c1=0; c1<m; ++c1
0340: 29 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 63 32  ).....for(int c2
0350: 3d 30 3b 20 63 32 3c 6d 3b 20 2b 2b 63 32 29 0d  =0; c2<m; ++c2).
0360: 0a 09 09 09 09 69 66 28 69 73 47 6f 6f 64 5b 63  .....if(isGood[c
0370: 31 2b 63 32 5d 3d 3d 27 59 27 29 0d 0a 09 09 09  1+c2]=='Y').....
0380: 09 09 6d 75 6c 74 5b 63 32 5d 20 2b 3d 20 63 6f  ..mult[c2] += co
0390: 75 6e 74 5b 69 5d 5b 63 31 5d 3b 0d 0a 09 09 09  unt[i][c1];.....
03a0: 66 6f 72 28 69 6e 74 20 63 3d 30 3b 20 63 3c 6d  for(int c=0; c<m
03b0: 3b 20 2b 2b 63 29 0d 0a 09 09 09 09 6d 75 6c 74  ; ++c)......mult
03c0: 5b 63 5d 20 25 3d 20 4d 4f 44 56 41 4c 3b 0d 0a  [c] %= MODVAL;..
03d0: 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 6b 3d 69  .....for(int k=i
03e0: 2b 31 3b 20 6b 3c 6e 3b 20 2b 2b 6b 29 0d 0a 09  +1; k<n; ++k)...
03f0: 09 09 7b 0d 0a 09 09 09 09 4c 4c 20 61 6e 73 20  ..{......LL ans 
0400: 3d 20 30 3b 0d 0a 09 09 09 09 66 6f 72 28 69 6e  = 0;......for(in
0410: 74 20 63 3d 30 3b 20 63 3c 6d 3b 20 2b 2b 63 29  t c=0; c<m; ++c)
0420: 0d 0a 09 09 09 09 09 61 6e 73 20 2b 3d 20 63 6f  .......ans += co
0430: 75 6e 74 5b 6b 5d 5b 63 5d 20 2a 20 6d 75 6c 74  unt[k][c] * mult
0440: 5b 63 5d 20 25 20 4d 4f 44 56 41 4c 3b 0d 0a 09  [c] % MODVAL;...
0450: 09 09 09 74 6f 74 61 6c 5f 61 6e 73 20 5e 3d 20  ...total_ans ^= 
0460: 69 6e 74 28 61 6e 73 20 25 20 4d 4f 44 56 41 4c  int(ans % MODVAL
0470: 29 3b 0d 0a 09 09 09 7d 0d 0a 09 09 7d 0d 0a 09  );.....}....}...
0480: 09 72 65 74 75 72 6e 20 74 6f 74 61 6c 5f 61 6e  .return total_an
0490: 73 3b 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f  s;...}..};....//
04a0: 20 42 45 47 49 4e 20 43 55 54 20 48 45 52 45 0d   BEGIN CUT HERE.
04b0: 0a 23 69 6e 63 6c 75 64 65 20 3c 63 74 69 6d 65  .#include <ctime
04c0: 3e 0d 0a 64 6f 75 62 6c 65 20 73 74 61 72 74 5f  >..double start_
04d0: 74 69 6d 65 3b 20 73 74 72 69 6e 67 20 74 69 6d  time; string tim
04e0: 65 72 28 29 0d 0a 20 7b 20 6f 73 74 72 69 6e 67  er().. { ostring
04f0: 73 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c  stream os; os <<
0500: 20 22 20 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c   " (" << int((cl
0510: 6f 63 6b 28 29 2d 73 74 61 72 74 5f 74 69 6d 65  ock()-start_time
0520: 29 2f 43 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43  )/CLOCKS_PER_SEC
0530: 2a 31 30 30 30 29 20 3c 3c 20 22 20 6d 73 65 63  *1000) << " msec
0540: 29 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74  )"; return os.st
0550: 72 28 29 3b 20 7d 0d 0a 74 65 6d 70 6c 61 74 65  r(); }..template
0560: 3c 74 79 70 65 6e 61 6d 65 20 54 3e 20 6f 73 74  <typename T> ost
0570: 72 65 61 6d 26 20 6f 70 65 72 61 74 6f 72 3c 3c  ream& operator<<
0580: 28 6f 73 74 72 65 61 6d 26 20 6f 73 2c 20 63 6f  (ostream& os, co
0590: 6e 73 74 20 76 65 63 74 6f 72 3c 54 3e 26 20 76  nst vector<T>& v
05a0: 29 0d 0a 20 7b 20 6f 73 20 3c 3c 20 22 7b 20 22  ).. { os << "{ "
05b0: 3b 0d 0a 20 20 20 66 6f 72 28 74 79 70 65 6e 61  ;..   for(typena
05c0: 6d 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f  me vector<T>::co
05d0: 6e 73 74 5f 69 74 65 72 61 74 6f 72 20 69 74 3d  nst_iterator it=
05e0: 76 2e 62 65 67 69 6e 28 29 3b 20 69 74 21 3d 76  v.begin(); it!=v
05f0: 2e 65 6e 64 28 29 3b 20 2b 2b 69 74 29 0d 0a 20  .end(); ++it).. 
0600: 20 20 6f 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20    os << '\"' << 
0610: 2a 69 74 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 28  *it << '\"' << (
0620: 69 74 2b 31 3d 3d 76 2e 65 6e 64 28 29 20 3f 20  it+1==v.end() ? 
0630: 22 22 20 3a 20 22 2c 20 22 29 3b 20 6f 73 20 3c  "" : ", "); os <
0640: 3c 20 22 20 7d 22 3b 20 72 65 74 75 72 6e 20 6f  < " }"; return o
0650: 73 3b 20 7d 0d 0a 76 6f 69 64 20 76 65 72 69 66  s; }..void verif
0660: 79 5f 63 61 73 65 28 63 6f 6e 73 74 20 69 6e 74  y_case(const int
0670: 26 20 45 78 70 65 63 74 65 64 2c 20 63 6f 6e 73  & Expected, cons
0680: 74 20 69 6e 74 26 20 52 65 63 65 69 76 65 64 29  t int& Received)
0690: 20 7b 0d 0a 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28   {.. bool ok = (
06a0: 45 78 70 65 63 74 65 64 20 3d 3d 20 52 65 63 65  Expected == Rece
06b0: 69 76 65 64 29 3b 0d 0a 20 69 66 28 6f 6b 29 20  ived);.. if(ok) 
06c0: 63 65 72 72 20 3c 3c 20 22 50 41 53 53 45 44 22  cerr << "PASSED"
06d0: 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 65   << timer() << e
06e0: 6e 64 6c 3b 20 20 65 6c 73 65 20 7b 20 63 65 72  ndl;  else { cer
06f0: 72 20 3c 3c 20 22 46 41 49 4c 45 44 22 20 3c 3c  r << "FAILED" <<
0700: 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c   timer() << endl
0710: 3b 0d 0a 20 63 65 72 72 20 3c 3c 20 22 5c 74 6f  ;.. cerr << "\to
0720: 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 65  : \"" << Expecte
0730: 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64  d << '\"' << end
0740: 6c 20 3c 3c 20 22 5c 74 78 3a 20 5c 22 22 20 3c  l << "\tx: \"" <
0750: 3c 20 52 65 63 65 69 76 65 64 20 3c 3c 20 27 5c  < Received << '\
0760: 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d  "' << endl; } }.
0770: 0a 23 64 65 66 69 6e 65 20 43 41 53 45 28 4e 29  .#define CASE(N)
0780: 20 7b 63 65 72 72 20 3c 3c 20 22 54 65 73 74 20   {cerr << "Test 
0790: 43 61 73 65 20 23 22 20 3c 3c 20 4e 20 3c 3c 20  Case #" << N << 
07a0: 22 2e 2e 2e 22 20 3c 3c 20 66 6c 75 73 68 3b 20  "..." << flush; 
07b0: 73 74 61 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b  start_time=clock
07c0: 28 29 3b 0d 0a 23 64 65 66 69 6e 65 20 45 4e 44  ();..#define END
07d0: 09 20 76 65 72 69 66 79 5f 63 61 73 65 28 5f 2c  . verify_case(_,
07e0: 20 42 61 67 41 6e 64 43 61 72 64 73 28 29 2e 67   BagAndCards().g
07f0: 65 74 48 61 73 68 28 6e 2c 20 6d 2c 20 78 2c 20  etHash(n, m, x, 
0800: 61 2c 20 62 2c 20 63 2c 20 69 73 47 6f 6f 64 29  a, b, c, isGood)
0810: 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 7b  );}..int main(){
0820: 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 69 6e  ....CASE(0)...in
0830: 74 20 6e 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20  t n = 2; ...int 
0840: 6d 20 3d 20 34 3b 20 0d 0a 09 69 6e 74 20 78 20  m = 4; ...int x 
0850: 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 61 20 3d 20  = 1; ...int a = 
0860: 31 3b 20 0d 0a 09 69 6e 74 20 62 20 3d 20 30 3b  1; ...int b = 0;
0870: 20 0d 0a 09 69 6e 74 20 63 20 3d 20 30 3b 20 0d   ...int c = 0; .
0880: 0a 09 73 74 72 69 6e 67 20 69 73 47 6f 6f 64 20  ..string isGood 
0890: 3d 20 22 4e 4e 59 59 4e 59 4e 22 3b 20 0d 0a 09  = "NNYYNYN"; ...
08a0: 69 6e 74 20 5f 20 3d 20 39 3b 20 0d 0a 45 4e 44  int _ = 9; ..END
08b0: 0d 0a 43 41 53 45 28 31 29 0d 0a 09 69 6e 74 20  ..CASE(1)...int 
08c0: 6e 20 3d 20 33 3b 20 0d 0a 09 69 6e 74 20 6d 20  n = 3; ...int m 
08d0: 3d 20 35 3b 20 0d 0a 09 69 6e 74 20 78 20 3d 20  = 5; ...int x = 
08e0: 31 3b 20 0d 0a 09 69 6e 74 20 61 20 3d 20 31 3b  1; ...int a = 1;
08f0: 20 0d 0a 09 69 6e 74 20 62 20 3d 20 31 3b 20 0d   ...int b = 1; .
0900: 0a 09 69 6e 74 20 63 20 3d 20 32 3b 20 0d 0a 09  ..int c = 2; ...
0910: 73 74 72 69 6e 67 20 69 73 47 6f 6f 64 20 3d 20  string isGood = 
0920: 22 4e 4e 59 59 4e 59 4e 59 4e 22 3b 20 0d 0a 09  "NNYYNYNYN"; ...
0930: 69 6e 74 20 5f 20 3d 20 31 35 33 32 3b 20 0d 0a  int _ = 1532; ..
0940: 45 4e 44 0d 0a 43 41 53 45 28 32 29 0d 0a 09 69  END..CASE(2)...i
0950: 6e 74 20 6e 20 3d 20 31 30 3b 20 0d 0a 09 69 6e  nt n = 10; ...in
0960: 74 20 6d 20 3d 20 32 30 3b 20 0d 0a 09 69 6e 74  t m = 20; ...int
0970: 20 78 20 3d 20 31 31 31 3b 20 0d 0a 09 69 6e 74   x = 111; ...int
0980: 20 61 20 3d 20 32 32 32 3b 20 0d 0a 09 69 6e 74   a = 222; ...int
0990: 20 62 20 3d 20 33 33 33 3b 20 0d 0a 09 69 6e 74   b = 333; ...int
09a0: 20 63 20 3d 20 34 34 34 3b 20 0d 0a 09 73 74 72   c = 444; ...str
09b0: 69 6e 67 20 69 73 47 6f 6f 64 20 3d 20 22 4e 4e  ing isGood = "NN
09c0: 4e 4e 4e 59 59 59 4e 4e 4e 59 59 59 59 59 59 4e  NNNYYYNNNYYYYYYN
09d0: 4e 59 59 59 59 4e 4e 4e 4e 59 4e 4e 59 59 59 4e  NYYYYNNNNYNNYYYN
09e0: 4e 4e 59 59 4e 22 20 3b 20 0d 0a 09 69 6e 74 20  NNYYN" ; ...int 
09f0: 5f 20 3d 20 34 35 30 37 35 30 36 38 33 3b 20 0d  _ = 450750683; .
0a00: 0a 45 4e 44 0d 0a 43 41 53 45 28 33 29 0d 0a 09  .END..CASE(3)...
0a10: 69 6e 74 20 6e 20 3d 20 32 3b 20 0d 0a 09 69 6e  int n = 2; ...in
0a20: 74 20 6d 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20  t m = 2; ...int 
0a30: 78 20 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 61 20  x = 1; ...int a 
0a40: 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 62 20 3d 20  = 1; ...int b = 
0a50: 30 3b 20 0d 0a 09 69 6e 74 20 63 20 3d 20 30 3b  0; ...int c = 0;
0a60: 20 0d 0a 09 73 74 72 69 6e 67 20 69 73 47 6f 6f   ...string isGoo
0a70: 64 20 3d 20 22 4e 4e 59 22 3b 20 0d 0a 09 69 6e  d = "NNY"; ...in
0a80: 74 20 5f 20 3d 20 31 3b 20 0d 0a 45 4e 44 0d 0a  t _ = 1; ..END..
0a90: 2f 2a 0d 0a 43 41 53 45 28 34 29 0d 0a 09 69 6e  /*..CASE(4)...in
0aa0: 74 20 6e 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 6d  t n = ; ...int m
0ab0: 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 78 20 3d 20   = ; ...int x = 
0ac0: 3b 20 0d 0a 09 69 6e 74 20 61 20 3d 20 3b 20 0d  ; ...int a = ; .
0ad0: 0a 09 69 6e 74 20 62 20 3d 20 3b 20 0d 0a 09 69  ..int b = ; ...i
0ae0: 6e 74 20 63 20 3d 20 3b 20 0d 0a 09 73 74 72 69  nt c = ; ...stri
0af0: 6e 67 20 69 73 47 6f 6f 64 20 3d 20 3b 20 0d 0a  ng isGood = ; ..
0b00: 09 69 6e 74 20 5f 20 3d 20 3b 20 0d 0a 45 4e 44  .int _ = ; ..END
0b10: 0d 0a 43 41 53 45 28 35 29 0d 0a 09 69 6e 74 20  ..CASE(5)...int 
0b20: 6e 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 6d 20 3d  n = ; ...int m =
0b30: 20 3b 20 0d 0a 09 69 6e 74 20 78 20 3d 20 3b 20   ; ...int x = ; 
0b40: 0d 0a 09 69 6e 74 20 61 20 3d 20 3b 20 0d 0a 09  ...int a = ; ...
0b50: 69 6e 74 20 62 20 3d 20 3b 20 0d 0a 09 69 6e 74  int b = ; ...int
0b60: 20 63 20 3d 20 3b 20 0d 0a 09 73 74 72 69 6e 67   c = ; ...string
0b70: 20 69 73 47 6f 6f 64 20 3d 20 3b 20 0d 0a 09 69   isGood = ; ...i
0b80: 6e 74 20 5f 20 3d 20 3b 20 0d 0a 45 4e 44 0d 0a  nt _ = ; ..END..
0b90: 2a 2f 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 20 43 55  */..}..// END CU
0ba0: 54 20 48 45 52 45 0d 0a                          T HERE..