Hex Artifact Content
Not logged in

Artifact 3f49b98f321c45030d669f5b1a5b186b61d78372:


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 20 3a 76 61 6c 28 30  ...mint() :val(0
01f0: 29 20 7b 7d 0d 0a 09 6d 69 6e 74 28 69 6e 74 20  ) {}...mint(int 
0200: 20 20 20 20 20 78 29 20 3a 76 61 6c 28 78 25 4d       x) :val(x%M
0210: 4f 44 56 41 4c 29 20 7b 7d 0d 0a 09 6d 69 6e 74  ODVAL) {}...mint
0220: 28 75 6e 73 69 67 6e 65 64 20 78 29 20 3a 76 61  (unsigned x) :va
0230: 6c 28 78 25 4d 4f 44 56 41 4c 29 20 7b 7d 0d 0a  l(x%MODVAL) {}..
0240: 09 6d 69 6e 74 28 4c 4c 20 20 20 20 20 20 20 78  .mint(LL       x
0250: 29 20 3a 76 61 6c 28 78 25 4d 4f 44 56 41 4c 29  ) :val(x%MODVAL)
0260: 20 7b 7d 0d 0a 7d 3b 0d 0a 6d 69 6e 74 26 20 6f   {}..};..mint& o
0270: 70 65 72 61 74 6f 72 2b 3d 28 6d 69 6e 74 26 20  perator+=(mint& 
0280: 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 72 65 74  x, mint y) { ret
0290: 75 72 6e 20 78 20 3d 20 78 2e 76 61 6c 20 2b 20  urn x = x.val + 
02a0: 79 2e 76 61 6c 3b 20 7d 0d 0a 6d 69 6e 74 26 20  y.val; }..mint& 
02b0: 6f 70 65 72 61 74 6f 72 2d 3d 28 6d 69 6e 74 26  operator-=(mint&
02c0: 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 72 65   x, mint y) { re
02d0: 74 75 72 6e 20 78 20 3d 20 78 2e 76 61 6c 20 2d  turn x = x.val -
02e0: 20 79 2e 76 61 6c 20 2b 20 4d 4f 44 56 41 4c 3b   y.val + MODVAL;
02f0: 20 7d 0d 0a 6d 69 6e 74 26 20 6f 70 65 72 61 74   }..mint& operat
0300: 6f 72 2a 3d 28 6d 69 6e 74 26 20 78 2c 20 6d 69  or*=(mint& x, mi
0310: 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78  nt y) { return x
0320: 20 3d 20 4c 4c 28 78 2e 76 61 6c 29 2a 79 2e 76   = LL(x.val)*y.v
0330: 61 6c 3b 20 7d 0d 0a 6d 69 6e 74 20 6f 70 65 72  al; }..mint oper
0340: 61 74 6f 72 2b 28 6d 69 6e 74 20 78 2c 20 6d 69  ator+(mint x, mi
0350: 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78  nt y) { return x
0360: 20 2b 3d 20 79 3b 20 7d 0d 0a 6d 69 6e 74 20 6f   += y; }..mint o
0370: 70 65 72 61 74 6f 72 2d 28 6d 69 6e 74 20 78 2c  perator-(mint x,
0380: 20 6d 69 6e 74 20 79 29 20 7b 20 72 65 74 75 72   mint y) { retur
0390: 6e 20 78 20 2d 3d 20 79 3b 20 7d 0d 0a 6d 69 6e  n x -= y; }..min
03a0: 74 20 6f 70 65 72 61 74 6f 72 2a 28 6d 69 6e 74  t operator*(mint
03b0: 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 72 65   x, mint y) { re
03c0: 74 75 72 6e 20 78 20 2a 3d 20 79 3b 20 7d 0d 0a  turn x *= y; }..
03d0: 0d 0a 6d 69 6e 74 20 50 4f 57 28 6d 69 6e 74 20  ..mint POW(mint 
03e0: 78 2c 20 4c 4c 20 65 29 20 7b 20 6d 69 6e 74 20  x, LL e) { mint 
03f0: 76 20 3d 20 31 3b 20 66 6f 72 20 28 3b 20 65 3b  v = 1; for (; e;
0400: 20 78 20 2a 3d 20 78 2c 20 65 20 3e 3e 3d 20 31   x *= x, e >>= 1
0410: 29 20 69 66 20 28 65 20 26 20 31 29 20 76 20 2a  ) if (e & 1) v *
0420: 3d 20 78 3b 20 72 65 74 75 72 6e 20 76 3b 20 7d  = x; return v; }
0430: 0d 0a 6d 69 6e 74 26 20 6f 70 65 72 61 74 6f 72  ..mint& operator
0440: 2f 3d 28 6d 69 6e 74 26 20 78 2c 20 6d 69 6e 74  /=(mint& x, mint
0450: 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78 20 2a   y) { return x *
0460: 3d 20 50 4f 57 28 79 2c 20 4d 4f 44 56 41 4c 20  = POW(y, MODVAL 
0470: 2d 20 32 29 3b 20 7d 0d 0a 6d 69 6e 74 20 6f 70  - 2); }..mint op
0480: 65 72 61 74 6f 72 2f 28 6d 69 6e 74 20 78 2c 20  erator/(mint x, 
0490: 6d 69 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e  mint y) { return
04a0: 20 78 20 2f 3d 20 79 3b 20 7d 0d 0a 0d 0a 4c 4c   x /= y; }....LL
04b0: 20 65 75 6c 65 72 50 68 69 28 4c 4c 20 6e 29 20   eulerPhi(LL n) 
04c0: 7b 0d 0a 09 4c 4c 20 61 6e 73 20 3d 20 6e 3b 0d  {...LL ans = n;.
04d0: 0a 09 66 6f 72 20 28 4c 4c 20 78 20 3d 20 32 3b  ..for (LL x = 2;
04e0: 20 78 2a 78 20 3c 3d 20 6e 3b 20 2b 2b 78 29 20   x*x <= n; ++x) 
04f0: 7b 0d 0a 09 09 69 66 20 28 6e 20 25 20 78 20 3d  {....if (n % x =
0500: 3d 20 30 29 20 7b 0d 0a 09 09 09 61 6e 73 20 2d  = 0) {.....ans -
0510: 3d 20 61 6e 73 20 2f 20 78 3b 0d 0a 09 09 09 77  = ans / x;.....w
0520: 68 69 6c 65 20 28 6e 20 25 20 78 20 3d 3d 20 30  hile (n % x == 0
0530: 29 20 6e 20 2f 3d 20 78 3b 0d 0a 09 09 7d 0d 0a  ) n /= x;....}..
0540: 09 7d 0d 0a 09 69 66 20 28 6e 20 3e 20 31 29 20  .}...if (n > 1) 
0550: 61 6e 73 20 2d 3d 20 61 6e 73 20 2f 20 6e 3b 0d  ans -= ans / n;.
0560: 0a 09 72 65 74 75 72 6e 20 61 6e 73 3b 0d 0a 7d  ..return ans;..}
0570: 0d 0a 0d 0a 63 6c 61 73 73 20 4d 6f 64 45 71 75  ....class ModEqu
0580: 61 74 69 6f 6e 20 7b 20 70 75 62 6c 69 63 3a 0d  ation { public:.
0590: 0a 09 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 63  ..vector <int> c
05a0: 6f 75 6e 74 28 69 6e 74 20 6e 2c 20 69 6e 74 20  ount(int n, int 
05b0: 4b 2c 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20  K, vector <int> 
05c0: 71 75 65 72 79 29 0d 0a 09 7b 0d 0a 09 09 69 6e  query)...{....in
05d0: 74 20 65 70 6b 20 3d 20 28 69 6e 74 29 65 75 6c  t epk = (int)eul
05e0: 65 72 50 68 69 28 4b 29 3b 0d 0a 0d 0a 09 09 76  erPhi(K);......v
05f0: 65 63 74 6f 72 3c 70 61 69 72 3c 69 6e 74 2c 20  ector<pair<int, 
0600: 69 6e 74 3e 3e 20 64 4b 3b 0d 0a 09 09 69 6e 74  int>> dK;....int
0610: 20 76 20 3d 20 4b 3b 0d 0a 09 09 66 6f 72 20 28   v = K;....for (
0620: 69 6e 74 20 70 20 3d 20 32 3b 20 70 2a 70 20 3c  int p = 2; p*p <
0630: 3d 20 76 3b 20 2b 2b 70 29 0d 0a 09 09 09 69 66  = v; ++p).....if
0640: 20 28 76 25 70 20 3d 3d 20 30 29 20 7b 0d 0a 09   (v%p == 0) {...
0650: 09 09 09 69 6e 74 20 63 6e 74 20 3d 20 30 3b 0d  ...int cnt = 0;.
0660: 0a 09 09 09 09 77 68 69 6c 65 20 28 76 25 70 20  .....while (v%p 
0670: 3d 3d 20 30 29 20 63 6e 74 2b 2b 2c 20 76 20 2f  == 0) cnt++, v /
0680: 3d 20 70 3b 0d 0a 09 09 09 09 64 4b 2e 65 6d 70  = p;......dK.emp
0690: 6c 61 63 65 5f 62 61 63 6b 28 70 2c 20 63 6e 74  lace_back(p, cnt
06a0: 29 3b 0d 0a 09 09 09 7d 0d 0a 09 09 69 66 20 28  );.....}....if (
06b0: 76 20 21 3d 20 31 29 0d 0a 09 09 09 64 4b 2e 65  v != 1).....dK.e
06c0: 6d 70 6c 61 63 65 5f 62 61 63 6b 28 76 2c 20 31  mplace_back(v, 1
06d0: 29 3b 0d 0a 0d 0a 09 09 76 65 63 74 6f 72 3c 69  );......vector<i
06e0: 6e 74 3e 20 61 6e 73 3b 0d 0a 09 09 66 6f 72 20  nt> ans;....for 
06f0: 28 69 6e 74 20 71 20 3a 20 71 75 65 72 79 29 0d  (int q : query).
0700: 0a 09 09 09 61 6e 73 2e 70 75 73 68 5f 62 61 63  ....ans.push_bac
0710: 6b 28 73 6f 6c 76 65 28 6e 2c 20 4b 2c 20 65 70  k(solve(n, K, ep
0720: 6b 2c 20 64 4b 2c 20 71 29 2e 76 61 6c 29 3b 0d  k, dK, q).val);.
0730: 0a 09 09 72 65 74 75 72 6e 20 61 6e 73 3b 0d 0a  ...return ans;..
0740: 09 7d 0d 0a 0d 0a 09 2f 2f 20 23 20 6f 66 20 73  .}.....// # of s
0750: 6f 6c 75 74 69 6f 6e 73 20 69 6e 20 6d 6f 64 20  olutions in mod 
0760: 4b 20 77 6f 72 6c 64 20 73 2e 74 2e 20 78 5b 30  K world s.t. x[0
0770: 5d 20 78 5b 31 5d 20 2e 2e 2e 20 78 5b 6e 2d 31  ] x[1] ... x[n-1
0780: 5d 20 3d 20 71 0d 0a 09 2f 2f 0d 0a 09 2f 2f 20  ] = q...//...// 
0790: 2a 20 6e 20 3c 3d 20 35 30 0d 0a 09 2f 2f 20 2a  * n <= 50...// *
07a0: 20 4b 20 3d 20 42 49 47 0d 0a 09 2f 2f 20 2a 20   K = BIG...// * 
07b0: 6d 75 73 74 20 62 65 20 73 6f 6c 76 65 64 20 31  must be solved 1
07c0: 30 30 30 20 74 69 6d 65 73 20 69 6e 20 74 69 6d  000 times in tim
07d0: 65 20 6c 69 6d 69 74 73 2e 0d 0a 09 6d 69 6e 74  e limits....mint
07e0: 20 73 6f 6c 76 65 28 69 6e 74 20 6e 2c 20 69 6e   solve(int n, in
07f0: 74 20 4b 2c 20 69 6e 74 20 65 70 6b 2c 20 63 6f  t K, int epk, co
0800: 6e 73 74 20 76 65 63 74 6f 72 3c 70 61 69 72 3c  nst vector<pair<
0810: 69 6e 74 2c 69 6e 74 3e 3e 26 20 64 4b 2c 20 69  int,int>>& dK, i
0820: 6e 74 20 71 29 0d 0a 09 7b 0d 0a 09 09 69 66 20  nt q)...{....if 
0830: 28 6e 20 3d 3d 20 31 29 0d 0a 09 09 09 72 65 74  (n == 1).....ret
0840: 75 72 6e 20 31 3b 0d 0a 0d 0a 09 09 6d 69 6e 74  urn 1;......mint
0850: 20 61 6e 73 20 3d 20 50 4f 57 28 65 70 6b 2c 20   ans = POW(epk, 
0860: 6e 2d 31 29 3b 0d 0a 0d 0a 09 09 66 6f 72 20 28  n-1);......for (
0870: 61 75 74 6f 20 70 65 20 3a 20 64 4b 29 20 7b 0d  auto pe : dK) {.
0880: 0a 09 09 09 69 6e 74 20 70 20 3d 20 70 65 2e 66  ....int p = pe.f
0890: 69 72 73 74 3b 0d 0a 09 09 09 69 6e 74 20 65 20  irst;.....int e 
08a0: 3d 20 70 65 2e 73 65 63 6f 6e 64 3b 0d 0a 09 09  = pe.second;....
08b0: 09 69 6e 74 20 71 71 20 3d 20 71 3b 0d 0a 09 09  .int qq = q;....
08c0: 09 69 6e 74 20 63 6e 74 20 3d 20 30 3b 0d 0a 09  .int cnt = 0;...
08d0: 09 09 77 68 69 6c 65 20 28 71 71 25 70 20 3d 3d  ..while (qq%p ==
08e0: 20 30 20 26 26 20 63 6e 74 3c 65 29 0d 0a 09 09   0 && cnt<e)....
08f0: 09 09 63 6e 74 2b 2b 2c 20 71 71 20 2f 3d 20 70  ..cnt++, qq /= p
0900: 3b 0d 0a 09 09 09 2f 2f 20 70 5e 63 6e 74 20 69  ;.....// p^cnt i
0910: 73 20 69 6e 20 67 63 64 28 4b 2c 71 29 0d 0a 09  s in gcd(K,q)...
0920: 09 09 69 66 20 28 63 6e 74 20 3c 20 65 29 20 7b  ..if (cnt < e) {
0930: 0d 0a 09 09 09 09 61 6e 73 20 2a 3d 20 50 4f 57  ......ans *= POW
0940: 28 6e 2c 20 63 6e 74 29 3b 20 2f 2f 20 64 69 76  (n, cnt); // div
0950: 20 64 75 70 73 2e 2e 2e 0d 0a 09 09 09 7d 0d 0a   dups........}..
0960: 09 09 09 65 6c 73 65 20 7b 0d 0a 09 09 09 09 61  ...else {......a
0970: 6e 73 20 2a 3d 20 50 4f 57 28 6e 2c 20 63 6e 74  ns *= POW(n, cnt
0980: 29 3b 20 2f 2f 20 6f 72 20 6d 6f 72 65 2e 2e 2e  ); // or more...
0990: 0d 0a 09 09 09 7d 0d 0a 09 09 7d 0d 0a 0d 0a 09  .....}....}.....
09a0: 09 72 65 74 75 72 6e 20 61 6e 73 3b 0d 0a 09 7d  .return ans;...}
09b0: 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 42 45 47 49 4e  ..};....// BEGIN
09c0: 20 43 55 54 20 48 45 52 45 0d 0a 23 69 6e 63 6c   CUT HERE..#incl
09d0: 75 64 65 20 3c 63 74 69 6d 65 3e 0d 0a 64 6f 75  ude <ctime>..dou
09e0: 62 6c 65 20 73 74 61 72 74 5f 74 69 6d 65 3b 20  ble start_time; 
09f0: 73 74 72 69 6e 67 20 74 69 6d 65 72 28 29 0d 0a  string timer()..
0a00: 20 7b 20 6f 73 74 72 69 6e 67 73 74 72 65 61 6d   { ostringstream
0a10: 20 6f 73 3b 20 6f 73 20 3c 3c 20 22 20 28 22 20   os; os << " (" 
0a20: 3c 3c 20 69 6e 74 28 28 63 6c 6f 63 6b 28 29 2d  << int((clock()-
0a30: 73 74 61 72 74 5f 74 69 6d 65 29 2f 43 4c 4f 43  start_time)/CLOC
0a40: 4b 53 5f 50 45 52 5f 53 45 43 2a 31 30 30 30 29  KS_PER_SEC*1000)
0a50: 20 3c 3c 20 22 20 6d 73 65 63 29 22 3b 20 72 65   << " msec)"; re
0a60: 74 75 72 6e 20 6f 73 2e 73 74 72 28 29 3b 20 7d  turn os.str(); }
0a70: 0d 0a 74 65 6d 70 6c 61 74 65 3c 74 79 70 65 6e  ..template<typen
0a80: 61 6d 65 20 54 3e 20 6f 73 74 72 65 61 6d 26 20  ame T> ostream& 
0a90: 6f 70 65 72 61 74 6f 72 3c 3c 28 6f 73 74 72 65  operator<<(ostre
0aa0: 61 6d 26 20 6f 73 2c 20 63 6f 6e 73 74 20 76 65  am& os, const ve
0ab0: 63 74 6f 72 3c 54 3e 26 20 76 29 0d 0a 20 7b 20  ctor<T>& v).. { 
0ac0: 6f 73 20 3c 3c 20 22 7b 20 22 3b 0d 0a 20 20 20  os << "{ ";..   
0ad0: 66 6f 72 28 74 79 70 65 6e 61 6d 65 20 76 65 63  for(typename vec
0ae0: 74 6f 72 3c 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74  tor<T>::const_it
0af0: 65 72 61 74 6f 72 20 69 74 3d 76 2e 62 65 67 69  erator it=v.begi
0b00: 6e 28 29 3b 20 69 74 21 3d 76 2e 65 6e 64 28 29  n(); it!=v.end()
0b10: 3b 20 2b 2b 69 74 29 0d 0a 20 20 20 6f 73 20 3c  ; ++it)..   os <
0b20: 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 74 20 3c 3c  < '\"' << *it <<
0b30: 20 27 5c 22 27 20 3c 3c 20 28 69 74 2b 31 3d 3d   '\"' << (it+1==
0b40: 76 2e 65 6e 64 28 29 20 3f 20 22 22 20 3a 20 22  v.end() ? "" : "
0b50: 2c 20 22 29 3b 20 6f 73 20 3c 3c 20 22 20 7d 22  , "); os << " }"
0b60: 3b 20 72 65 74 75 72 6e 20 6f 73 3b 20 7d 0d 0a  ; return os; }..
0b70: 76 6f 69 64 20 76 65 72 69 66 79 5f 63 61 73 65  void verify_case
0b80: 28 63 6f 6e 73 74 20 76 65 63 74 6f 72 20 3c 69  (const vector <i
0b90: 6e 74 3e 26 20 45 78 70 65 63 74 65 64 2c 20 63  nt>& Expected, c
0ba0: 6f 6e 73 74 20 76 65 63 74 6f 72 20 3c 69 6e 74  onst vector <int
0bb0: 3e 26 20 52 65 63 65 69 76 65 64 29 20 7b 0d 0a  >& Received) {..
0bc0: 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28 45 78 70 65   bool ok = (Expe
0bd0: 63 74 65 64 20 3d 3d 20 52 65 63 65 69 76 65 64  cted == Received
0be0: 29 3b 0d 0a 20 69 66 28 6f 6b 29 20 63 65 72 72  );.. if(ok) cerr
0bf0: 20 3c 3c 20 22 50 41 53 53 45 44 22 20 3c 3c 20   << "PASSED" << 
0c00: 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b  timer() << endl;
0c10: 20 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c 3c    else { cerr <<
0c20: 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 69 6d   "FAILED" << tim
0c30: 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 0d 0a 20  er() << endl;.. 
0c40: 63 65 72 72 20 3c 3c 20 22 5c 74 6f 3a 20 22 20  cerr << "\to: " 
0c50: 3c 3c 20 45 78 70 65 63 74 65 64 20 3c 3c 20 65  << Expected << e
0c60: 6e 64 6c 20 3c 3c 20 22 5c 74 78 3a 20 22 20 3c  ndl << "\tx: " <
0c70: 3c 20 52 65 63 65 69 76 65 64 20 3c 3c 20 65 6e  < Received << en
0c80: 64 6c 3b 20 7d 20 7d 0d 0a 23 64 65 66 69 6e 65  dl; } }..#define
0c90: 20 43 41 53 45 28 4e 29 20 7b 63 65 72 72 20 3c   CASE(N) {cerr <
0ca0: 3c 20 22 54 65 73 74 20 43 61 73 65 20 23 22 20  < "Test Case #" 
0cb0: 3c 3c 20 4e 20 3c 3c 20 22 2e 2e 2e 22 20 3c 3c  << N << "..." <<
0cc0: 20 66 6c 75 73 68 3b 20 73 74 61 72 74 5f 74 69   flush; start_ti
0cd0: 6d 65 3d 63 6c 6f 63 6b 28 29 3b 0d 0a 23 64 65  me=clock();..#de
0ce0: 66 69 6e 65 20 45 4e 44 09 20 76 65 72 69 66 79  fine END. verify
0cf0: 5f 63 61 73 65 28 5f 2c 20 4d 6f 64 45 71 75 61  _case(_, ModEqua
0d00: 74 69 6f 6e 28 29 2e 63 6f 75 6e 74 28 6e 2c 20  tion().count(n, 
0d10: 4b 2c 20 71 75 65 72 79 29 29 3b 7d 0d 0a 69 6e  K, query));}..in
0d20: 74 20 6d 61 69 6e 28 29 7b 0d 0a 0d 0a 43 41 53  t main(){....CAS
0d30: 45 28 30 29 0d 0a 09 69 6e 74 20 6e 20 3d 20 32  E(0)...int n = 2
0d40: 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d 20 32 3b 20  ; ...int K = 2; 
0d50: 0d 0a 09 69 6e 74 20 71 75 65 72 79 5f 5b 5d 20  ...int query_[] 
0d60: 3d 20 7b 30 2c 31 7d 3b 0d 0a 09 20 20 76 65 63  = {0,1};...  vec
0d70: 74 6f 72 20 3c 69 6e 74 3e 20 71 75 65 72 79 28  tor <int> query(
0d80: 71 75 65 72 79 5f 2c 20 71 75 65 72 79 5f 2b 73  query_, query_+s
0d90: 69 7a 65 6f 66 28 71 75 65 72 79 5f 29 2f 73 69  izeof(query_)/si
0da0: 7a 65 6f 66 28 2a 71 75 65 72 79 5f 29 29 3b 20  zeof(*query_)); 
0db0: 0d 0a 09 69 6e 74 20 5f 5f 5b 5d 20 3d 20 7b 33  ...int __[] = {3
0dc0: 2c 20 31 20 7d 3b 0d 0a 09 20 20 76 65 63 74 6f  , 1 };...  vecto
0dd0: 72 20 3c 69 6e 74 3e 20 5f 28 5f 5f 2c 20 5f 5f  r <int> _(__, __
0de0: 2b 73 69 7a 65 6f 66 28 5f 5f 29 2f 73 69 7a 65  +sizeof(__)/size
0df0: 6f 66 28 2a 5f 5f 29 29 3b 20 0d 0a 45 4e 44 0d  of(*__)); ..END.
0e00: 0a 43 41 53 45 28 31 29 0d 0a 09 69 6e 74 20 6e  .CASE(1)...int n
0e10: 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d   = 2; ...int K =
0e20: 20 34 3b 20 0d 0a 09 69 6e 74 20 71 75 65 72 79   4; ...int query
0e30: 5f 5b 5d 20 3d 20 7b 30 2c 31 2c 32 2c 33 7d 3b  _[] = {0,1,2,3};
0e40: 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74  ...  vector <int
0e50: 3e 20 71 75 65 72 79 28 71 75 65 72 79 5f 2c 20  > query(query_, 
0e60: 71 75 65 72 79 5f 2b 73 69 7a 65 6f 66 28 71 75  query_+sizeof(qu
0e70: 65 72 79 5f 29 2f 73 69 7a 65 6f 66 28 2a 71 75  ery_)/sizeof(*qu
0e80: 65 72 79 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 5f  ery_)); ...int _
0e90: 5f 5b 5d 20 3d 20 7b 38 2c 20 32 2c 20 34 2c 20  _[] = {8, 2, 4, 
0ea0: 32 20 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20  2 };...  vector 
0eb0: 3c 69 6e 74 3e 20 5f 28 5f 5f 2c 20 5f 5f 2b 73  <int> _(__, __+s
0ec0: 69 7a 65 6f 66 28 5f 5f 29 2f 73 69 7a 65 6f 66  izeof(__)/sizeof
0ed0: 28 2a 5f 5f 29 29 3b 20 0d 0a 45 4e 44 0d 0a 43  (*__)); ..END..C
0ee0: 41 53 45 28 32 29 0d 0a 09 69 6e 74 20 6e 20 3d  ASE(2)...int n =
0ef0: 20 36 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d 20 36   6; ...int K = 6
0f00: 3b 20 0d 0a 09 69 6e 74 20 71 75 65 72 79 5f 5b  ; ...int query_[
0f10: 5d 20 3d 20 7b 34 7d 3b 0d 0a 09 20 20 76 65 63  ] = {4};...  vec
0f20: 74 6f 72 20 3c 69 6e 74 3e 20 71 75 65 72 79 28  tor <int> query(
0f30: 71 75 65 72 79 5f 2c 20 71 75 65 72 79 5f 2b 73  query_, query_+s
0f40: 69 7a 65 6f 66 28 71 75 65 72 79 5f 29 2f 73 69  izeof(query_)/si
0f50: 7a 65 6f 66 28 2a 71 75 65 72 79 5f 29 29 3b 20  zeof(*query_)); 
0f60: 0d 0a 09 69 6e 74 20 5f 5f 5b 5d 20 3d 20 7b 32  ...int __[] = {2
0f70: 30 31 36 20 7d 3b 0d 0a 09 20 20 76 65 63 74 6f  016 };...  vecto
0f80: 72 20 3c 69 6e 74 3e 20 5f 28 5f 5f 2c 20 5f 5f  r <int> _(__, __
0f90: 2b 73 69 7a 65 6f 66 28 5f 5f 29 2f 73 69 7a 65  +sizeof(__)/size
0fa0: 6f 66 28 2a 5f 5f 29 29 3b 20 0d 0a 45 4e 44 0d  of(*__)); ..END.
0fb0: 0a 43 41 53 45 28 33 29 0d 0a 09 69 6e 74 20 6e  .CASE(3)...int n
0fc0: 20 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d   = 1; ...int K =
0fd0: 20 32 3b 20 0d 0a 09 69 6e 74 20 71 75 65 72 79   2; ...int query
0fe0: 5f 5b 5d 20 3d 20 7b 30 2c 30 2c 30 2c 31 2c 31  _[] = {0,0,0,1,1
0ff0: 2c 31 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20  ,1};...  vector 
1000: 3c 69 6e 74 3e 20 71 75 65 72 79 28 71 75 65 72  <int> query(quer
1010: 79 5f 2c 20 71 75 65 72 79 5f 2b 73 69 7a 65 6f  y_, query_+sizeo
1020: 66 28 71 75 65 72 79 5f 29 2f 73 69 7a 65 6f 66  f(query_)/sizeof
1030: 28 2a 71 75 65 72 79 5f 29 29 3b 20 0d 0a 09 69  (*query_)); ...i
1040: 6e 74 20 5f 5f 5b 5d 20 3d 20 7b 31 2c 20 31 2c  nt __[] = {1, 1,
1050: 20 31 2c 20 31 2c 20 31 2c 20 31 20 7d 3b 0d 0a   1, 1, 1, 1 };..
1060: 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20  .  vector <int> 
1070: 5f 28 5f 5f 2c 20 5f 5f 2b 73 69 7a 65 6f 66 28  _(__, __+sizeof(
1080: 5f 5f 29 2f 73 69 7a 65 6f 66 28 2a 5f 5f 29 29  __)/sizeof(*__))
1090: 3b 20 0d 0a 45 4e 44 0d 0a 2f 2a 0d 0a 43 41 53  ; ..END../*..CAS
10a0: 45 28 34 29 0d 0a 09 69 6e 74 20 6e 20 3d 20 3b  E(4)...int n = ;
10b0: 20 0d 0a 09 69 6e 74 20 4b 20 3d 20 3b 20 0d 0a   ...int K = ; ..
10c0: 09 69 6e 74 20 71 75 65 72 79 5f 5b 5d 20 3d 20  .int query_[] = 
10d0: 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 6e  ;...  vector <in
10e0: 74 3e 20 71 75 65 72 79 28 71 75 65 72 79 5f 2c  t> query(query_,
10f0: 20 71 75 65 72 79 5f 2b 73 69 7a 65 6f 66 28 71   query_+sizeof(q
1100: 75 65 72 79 5f 29 2f 73 69 7a 65 6f 66 28 2a 71  uery_)/sizeof(*q
1110: 75 65 72 79 5f 29 29 3b 20 0d 0a 09 69 6e 74 20  uery_)); ...int 
1120: 5f 5f 5b 5d 20 3d 20 3b 0d 0a 09 20 20 76 65 63  __[] = ;...  vec
1130: 74 6f 72 20 3c 69 6e 74 3e 20 5f 28 5f 5f 2c 20  tor <int> _(__, 
1140: 5f 5f 2b 73 69 7a 65 6f 66 28 5f 5f 29 2f 73 69  __+sizeof(__)/si
1150: 7a 65 6f 66 28 2a 5f 5f 29 29 3b 20 0d 0a 45 4e  zeof(*__)); ..EN
1160: 44 0d 0a 43 41 53 45 28 35 29 0d 0a 09 69 6e 74  D..CASE(5)...int
1170: 20 6e 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 4b 20   n = ; ...int K 
1180: 3d 20 3b 20 0d 0a 09 69 6e 74 20 71 75 65 72 79  = ; ...int query
1190: 5f 5b 5d 20 3d 20 3b 0d 0a 09 20 20 76 65 63 74  _[] = ;...  vect
11a0: 6f 72 20 3c 69 6e 74 3e 20 71 75 65 72 79 28 71  or <int> query(q
11b0: 75 65 72 79 5f 2c 20 71 75 65 72 79 5f 2b 73 69  uery_, query_+si
11c0: 7a 65 6f 66 28 71 75 65 72 79 5f 29 2f 73 69 7a  zeof(query_)/siz
11d0: 65 6f 66 28 2a 71 75 65 72 79 5f 29 29 3b 20 0d  eof(*query_)); .
11e0: 0a 09 69 6e 74 20 5f 5f 5b 5d 20 3d 20 3b 0d 0a  ..int __[] = ;..
11f0: 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20  .  vector <int> 
1200: 5f 28 5f 5f 2c 20 5f 5f 2b 73 69 7a 65 6f 66 28  _(__, __+sizeof(
1210: 5f 5f 29 2f 73 69 7a 65 6f 66 28 2a 5f 5f 29 29  __)/sizeof(*__))
1220: 3b 20 0d 0a 45 4e 44 0d 0a 2a 2f 0d 0a 7d 0d 0a  ; ..END..*/..}..
1230: 2f 2f 20 45 4e 44 20 43 55 54 20 48 45 52 45 0d  // END CUT HERE.
1240: 0a                                               .