Artifact 7fcb01fe631559ac2437f67df56803edae5ef48a:
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 63 6c 61 73 73 ong LL;....class
0140: 20 44 69 76 69 73 69 62 69 6c 69 74 79 52 75 6c DivisibilityRul
0150: 65 73 0d 0a 7b 0d 0a 70 75 62 6c 69 63 3a 0d 0a es..{..public:..
0160: 09 69 6e 74 20 73 69 6d 69 6c 61 72 28 69 6e 74 .int similar(int
0170: 20 6e 75 6d 65 72 61 74 69 6f 6e 42 61 73 65 2c numerationBase,
0180: 20 69 6e 74 20 64 69 76 69 73 6f 72 29 20 0d 0a int divisor) ..
0190: 09 7b 0d 0a 09 09 76 65 63 74 6f 72 3c 20 76 65 .{....vector< ve
01a0: 63 74 6f 72 3c 69 6e 74 3e 20 3e 20 6d 62 3b 0d ctor<int> > mb;.
01b0: 0a 09 09 66 6f 72 28 69 6e 74 20 6b 3d 32 3b 20 ...for(int k=2;
01c0: 6b 3c 6e 75 6d 65 72 61 74 69 6f 6e 42 61 73 65 k<numerationBase
01d0: 3b 20 2b 2b 6b 29 0d 0a 09 09 09 6d 62 2e 70 75 ; ++k).....mb.pu
01e0: 73 68 5f 62 61 63 6b 28 20 6d 62 61 73 65 28 6e sh_back( mbase(n
01f0: 75 6d 65 72 61 74 69 6f 6e 42 61 73 65 2c 20 6b umerationBase, k
0200: 29 20 29 3b 0d 0a 09 09 72 65 74 75 72 6e 20 63 ) );....return c
0210: 6f 75 6e 74 28 20 6d 62 2e 62 65 67 69 6e 28 29 ount( mb.begin()
0220: 2c 20 6d 62 2e 65 6e 64 28 29 2c 20 6d 62 5b 64 , mb.end(), mb[d
0230: 69 76 69 73 6f 72 2d 32 5d 20 29 3b 0d 0a 09 7d ivisor-2] );...}
0240: 0d 0a 0d 0a 09 76 65 63 74 6f 72 3c 69 6e 74 3e .....vector<int>
0250: 20 6d 62 61 73 65 28 69 6e 74 20 6e 2c 20 69 6e mbase(int n, in
0260: 74 20 6b 29 0d 0a 09 7b 0d 0a 09 09 76 65 63 74 t k)...{....vect
0270: 6f 72 3c 69 6e 74 3e 20 72 3b 0d 0a 09 09 69 6e or<int> r;....in
0280: 74 20 62 20 3d 20 31 3b 0d 0a 09 09 66 6f 72 28 t b = 1;....for(
0290: 69 6e 74 20 69 3d 30 3b 20 69 3c 6e 3b 20 2b 2b int i=0; i<n; ++
02a0: 69 29 20 7b 0d 0a 09 09 09 72 2e 70 75 73 68 5f i) {.....r.push_
02b0: 62 61 63 6b 28 62 29 3b 0d 0a 09 09 09 62 20 3d back(b);.....b =
02c0: 20 28 62 2a 6e 29 25 6b 3b 0d 0a 09 09 7d 0d 0a (b*n)%k;....}..
02d0: 09 09 72 65 74 75 72 6e 20 72 3b 0d 0a 09 7d 0d ..return r;...}.
02e0: 0a 0d 0a 2f 2f 20 42 45 47 49 4e 20 43 55 54 20 ...// BEGIN CUT
02f0: 48 45 52 45 0d 0a 09 70 75 62 6c 69 63 3a 0d 0a HERE...public:..
0300: 09 76 6f 69 64 20 72 75 6e 5f 74 65 73 74 28 69 .void run_test(i
0310: 6e 74 20 43 61 73 65 29 20 7b 20 69 66 20 28 28 nt Case) { if ((
0320: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
0330: 43 61 73 65 20 3d 3d 20 30 29 29 20 74 65 73 74 Case == 0)) test
0340: 5f 63 61 73 65 5f 30 28 29 3b 20 69 66 20 28 28 _case_0(); if ((
0350: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
0360: 43 61 73 65 20 3d 3d 20 31 29 29 20 74 65 73 74 Case == 1)) test
0370: 5f 63 61 73 65 5f 31 28 29 3b 20 69 66 20 28 28 _case_1(); if ((
0380: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
0390: 43 61 73 65 20 3d 3d 20 32 29 29 20 74 65 73 74 Case == 2)) test
03a0: 5f 63 61 73 65 5f 32 28 29 3b 20 69 66 20 28 28 _case_2(); if ((
03b0: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
03c0: 43 61 73 65 20 3d 3d 20 33 29 29 20 74 65 73 74 Case == 3)) test
03d0: 5f 63 61 73 65 5f 33 28 29 3b 20 69 66 20 28 28 _case_3(); if ((
03e0: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
03f0: 43 61 73 65 20 3d 3d 20 34 29 29 20 74 65 73 74 Case == 4)) test
0400: 5f 63 61 73 65 5f 34 28 29 3b 20 69 66 20 28 28 _case_4(); if ((
0410: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
0420: 43 61 73 65 20 3d 3d 20 35 29 29 20 74 65 73 74 Case == 5)) test
0430: 5f 63 61 73 65 5f 35 28 29 3b 20 7d 0d 0a 09 70 _case_5(); }...p
0440: 72 69 76 61 74 65 3a 0d 0a 09 74 65 6d 70 6c 61 rivate:...templa
0450: 74 65 20 3c 74 79 70 65 6e 61 6d 65 20 54 3e 20 te <typename T>
0460: 73 74 72 69 6e 67 20 70 72 69 6e 74 5f 61 72 72 string print_arr
0470: 61 79 28 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c ay(const vector<
0480: 54 3e 20 26 56 29 20 7b 20 6f 73 74 72 69 6e 67 T> &V) { ostring
0490: 73 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c stream os; os <<
04a0: 20 22 7b 20 22 3b 20 66 6f 72 20 28 74 79 70 65 "{ "; for (type
04b0: 6e 61 6d 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a name vector<T>::
04c0: 63 6f 6e 73 74 5f 69 74 65 72 61 74 6f 72 20 69 const_iterator i
04d0: 74 65 72 20 3d 20 56 2e 62 65 67 69 6e 28 29 3b ter = V.begin();
04e0: 20 69 74 65 72 20 21 3d 20 56 2e 65 6e 64 28 29 iter != V.end()
04f0: 3b 20 2b 2b 69 74 65 72 29 20 6f 73 20 3c 3c 20 ; ++iter) os <<
0500: 27 5c 22 27 20 3c 3c 20 2a 69 74 65 72 20 3c 3c '\"' << *iter <<
0510: 20 22 5c 22 2c 22 3b 20 6f 73 20 3c 3c 20 22 20 "\","; os << "
0520: 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 }"; return os.st
0530: 72 28 29 3b 20 7d 0d 0a 09 76 6f 69 64 20 76 65 r(); }...void ve
0540: 72 69 66 79 5f 63 61 73 65 28 69 6e 74 20 43 61 rify_case(int Ca
0550: 73 65 2c 20 63 6f 6e 73 74 20 69 6e 74 20 26 45 se, const int &E
0560: 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 20 69 xpected, const i
0570: 6e 74 20 26 52 65 63 65 69 76 65 64 29 20 7b 20 nt &Received) {
0580: 63 65 72 72 20 3c 3c 20 22 54 65 73 74 20 43 61 cerr << "Test Ca
0590: 73 65 20 23 22 20 3c 3c 20 43 61 73 65 20 3c 3c se #" << Case <<
05a0: 20 22 2e 2e 2e 22 3b 20 69 66 20 28 45 78 70 65 "..."; if (Expe
05b0: 63 74 65 64 20 3d 3d 20 52 65 63 65 69 76 65 64 cted == Received
05c0: 29 20 63 65 72 72 20 3c 3c 20 22 50 41 53 53 45 ) cerr << "PASSE
05d0: 44 22 20 3c 3c 20 65 6e 64 6c 3b 20 65 6c 73 65 D" << endl; else
05e0: 20 7b 20 63 65 72 72 20 3c 3c 20 22 46 41 49 4c { cerr << "FAIL
05f0: 45 44 22 20 3c 3c 20 65 6e 64 6c 3b 20 63 65 72 ED" << endl; cer
0600: 72 20 3c 3c 20 22 5c 74 45 78 70 65 63 74 65 64 r << "\tExpected
0610: 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 65 : \"" << Expecte
0620: 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 d << '\"' << end
0630: 6c 3b 20 63 65 72 72 20 3c 3c 20 22 5c 74 52 65 l; cerr << "\tRe
0640: 63 65 69 76 65 64 3a 20 5c 22 22 20 3c 3c 20 52 ceived: \"" << R
0650: 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 20 eceived << '\"'
0660: 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a 09 76 << endl; } }...v
0670: 6f 69 64 20 74 65 73 74 5f 63 61 73 65 5f 30 28 oid test_case_0(
0680: 29 20 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 31 ) { int Arg0 = 1
0690: 30 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 33 3b 0; int Arg1 = 3;
06a0: 20 69 6e 74 20 41 72 67 32 20 3d 20 32 3b 20 76 int Arg2 = 2; v
06b0: 65 72 69 66 79 5f 63 61 73 65 28 30 2c 20 41 72 erify_case(0, Ar
06c0: 67 32 2c 20 73 69 6d 69 6c 61 72 28 41 72 67 30 g2, similar(Arg0
06d0: 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76 6f , Arg1)); }...vo
06e0: 69 64 20 74 65 73 74 5f 63 61 73 65 5f 31 28 29 id test_case_1()
06f0: 20 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 31 30 { int Arg0 = 10
0700: 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 35 3b 20 ; int Arg1 = 5;
0710: 69 6e 74 20 41 72 67 32 20 3d 20 32 3b 20 76 65 int Arg2 = 2; ve
0720: 72 69 66 79 5f 63 61 73 65 28 31 2c 20 41 72 67 rify_case(1, Arg
0730: 32 2c 20 73 69 6d 69 6c 61 72 28 41 72 67 30 2c 2, similar(Arg0,
0740: 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76 6f 69 Arg1)); }...voi
0750: 64 20 74 65 73 74 5f 63 61 73 65 5f 32 28 29 20 d test_case_2()
0760: 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 35 31 31 { int Arg0 = 511
0770: 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 33 32 3b ; int Arg1 = 32;
0780: 20 69 6e 74 20 41 72 67 32 20 3d 20 31 30 3b 20 int Arg2 = 10;
0790: 76 65 72 69 66 79 5f 63 61 73 65 28 32 2c 20 41 verify_case(2, A
07a0: 72 67 32 2c 20 73 69 6d 69 6c 61 72 28 41 72 67 rg2, similar(Arg
07b0: 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76 0, Arg1)); }...v
07c0: 6f 69 64 20 74 65 73 74 5f 63 61 73 65 5f 33 28 oid test_case_3(
07d0: 29 20 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 33 ) { int Arg0 = 3
07e0: 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 32 3b 20 ; int Arg1 = 2;
07f0: 69 6e 74 20 41 72 67 32 20 3d 20 31 3b 20 76 65 int Arg2 = 1; ve
0800: 72 69 66 79 5f 63 61 73 65 28 33 2c 20 41 72 67 rify_case(3, Arg
0810: 32 2c 20 73 69 6d 69 6c 61 72 28 41 72 67 30 2c 2, similar(Arg0,
0820: 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76 6f 69 Arg1)); }...voi
0830: 64 20 74 65 73 74 5f 63 61 73 65 5f 34 28 29 20 d test_case_4()
0840: 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 31 30 30 { int Arg0 = 100
0850: 30 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 39 39 0; int Arg1 = 99
0860: 39 3b 20 69 6e 74 20 41 72 67 32 20 3d 20 37 3b 9; int Arg2 = 7;
0870: 20 76 65 72 69 66 79 5f 63 61 73 65 28 34 2c 20 verify_case(4,
0880: 41 72 67 32 2c 20 73 69 6d 69 6c 61 72 28 41 72 Arg2, similar(Ar
0890: 67 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 g0, Arg1)); }...
08a0: 76 6f 69 64 20 74 65 73 74 5f 63 61 73 65 5f 35 void test_case_5
08b0: 28 29 20 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 () { int Arg0 =
08c0: 36 35 35 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 655; int Arg1 =
08d0: 35 33 32 3b 20 69 6e 74 20 41 72 67 32 20 3d 20 532; int Arg2 =
08e0: 31 3b 20 76 65 72 69 66 79 5f 63 61 73 65 28 35 1; verify_case(5
08f0: 2c 20 41 72 67 32 2c 20 73 69 6d 69 6c 61 72 28 , Arg2, similar(
0900: 41 72 67 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d Arg0, Arg1)); }.
0910: 0a 0d 0a 2f 2f 20 45 4e 44 20 43 55 54 20 48 45 ...// END CUT HE
0920: 52 45 0d 0a 7d 3b 0d 0a 2f 2f 20 42 45 47 49 4e RE..};..// BEGIN
0930: 20 43 55 54 20 48 45 52 45 20 0d 0a 69 6e 74 20 CUT HERE ..int
0940: 6d 61 69 6e 28 29 20 7b 20 44 69 76 69 73 69 62 main() { Divisib
0950: 69 6c 69 74 79 52 75 6c 65 73 28 29 2e 72 75 6e ilityRules().run
0960: 5f 74 65 73 74 28 2d 31 29 3b 20 7d 0d 0a 2f 2f _test(-1); }..//
0970: 20 45 4e 44 20 43 55 54 20 48 45 52 45 0d 0a END CUT HERE..