Hex Artifact Content
Not logged in

Artifact e110c16d9a9bffd53f881091cd58d56f4e3f1779:


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 63 6f 6d 70  ..#include <comp
00d0: 6c 65 78 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c  lex>..#include <
00e0: 71 75 65 75 65 3e 0d 0a 23 69 6e 63 6c 75 64 65  queue>..#include
00f0: 20 3c 73 74 61 63 6b 3e 0d 0a 23 69 6e 63 6c 75   <stack>..#inclu
0100: 64 65 20 3c 63 6d 61 74 68 3e 0d 0a 75 73 69 6e  de <cmath>..usin
0110: 67 20 6e 61 6d 65 73 70 61 63 65 20 73 74 64 3b  g namespace std;
0120: 0d 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c  ..typedef long l
0130: 6f 6e 67 20 4c 4c 3b 0d 0a 0d 0a 73 74 61 74 69  ong LL;....stati
0140: 63 20 63 6f 6e 73 74 20 4c 4c 20 4d 4f 44 56 41  c const LL MODVA
0150: 4c 20 3d 20 31 32 33 34 35 36 37 38 39 31 3b 0d  L = 1234567891;.
0160: 0a 0d 0a 4c 4c 20 41 44 44 28 4c 4c 20 78 2c 20  ...LL ADD(LL x, 
0170: 4c 4c 20 79 29 20 7b 20 72 65 74 75 72 6e 20 28  LL y) { return (
0180: 78 2b 79 29 25 4d 4f 44 56 41 4c 3b 20 7d 0d 0a  x+y)%MODVAL; }..
0190: 4c 4c 20 53 55 42 28 4c 4c 20 78 2c 20 4c 4c 20  LL SUB(LL x, LL 
01a0: 79 29 20 7b 20 72 65 74 75 72 6e 20 28 78 2d 79  y) { return (x-y
01b0: 2b 4d 4f 44 56 41 4c 29 25 4d 4f 44 56 41 4c 3b  +MODVAL)%MODVAL;
01c0: 20 7d 0d 0a 4c 4c 20 4d 55 4c 28 4c 4c 20 78 2c   }..LL MUL(LL x,
01d0: 20 4c 4c 20 79 29 20 7b 20 72 65 74 75 72 6e 20   LL y) { return 
01e0: 28 78 2a 79 29 25 4d 4f 44 56 41 4c 3b 20 7d 0d  (x*y)%MODVAL; }.
01f0: 0a 4c 4c 20 50 4f 57 28 4c 4c 20 78 2c 20 4c 4c  .LL POW(LL x, LL
0200: 20 65 29 20 7b 0d 0a 09 4c 4c 20 76 20 3d 20 31   e) {...LL v = 1
0210: 3b 0d 0a 09 66 6f 72 28 3b 65 3b 78 3d 4d 55 4c  ;...for(;e;x=MUL
0220: 28 78 2c 78 29 2c 65 3e 3e 3d 31 29 0d 0a 09 09  (x,x),e>>=1)....
0230: 69 66 28 65 26 31 29 0d 0a 09 09 09 76 20 3d 20  if(e&1).....v = 
0240: 4d 55 4c 28 76 2c 20 78 29 3b 0d 0a 09 72 65 74  MUL(v, x);...ret
0250: 75 72 6e 20 76 3b 0d 0a 7d 0d 0a 4c 4c 20 44 49  urn v;..}..LL DI
0260: 56 28 4c 4c 20 78 2c 20 4c 4c 20 79 29 20 7b 20  V(LL x, LL y) { 
0270: 72 65 74 75 72 6e 20 4d 55 4c 28 78 2c 20 50 4f  return MUL(x, PO
0280: 57 28 79 2c 20 4d 4f 44 56 41 4c 2d 32 29 29 3b  W(y, MODVAL-2));
0290: 20 7d 0d 0a 4c 4c 20 43 28 4c 4c 20 6e 2c 20 4c   }..LL C(LL n, L
02a0: 4c 20 6b 29 20 7b 0d 0a 09 4c 4c 20 76 20 3d 20  L k) {...LL v = 
02b0: 31 3b 0d 0a 09 66 6f 72 28 4c 4c 20 69 3d 31 3b  1;...for(LL i=1;
02c0: 20 69 3c 3d 6b 3b 20 2b 2b 69 29 0d 0a 09 09 76   i<=k; ++i)....v
02d0: 20 3d 20 44 49 56 28 4d 55 4c 28 76 2c 20 6e 2d   = DIV(MUL(v, n-
02e0: 69 2b 31 29 2c 20 69 29 3b 0d 0a 09 72 65 74 75  i+1), i);...retu
02f0: 72 6e 20 76 3b 0d 0a 7d 0d 0a 4c 4c 20 47 53 53  rn v;..}..LL GSS
0300: 28 4c 4c 20 6b 2c 20 4c 4c 20 62 2c 20 4c 4c 20  (LL k, LL b, LL 
0310: 65 29 20 2f 2f 20 6b 5e 62 20 2b 20 6b 5e 62 2b  e) // k^b + k^b+
0320: 31 20 2b 20 2e 2e 2e 20 2b 20 6b 5e 65 0d 0a 7b  1 + ... + k^e..{
0330: 0d 0a 09 69 66 28 20 62 20 3e 20 20 65 20 29 20  ...if( b >  e ) 
0340: 72 65 74 75 72 6e 20 30 3b 0d 0a 09 69 66 28 20  return 0;...if( 
0350: 6b 20 3c 3d 20 31 20 29 20 72 65 74 75 72 6e 20  k <= 1 ) return 
0360: 6b 2a 28 65 2d 62 2b 31 29 3b 0d 0a 09 72 65 74  k*(e-b+1);...ret
0370: 75 72 6e 20 44 49 56 28 53 55 42 28 50 4f 57 28  urn DIV(SUB(POW(
0380: 6b 2c 20 65 2b 31 29 2c 20 50 4f 57 28 6b 2c 62  k, e+1), POW(k,b
0390: 29 29 2c 20 6b 2d 31 29 3b 0d 0a 7d 0d 0a 0d 0a  )), k-1);..}....
03a0: 63 6c 61 73 73 20 54 68 65 4c 6f 6e 67 50 61 6c  class TheLongPal
03b0: 69 6e 64 72 6f 6d 65 0d 0a 7b 0d 0a 70 75 62 6c  indrome..{..publ
03c0: 69 63 3a 0d 0a 09 69 6e 74 20 63 6f 75 6e 74 28  ic:...int count(
03d0: 69 6e 74 20 6e 2c 20 69 6e 74 20 6b 29 20 0d 0a  int n, int k) ..
03e0: 09 7b 0d 0a 09 09 4c 4c 20 61 6e 73 20 3d 20 30  .{....LL ans = 0
03f0: 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 69 3d 31  ;....for(int i=1
0400: 3b 20 69 3c 3d 6b 3b 20 2b 2b 69 29 0d 0a 09 09  ; i<=k; ++i)....
0410: 7b 0d 0a 09 09 09 4c 4c 20 61 61 20 3d 20 66 28  {.....LL aa = f(
0420: 69 2c 69 2c 6e 29 3b 0d 0a 09 09 09 66 6f 72 28  i,i,n);.....for(
0430: 69 6e 74 20 6a 3d 69 2d 31 3b 20 6a 3e 3d 31 3b  int j=i-1; j>=1;
0440: 20 2d 2d 6a 29 0d 0a 09 09 09 09 61 61 20 3d 20   --j)......aa = 
0450: 28 69 2d 6a 29 25 32 0d 0a 09 09 09 09 20 20 3f  (i-j)%2......  ?
0460: 20 53 55 42 28 61 61 2c 20 4d 55 4c 28 66 28 6a   SUB(aa, MUL(f(j
0470: 2c 69 2c 6e 29 2c 20 43 28 69 2c 6a 29 29 29 0d  ,i,n), C(i,j))).
0480: 0a 09 09 09 09 20 20 3a 20 41 44 44 28 61 61 2c  .....  : ADD(aa,
0490: 20 4d 55 4c 28 66 28 6a 2c 69 2c 6e 29 2c 20 43   MUL(f(j,i,n), C
04a0: 28 69 2c 6a 29 29 29 3b 0d 0a 09 09 09 61 6e 73  (i,j)));.....ans
04b0: 20 3d 20 41 44 44 28 61 6e 73 2c 20 4d 55 4c 28   = ADD(ans, MUL(
04c0: 61 61 2c 20 43 28 32 36 2c 69 29 29 29 3b 0d 0a  aa, C(26,i)));..
04d0: 09 09 7d 0d 0a 09 09 72 65 74 75 72 6e 20 61 6e  ..}....return an
04e0: 73 3b 0d 0a 09 7d 0d 0a 0d 0a 09 4c 4c 20 66 28  s;...}.....LL f(
04f0: 4c 4c 20 69 2c 20 4c 4c 20 62 2c 20 4c 4c 20 6e  LL i, LL b, LL n
0500: 29 20 2f 2f 20 69 5e 62 20 2b 20 69 5e 62 20 2b  ) // i^b + i^b +
0510: 20 69 5e 62 2b 31 20 2b 20 69 5e 62 2b 31 20 2b   i^b+1 + i^b+1 +
0520: 20 2e 2e 2e 20 2b 20 28 6e 74 68 29 0d 0a 09 7b   ... + (nth)...{
0530: 0d 0a 09 09 4c 4c 20 76 20 3d 20 4d 55 4c 28 47  ....LL v = MUL(G
0540: 53 53 28 69 2c 20 62 2c 20 6e 2f 32 29 2c 20 32  SS(i, b, n/2), 2
0550: 29 3b 0d 0a 09 09 69 66 28 20 6e 25 32 20 3d 3d  );....if( n%2 ==
0560: 20 31 20 29 0d 0a 09 09 09 76 20 3d 20 41 44 44   1 ).....v = ADD
0570: 28 76 2c 20 50 4f 57 28 69 2c 20 6e 2f 32 2b 31  (v, POW(i, n/2+1
0580: 29 29 3b 0d 0a 09 09 72 65 74 75 72 6e 20 76 3b  ));....return v;
0590: 0d 0a 09 7d 0d 0a 0d 0a 2f 2f 20 42 45 47 49 4e  ...}....// BEGIN
05a0: 20 43 55 54 20 48 45 52 45 0d 0a 09 70 75 62 6c   CUT HERE...publ
05b0: 69 63 3a 0d 0a 09 76 6f 69 64 20 72 75 6e 5f 74  ic:...void run_t
05c0: 65 73 74 28 69 6e 74 20 43 61 73 65 29 20 7b 20  est(int Case) { 
05d0: 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 29  if ((Case == -1)
05e0: 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 30 29 29   || (Case == 0))
05f0: 20 74 65 73 74 5f 63 61 73 65 5f 30 28 29 3b 20   test_case_0(); 
0600: 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 29  if ((Case == -1)
0610: 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 31 29 29   || (Case == 1))
0620: 20 74 65 73 74 5f 63 61 73 65 5f 31 28 29 3b 20   test_case_1(); 
0630: 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 29  if ((Case == -1)
0640: 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 32 29 29   || (Case == 2))
0650: 20 74 65 73 74 5f 63 61 73 65 5f 32 28 29 3b 20   test_case_2(); 
0660: 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 29  if ((Case == -1)
0670: 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 33 29 29   || (Case == 3))
0680: 20 74 65 73 74 5f 63 61 73 65 5f 33 28 29 3b 20   test_case_3(); 
0690: 7d 0d 0a 09 70 72 69 76 61 74 65 3a 0d 0a 09 74  }...private:...t
06a0: 65 6d 70 6c 61 74 65 20 3c 74 79 70 65 6e 61 6d  emplate <typenam
06b0: 65 20 54 3e 20 73 74 72 69 6e 67 20 70 72 69 6e  e T> string prin
06c0: 74 5f 61 72 72 61 79 28 63 6f 6e 73 74 20 76 65  t_array(const ve
06d0: 63 74 6f 72 3c 54 3e 20 26 56 29 20 7b 20 6f 73  ctor<T> &V) { os
06e0: 74 72 69 6e 67 73 74 72 65 61 6d 20 6f 73 3b 20  tringstream os; 
06f0: 6f 73 20 3c 3c 20 22 7b 20 22 3b 20 66 6f 72 20  os << "{ "; for 
0700: 28 74 79 70 65 6e 61 6d 65 20 76 65 63 74 6f 72  (typename vector
0710: 3c 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 61  <T>::const_itera
0720: 74 6f 72 20 69 74 65 72 20 3d 20 56 2e 62 65 67  tor iter = V.beg
0730: 69 6e 28 29 3b 20 69 74 65 72 20 21 3d 20 56 2e  in(); iter != V.
0740: 65 6e 64 28 29 3b 20 2b 2b 69 74 65 72 29 20 6f  end(); ++iter) o
0750: 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 74  s << '\"' << *it
0760: 65 72 20 3c 3c 20 22 5c 22 2c 22 3b 20 6f 73 20  er << "\","; os 
0770: 3c 3c 20 22 20 7d 22 3b 20 72 65 74 75 72 6e 20  << " }"; return 
0780: 6f 73 2e 73 74 72 28 29 3b 20 7d 0d 0a 09 76 6f  os.str(); }...vo
0790: 69 64 20 76 65 72 69 66 79 5f 63 61 73 65 28 69  id verify_case(i
07a0: 6e 74 20 43 61 73 65 2c 20 63 6f 6e 73 74 20 69  nt Case, const i
07b0: 6e 74 20 26 45 78 70 65 63 74 65 64 2c 20 63 6f  nt &Expected, co
07c0: 6e 73 74 20 69 6e 74 20 26 52 65 63 65 69 76 65  nst int &Receive
07d0: 64 29 20 7b 20 63 65 72 72 20 3c 3c 20 22 54 65  d) { cerr << "Te
07e0: 73 74 20 43 61 73 65 20 23 22 20 3c 3c 20 43 61  st Case #" << Ca
07f0: 73 65 20 3c 3c 20 22 2e 2e 2e 22 3b 20 69 66 20  se << "..."; if 
0800: 28 45 78 70 65 63 74 65 64 20 3d 3d 20 52 65 63  (Expected == Rec
0810: 65 69 76 65 64 29 20 63 65 72 72 20 3c 3c 20 22  eived) cerr << "
0820: 50 41 53 53 45 44 22 20 3c 3c 20 65 6e 64 6c 3b  PASSED" << endl;
0830: 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c 3c 20   else { cerr << 
0840: 22 46 41 49 4c 45 44 22 20 3c 3c 20 65 6e 64 6c  "FAILED" << endl
0850: 3b 20 63 65 72 72 20 3c 3c 20 22 5c 74 45 78 70  ; cerr << "\tExp
0860: 65 63 74 65 64 3a 20 5c 22 22 20 3c 3c 20 45 78  ected: \"" << Ex
0870: 70 65 63 74 65 64 20 3c 3c 20 27 5c 22 27 20 3c  pected << '\"' <
0880: 3c 20 65 6e 64 6c 3b 20 63 65 72 72 20 3c 3c 20  < endl; cerr << 
0890: 22 5c 74 52 65 63 65 69 76 65 64 3a 20 5c 22 22  "\tReceived: \""
08a0: 20 3c 3c 20 52 65 63 65 69 76 65 64 20 3c 3c 20   << Received << 
08b0: 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20  '\"' << endl; } 
08c0: 7d 0d 0a 09 76 6f 69 64 20 74 65 73 74 5f 63 61  }...void test_ca
08d0: 73 65 5f 30 28 29 20 7b 20 69 6e 74 20 41 72 67  se_0() { int Arg
08e0: 30 20 3d 20 31 3b 20 69 6e 74 20 41 72 67 31 20  0 = 1; int Arg1 
08f0: 3d 20 31 3b 20 69 6e 74 20 41 72 67 32 20 3d 20  = 1; int Arg2 = 
0900: 32 36 3b 20 76 65 72 69 66 79 5f 63 61 73 65 28  26; verify_case(
0910: 30 2c 20 41 72 67 32 2c 20 63 6f 75 6e 74 28 41  0, Arg2, count(A
0920: 72 67 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a  rg0, Arg1)); }..
0930: 09 76 6f 69 64 20 74 65 73 74 5f 63 61 73 65 5f  .void test_case_
0940: 31 28 29 20 7b 20 69 6e 74 20 41 72 67 30 20 3d  1() { int Arg0 =
0950: 20 32 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 31   2; int Arg1 = 1
0960: 30 3b 20 69 6e 74 20 41 72 67 32 20 3d 20 35 32  0; int Arg2 = 52
0970: 3b 20 76 65 72 69 66 79 5f 63 61 73 65 28 31 2c  ; verify_case(1,
0980: 20 41 72 67 32 2c 20 63 6f 75 6e 74 28 41 72 67   Arg2, count(Arg
0990: 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76  0, Arg1)); }...v
09a0: 6f 69 64 20 74 65 73 74 5f 63 61 73 65 5f 32 28  oid test_case_2(
09b0: 29 20 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 33  ) { int Arg0 = 3
09c0: 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 32 3b 20  ; int Arg1 = 2; 
09d0: 69 6e 74 20 41 72 67 32 20 3d 20 37 32 38 3b 20  int Arg2 = 728; 
09e0: 76 65 72 69 66 79 5f 63 61 73 65 28 32 2c 20 41  verify_case(2, A
09f0: 72 67 32 2c 20 63 6f 75 6e 74 28 41 72 67 30 2c  rg2, count(Arg0,
0a00: 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76 6f 69   Arg1)); }...voi
0a10: 64 20 74 65 73 74 5f 63 61 73 65 5f 33 28 29 20  d test_case_3() 
0a20: 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 34 34 3b  { int Arg0 = 44;
0a30: 20 69 6e 74 20 41 72 67 31 20 3d 20 37 3b 20 69   int Arg1 = 7; i
0a40: 6e 74 20 41 72 67 32 20 3d 20 32 34 30 32 34 39  nt Arg2 = 240249
0a50: 37 38 31 3b 20 76 65 72 69 66 79 5f 63 61 73 65  781; verify_case
0a60: 28 33 2c 20 41 72 67 32 2c 20 63 6f 75 6e 74 28  (3, Arg2, count(
0a70: 41 72 67 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d  Arg0, Arg1)); }.
0a80: 0a 0d 0a 2f 2f 20 45 4e 44 20 43 55 54 20 48 45  ...// END CUT HE
0a90: 52 45 0d 0a 7d 3b 0d 0a 2f 2f 20 42 45 47 49 4e  RE..};..// BEGIN
0aa0: 20 43 55 54 20 48 45 52 45 20 0d 0a 69 6e 74 20   CUT HERE ..int 
0ab0: 6d 61 69 6e 28 29 20 7b 20 54 68 65 4c 6f 6e 67  main() { TheLong
0ac0: 50 61 6c 69 6e 64 72 6f 6d 65 28 29 2e 72 75 6e  Palindrome().run
0ad0: 5f 74 65 73 74 28 2d 31 29 3b 20 7d 0d 0a 2f 2f  _test(-1); }..//
0ae0: 20 45 4e 44 20 43 55 54 20 48 45 52 45 0d 0a      END CUT HERE..