Artifact 29d8b3c8147890dbd94157bb268f511e5543cd33:
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 63 73 74 72 69 6e 67 3e 0d 0a 75 73 69 6e 67 <cstring>..using
0150: 20 6e 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d namespace std;.
0160: 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f .typedef long lo
0170: 6e 67 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 ng LL;..typedef
0180: 63 6f 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 complex<double>
0190: 43 4d 50 3b 0d 0a 0d 0a 73 74 61 74 69 63 20 63 CMP;....static c
01a0: 6f 6e 73 74 20 69 6e 74 20 4d 4f 44 56 41 4c 20 onst int MODVAL
01b0: 3d 20 31 30 30 30 30 30 30 30 30 37 3b 0d 0a 0d = 1000000007;...
01c0: 0a 62 6f 6f 6c 20 69 73 5f 73 71 75 61 72 65 5f .bool is_square_
01d0: 66 72 65 65 28 69 6e 74 20 6e 29 0d 0a 7b 0d 0a free(int n)..{..
01e0: 09 66 6f 72 28 69 6e 74 20 70 3d 32 3b 20 70 2a .for(int p=2; p*
01f0: 70 3c 3d 6e 3b 20 2b 2b 70 29 0d 0a 09 09 69 66 p<=n; ++p)....if
0200: 28 20 6e 25 28 70 2a 70 29 20 3d 3d 20 30 20 29 ( n%(p*p) == 0 )
0210: 0d 0a 09 09 09 72 65 74 75 72 6e 20 66 61 6c 73 .....return fals
0220: 65 3b 0d 0a 09 72 65 74 75 72 6e 20 74 72 75 65 e;...return true
0230: 3b 0d 0a 7d 0d 0a 0d 0a 73 74 61 74 69 63 20 63 ;..}....static c
0240: 6f 6e 73 74 20 69 6e 74 20 53 50 5b 5d 20 3d 20 onst int SP[] =
0250: 7b 32 2c 33 2c 35 2c 37 2c 31 31 2c 31 33 2c 31 {2,3,5,7,11,13,1
0260: 37 2c 31 39 7d 3b 20 2f 2f 20 3c 3d 20 73 71 72 7,19}; // <= sqr
0270: 74 28 4e 29 0d 0a 73 74 61 74 69 63 20 63 6f 6e t(N)..static con
0280: 73 74 20 69 6e 74 20 53 50 4e 20 20 3d 20 73 69 st int SPN = si
0290: 7a 65 6f 66 28 53 50 29 2f 73 69 7a 65 6f 66 28 zeof(SP)/sizeof(
02a0: 2a 53 50 29 3b 0d 0a 0d 0a 69 6e 74 20 73 6d 61 *SP);....int sma
02b0: 6c 6c 5f 70 72 69 6d 65 5f 6d 61 73 6b 28 69 6e ll_prime_mask(in
02c0: 74 20 6e 29 0d 0a 7b 0d 0a 09 69 6e 74 20 6d 20 t n)..{...int m
02d0: 3d 20 30 3b 0d 0a 09 66 6f 72 28 69 6e 74 20 69 = 0;...for(int i
02e0: 3d 30 3b 20 69 3c 53 50 4e 3b 20 2b 2b 69 29 0d =0; i<SPN; ++i).
02f0: 0a 09 09 69 66 28 20 6e 25 53 50 5b 69 5d 20 3d ...if( n%SP[i] =
0300: 3d 20 30 20 29 0d 0a 09 09 09 6d 20 7c 3d 20 31 = 0 ).....m |= 1
0310: 3c 3c 69 3b 0d 0a 09 72 65 74 75 72 6e 20 6d 3b <<i;...return m;
0320: 0d 0a 7d 0d 0a 0d 0a 69 6e 74 20 62 69 67 5f 70 ..}....int big_p
0330: 72 69 6d 65 5f 66 61 63 74 6f 72 28 69 6e 74 20 rime_factor(int
0340: 6e 29 0d 0a 7b 0d 0a 09 66 6f 72 28 69 6e 74 20 n)..{...for(int
0350: 69 3d 30 3b 20 69 3c 53 50 4e 3b 20 2b 2b 69 29 i=0; i<SPN; ++i)
0360: 0d 0a 09 09 77 68 69 6c 65 28 20 6e 25 53 50 5b ....while( n%SP[
0370: 69 5d 20 3d 3d 20 30 20 29 0d 0a 09 09 09 6e 20 i] == 0 ).....n
0380: 2f 3d 20 53 50 5b 69 5d 3b 0d 0a 09 72 65 74 75 /= SP[i];...retu
0390: 72 6e 20 6e 3b 0d 0a 7d 0d 0a 0d 0a 63 6c 61 73 rn n;..}....clas
03a0: 73 20 53 71 75 61 72 65 46 72 65 65 53 65 74 73 s SquareFreeSets
03b0: 20 7b 0d 0a 70 75 62 6c 69 63 3a 0d 0a 09 69 6e {..public:...in
03c0: 74 20 63 6f 75 6e 74 50 65 72 66 65 63 74 28 69 t countPerfect(i
03d0: 6e 74 20 4e 2c 20 69 6e 74 20 4b 29 20 0d 0a 09 nt N, int K) ...
03e0: 7b 0d 0a 09 09 2f 2f 20 77 61 79 5b 6b 5d 5b 6d {....// way[k][m
03f0: 5d 20 3d 20 23 20 6f 66 20 77 61 79 73 20 74 6f ] = # of ways to
0400: 20 6d 61 6b 65 20 61 20 6b 2d 65 6c 65 6d 20 70 make a k-elem p
0410: 73 66 2d 73 65 74 20 77 69 74 68 20 73 6d 61 6c sf-set with smal
0420: 6c 20 70 72 69 6d 65 20 66 61 63 74 6f 72 73 20 l prime factors
0430: 6d 0d 0a 09 09 69 6e 74 20 77 61 79 5b 35 30 31 m....int way[501
0440: 5d 5b 31 3c 3c 53 50 4e 5d 20 3d 20 7b 7d 3b 0d ][1<<SPN] = {};.
0450: 0a 09 09 77 61 79 5b 30 5d 5b 30 5d 20 3d 20 31 ...way[0][0] = 1
0460: 3b 0d 0a 0d 0a 09 09 66 6f 72 28 69 6e 74 20 6e ;......for(int n
0470: 3d 32 3b 20 6e 3c 3d 4e 3b 20 2b 2b 6e 29 0d 0a =2; n<=N; ++n)..
0480: 09 09 09 69 66 28 20 69 73 5f 73 71 75 61 72 65 ...if( is_square
0490: 5f 66 72 65 65 28 6e 29 20 29 0d 0a 09 09 09 09 _free(n) )......
04a0: 69 66 28 20 62 69 67 5f 70 72 69 6d 65 5f 66 61 if( big_prime_fa
04b0: 63 74 6f 72 28 6e 29 20 3d 3d 20 31 20 29 0d 0a ctor(n) == 1 )..
04c0: 09 09 09 09 7b 0d 0a 09 09 09 09 09 69 6e 74 20 ....{.......int
04d0: 6d 61 73 6b 20 3d 20 73 6d 61 6c 6c 5f 70 72 69 mask = small_pri
04e0: 6d 65 5f 6d 61 73 6b 28 6e 29 3b 0d 0a 09 09 09 me_mask(n);.....
04f0: 09 09 66 6f 72 28 69 6e 74 20 6b 3d 4b 3b 20 6b ..for(int k=K; k
0500: 3e 3d 31 3b 20 2d 2d 6b 29 0d 0a 09 09 09 09 09 >=1; --k).......
0510: 09 66 6f 72 28 69 6e 74 20 6d 3d 30 3b 20 6d 3c .for(int m=0; m<
0520: 28 31 3c 3c 53 50 4e 29 3b 20 2b 2b 6d 29 0d 0a (1<<SPN); ++m)..
0530: 09 09 09 09 09 09 09 69 66 28 20 28 6d 7c 6d 61 .......if( (m|ma
0540: 73 6b 29 20 3d 3d 20 6d 20 29 0d 0a 09 09 09 09 sk) == m )......
0550: 09 09 09 09 77 61 79 5b 6b 5d 5b 6d 5d 20 3d 20 ....way[k][m] =
0560: 28 77 61 79 5b 6b 5d 5b 6d 5d 20 2b 20 77 61 79 (way[k][m] + way
0570: 5b 6b 2d 31 5d 5b 6d 26 7e 6d 61 73 6b 5d 29 20 [k-1][m&~mask])
0580: 25 20 4d 4f 44 56 41 4c 3b 0d 0a 09 09 09 09 7d % MODVAL;......}
0590: 0d 0a 09 09 09 09 65 6c 73 65 20 69 66 28 20 62 ......else if( b
05a0: 69 67 5f 70 72 69 6d 65 5f 66 61 63 74 6f 72 28 ig_prime_factor(
05b0: 6e 29 20 3d 3d 20 6e 20 29 0d 0a 09 09 09 09 7b n) == n )......{
05c0: 0d 0a 09 09 09 09 09 76 65 63 74 6f 72 3c 69 6e .......vector<in
05d0: 74 3e 20 6d 61 73 6b 3b 0d 0a 09 09 09 09 09 66 t> mask;.......f
05e0: 6f 72 28 69 6e 74 20 70 3d 31 3b 20 6e 2a 70 3c or(int p=1; n*p<
05f0: 3d 4e 3b 20 2b 2b 70 29 0d 0a 09 09 09 09 09 09 =N; ++p)........
0600: 69 66 28 20 69 73 5f 73 71 75 61 72 65 5f 66 72 if( is_square_fr
0610: 65 65 28 70 29 20 29 0d 0a 09 09 09 09 09 09 09 ee(p) ).........
0620: 6d 61 73 6b 2e 70 75 73 68 5f 62 61 63 6b 28 20 mask.push_back(
0630: 73 6d 61 6c 6c 5f 70 72 69 6d 65 5f 6d 61 73 6b small_prime_mask
0640: 28 70 29 20 29 3b 0d 0a 0d 0a 09 09 09 09 09 66 (p) );.........f
0650: 6f 72 28 69 6e 74 20 6b 3d 4b 3b 20 6b 3e 3d 31 or(int k=K; k>=1
0660: 3b 20 2d 2d 6b 29 0d 0a 09 09 09 09 09 09 66 6f ; --k)........fo
0670: 72 28 69 6e 74 20 6d 3d 30 3b 20 6d 3c 28 31 3c r(int m=0; m<(1<
0680: 3c 53 50 4e 29 3b 20 2b 2b 6d 29 0d 0a 09 09 09 <SPN); ++m).....
0690: 09 09 09 09 66 6f 72 28 69 6e 74 20 6a 3d 30 3b ....for(int j=0;
06a0: 20 6a 3c 6d 61 73 6b 2e 73 69 7a 65 28 29 3b 20 j<mask.size();
06b0: 2b 2b 6a 29 0d 0a 09 09 09 09 09 09 09 09 69 66 ++j)..........if
06c0: 28 20 28 6d 7c 6d 61 73 6b 5b 6a 5d 29 20 3d 3d ( (m|mask[j]) ==
06d0: 20 6d 20 29 0d 0a 09 09 09 09 09 09 09 09 09 77 m )...........w
06e0: 61 79 5b 6b 5d 5b 6d 5d 20 3d 20 28 77 61 79 5b ay[k][m] = (way[
06f0: 6b 5d 5b 6d 5d 20 2b 20 77 61 79 5b 6b 2d 31 5d k][m] + way[k-1]
0700: 5b 6d 26 7e 6d 61 73 6b 5b 6a 5d 5d 29 20 25 20 [m&~mask[j]]) %
0710: 4d 4f 44 56 41 4c 3b 0d 0a 09 09 09 09 7d 0d 0a MODVAL;......}..
0720: 0d 0a 09 09 2f 2f 20 73 75 6d 6d 69 6e 67 20 75 ....// summing u
0730: 70 0d 0a 09 09 69 6e 74 20 73 75 6d 20 3d 20 30 p....int sum = 0
0740: 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 6b 3d 31 ;....for(int k=1
0750: 3b 20 6b 3c 3d 4b 3b 20 2b 2b 6b 29 0d 0a 09 09 ; k<=K; ++k)....
0760: 09 66 6f 72 28 69 6e 74 20 6d 3d 30 3b 20 6d 3c .for(int m=0; m<
0770: 28 31 3c 3c 53 50 4e 29 3b 20 2b 2b 6d 29 0d 0a (1<<SPN); ++m)..
0780: 09 09 09 09 73 75 6d 20 3d 20 28 73 75 6d 20 2b ....sum = (sum +
0790: 20 77 61 79 5b 6b 5d 5b 6d 5d 29 20 25 20 4d 4f way[k][m]) % MO
07a0: 44 56 41 4c 3b 0d 0a 09 09 72 65 74 75 72 6e 20 DVAL;....return
07b0: 73 75 6d 3b 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a sum;...}..};....
07c0: 2f 2f 20 42 45 47 49 4e 20 43 55 54 20 48 45 52 // BEGIN CUT HER
07d0: 45 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 63 74 69 E..#include <cti
07e0: 6d 65 3e 0d 0a 64 6f 75 62 6c 65 20 73 74 61 72 me>..double star
07f0: 74 5f 74 69 6d 65 3b 73 74 72 69 6e 67 20 74 69 t_time;string ti
0800: 6d 65 72 28 29 20 7b 20 6f 73 74 72 69 6e 67 73 mer() { ostrings
0810: 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 tream os; os <<
0820: 22 20 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f " (" << int((clo
0830: 63 6b 28 29 2d 73 74 61 72 74 5f 74 69 6d 65 29 ck()-start_time)
0840: 2f 43 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a /CLOCKS_PER_SEC*
0850: 31 30 30 30 29 20 3c 3c 20 22 20 6d 73 65 63 29 1000) << " msec)
0860: 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 "; return os.str
0870: 28 29 3b 20 7d 0d 0a 0d 0a 74 65 6d 70 6c 61 74 (); }....templat
0880: 65 20 3c 74 79 70 65 6e 61 6d 65 20 54 3e 20 73 e <typename T> s
0890: 74 72 69 6e 67 20 70 72 69 6e 74 5f 61 72 72 61 tring print_arra
08a0: 79 28 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c 54 y(const vector<T
08b0: 3e 20 26 56 29 20 7b 20 6f 73 74 72 69 6e 67 73 > &V) { ostrings
08c0: 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 tream os; os <<
08d0: 22 7b 20 22 3b 20 66 6f 72 20 28 74 79 70 65 6e "{ "; for (typen
08e0: 61 6d 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a 63 ame vector<T>::c
08f0: 6f 6e 73 74 5f 69 74 65 72 61 74 6f 72 20 69 74 onst_iterator it
0900: 65 72 20 3d 20 56 2e 62 65 67 69 6e 28 29 3b 20 er = V.begin();
0910: 69 74 65 72 20 21 3d 20 56 2e 65 6e 64 28 29 3b iter != V.end();
0920: 20 2b 2b 69 74 65 72 29 20 6f 73 20 3c 3c 20 27 ++iter) os << '
0930: 5c 22 27 20 3c 3c 20 2a 69 74 65 72 20 3c 3c 20 \"' << *iter <<
0940: 22 5c 22 2c 22 3b 20 6f 73 20 3c 3c 20 22 20 7d "\","; os << " }
0950: 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 "; return os.str
0960: 28 29 3b 20 7d 0d 0a 69 6e 74 20 76 65 72 69 66 (); }..int verif
0970: 79 5f 63 61 73 65 28 63 6f 6e 73 74 20 69 6e 74 y_case(const int
0980: 20 26 45 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 &Expected, cons
0990: 74 20 69 6e 74 20 26 52 65 63 65 69 76 65 64 29 t int &Received)
09a0: 20 7b 20 69 66 20 28 45 78 70 65 63 74 65 64 20 { if (Expected
09b0: 3d 3d 20 52 65 63 65 69 76 65 64 29 20 63 65 72 == Received) cer
09c0: 72 20 3c 3c 20 22 50 41 53 53 45 44 22 20 3c 3c r << "PASSED" <<
09d0: 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c timer() << endl
09e0: 3b 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c 3c ; else { cerr <<
09f0: 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 69 6d "FAILED" << tim
0a00: 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 20 63 65 er() << endl; ce
0a10: 72 72 20 3c 3c 20 22 5c 74 45 78 70 65 63 74 65 rr << "\tExpecte
0a20: 64 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 d: \"" << Expect
0a30: 65 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e ed << '\"' << en
0a40: 64 6c 3b 20 63 65 72 72 20 3c 3c 20 22 5c 74 52 dl; cerr << "\tR
0a50: 65 63 65 69 76 65 64 3a 20 5c 22 22 20 3c 3c 20 eceived: \"" <<
0a60: 52 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 Received << '\"'
0a70: 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 72 65 74 75 << endl; } retu
0a80: 72 6e 20 30 3b 7d 0d 0a 0d 0a 74 65 6d 70 6c 61 rn 0;}....templa
0a90: 74 65 3c 69 6e 74 20 4e 3e 20 73 74 72 75 63 74 te<int N> struct
0aa0: 20 43 61 73 65 5f 20 7b 20 43 61 73 65 5f 28 29 Case_ { Case_()
0ab0: 7b 73 74 61 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 {start_time=cloc
0ac0: 6b 28 29 3b 7d 20 7d 3b 0d 0a 63 68 61 72 20 54 k();} };..char T
0ad0: 65 73 74 5f 28 2e 2e 2e 29 3b 0d 0a 69 6e 74 20 est_(...);..int
0ae0: 54 65 73 74 5f 28 43 61 73 65 5f 3c 30 3e 29 20 Test_(Case_<0>)
0af0: 7b 0d 0a 09 69 6e 74 20 4e 20 3d 20 35 3b 20 0d {...int N = 5; .
0b00: 0a 09 69 6e 74 20 4b 20 3d 20 31 3b 20 0d 0a 09 ..int K = 1; ...
0b10: 69 6e 74 20 52 65 74 56 61 6c 20 3d 20 33 3b 20 int RetVal = 3;
0b20: 0d 0a 09 72 65 74 75 72 6e 20 76 65 72 69 66 79 ...return verify
0b30: 5f 63 61 73 65 28 52 65 74 56 61 6c 2c 20 53 71 _case(RetVal, Sq
0b40: 75 61 72 65 46 72 65 65 53 65 74 73 28 29 2e 63 uareFreeSets().c
0b50: 6f 75 6e 74 50 65 72 66 65 63 74 28 4e 2c 20 4b ountPerfect(N, K
0b60: 29 29 3b 20 7d 0d 0a 69 6e 74 20 54 65 73 74 5f )); }..int Test_
0b70: 28 43 61 73 65 5f 3c 31 3e 29 20 7b 0d 0a 09 69 (Case_<1>) {...i
0b80: 6e 74 20 4e 20 3d 20 35 3b 20 0d 0a 09 69 6e 74 nt N = 5; ...int
0b90: 20 4b 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 52 K = 2; ...int R
0ba0: 65 74 56 61 6c 20 3d 20 36 3b 20 0d 0a 09 72 65 etVal = 6; ...re
0bb0: 74 75 72 6e 20 76 65 72 69 66 79 5f 63 61 73 65 turn verify_case
0bc0: 28 52 65 74 56 61 6c 2c 20 53 71 75 61 72 65 46 (RetVal, SquareF
0bd0: 72 65 65 53 65 74 73 28 29 2e 63 6f 75 6e 74 50 reeSets().countP
0be0: 65 72 66 65 63 74 28 4e 2c 20 4b 29 29 3b 20 7d erfect(N, K)); }
0bf0: 0d 0a 69 6e 74 20 54 65 73 74 5f 28 43 61 73 65 ..int Test_(Case
0c00: 5f 3c 32 3e 29 20 7b 0d 0a 09 69 6e 74 20 4e 20 _<2>) {...int N
0c10: 3d 20 35 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d 20 = 5; ...int K =
0c20: 33 3b 20 0d 0a 09 69 6e 74 20 52 65 74 56 61 6c 3; ...int RetVal
0c30: 20 3d 20 37 3b 20 0d 0a 09 72 65 74 75 72 6e 20 = 7; ...return
0c40: 76 65 72 69 66 79 5f 63 61 73 65 28 52 65 74 56 verify_case(RetV
0c50: 61 6c 2c 20 53 71 75 61 72 65 46 72 65 65 53 65 al, SquareFreeSe
0c60: 74 73 28 29 2e 63 6f 75 6e 74 50 65 72 66 65 63 ts().countPerfec
0c70: 74 28 4e 2c 20 4b 29 29 3b 20 7d 0d 0a 69 6e 74 t(N, K)); }..int
0c80: 20 54 65 73 74 5f 28 43 61 73 65 5f 3c 33 3e 29 Test_(Case_<3>)
0c90: 20 7b 0d 0a 09 69 6e 74 20 4e 20 3d 20 36 3b 20 {...int N = 6;
0ca0: 0d 0a 09 69 6e 74 20 4b 20 3d 20 33 3b 20 0d 0a ...int K = 3; ..
0cb0: 09 69 6e 74 20 52 65 74 56 61 6c 20 3d 20 39 3b .int RetVal = 9;
0cc0: 20 0d 0a 09 72 65 74 75 72 6e 20 76 65 72 69 66 ...return verif
0cd0: 79 5f 63 61 73 65 28 52 65 74 56 61 6c 2c 20 53 y_case(RetVal, S
0ce0: 71 75 61 72 65 46 72 65 65 53 65 74 73 28 29 2e quareFreeSets().
0cf0: 63 6f 75 6e 74 50 65 72 66 65 63 74 28 4e 2c 20 countPerfect(N,
0d00: 4b 29 29 3b 20 7d 0d 0a 69 6e 74 20 54 65 73 74 K)); }..int Test
0d10: 5f 28 43 61 73 65 5f 3c 34 3e 29 20 7b 0d 0a 09 _(Case_<4>) {...
0d20: 69 6e 74 20 4e 20 3d 20 32 38 3b 20 0d 0a 09 69 int N = 28; ...i
0d30: 6e 74 20 4b 20 3d 20 34 31 3b 20 0d 0a 09 69 6e nt K = 41; ...in
0d40: 74 20 52 65 74 56 61 6c 20 3d 20 31 35 39 39 3b t RetVal = 1599;
0d50: 20 0d 0a 09 72 65 74 75 72 6e 20 76 65 72 69 66 ...return verif
0d60: 79 5f 63 61 73 65 28 52 65 74 56 61 6c 2c 20 53 y_case(RetVal, S
0d70: 71 75 61 72 65 46 72 65 65 53 65 74 73 28 29 2e quareFreeSets().
0d80: 63 6f 75 6e 74 50 65 72 66 65 63 74 28 4e 2c 20 countPerfect(N,
0d90: 4b 29 29 3b 20 7d 0d 0a 69 6e 74 20 54 65 73 74 K)); }..int Test
0da0: 5f 28 43 61 73 65 5f 3c 35 3e 29 20 7b 0d 0a 09 _(Case_<5>) {...
0db0: 69 6e 74 20 4e 20 3d 20 35 30 30 3b 20 0d 0a 09 int N = 500; ...
0dc0: 69 6e 74 20 4b 20 3d 20 35 30 30 3b 20 0d 0a 09 int K = 500; ...
0dd0: 69 6e 74 20 52 65 74 56 61 6c 20 3d 20 35 39 39 int RetVal = 599
0de0: 36 34 31 30 34 3b 20 0d 0a 09 72 65 74 75 72 6e 64104; ...return
0df0: 20 76 65 72 69 66 79 5f 63 61 73 65 28 52 65 74 verify_case(Ret
0e00: 56 61 6c 2c 20 53 71 75 61 72 65 46 72 65 65 53 Val, SquareFreeS
0e10: 65 74 73 28 29 2e 63 6f 75 6e 74 50 65 72 66 65 ets().countPerfe
0e20: 63 74 28 4e 2c 20 4b 29 29 3b 20 7d 0d 0a 69 6e ct(N, K)); }..in
0e30: 74 20 54 65 73 74 5f 28 43 61 73 65 5f 3c 36 3e t Test_(Case_<6>
0e40: 29 20 7b 0d 0a 09 69 6e 74 20 4e 20 3d 20 31 30 ) {...int N = 10
0e50: 30 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d 20 31 3b 0; ...int K = 1;
0e60: 0d 0a 09 69 6e 74 20 52 65 74 56 61 6c 20 3d 20 ...int RetVal =
0e70: 36 30 3b 20 0d 0a 09 72 65 74 75 72 6e 20 76 65 60; ...return ve
0e80: 72 69 66 79 5f 63 61 73 65 28 52 65 74 56 61 6c rify_case(RetVal
0e90: 2c 20 53 71 75 61 72 65 46 72 65 65 53 65 74 73 , SquareFreeSets
0ea0: 28 29 2e 63 6f 75 6e 74 50 65 72 66 65 63 74 28 ().countPerfect(
0eb0: 4e 2c 20 4b 29 29 3b 20 7d 0d 0a 0d 0a 74 65 6d N, K)); }....tem
0ec0: 70 6c 61 74 65 3c 69 6e 74 20 4e 3e 20 76 6f 69 plate<int N> voi
0ed0: 64 20 52 75 6e 5f 28 29 20 7b 20 63 65 72 72 20 d Run_() { cerr
0ee0: 3c 3c 20 22 54 65 73 74 20 43 61 73 65 20 23 22 << "Test Case #"
0ef0: 20 3c 3c 20 4e 20 3c 3c 20 22 2e 2e 2e 22 20 3c << N << "..." <
0f00: 3c 20 66 6c 75 73 68 3b 20 54 65 73 74 5f 28 43 < flush; Test_(C
0f10: 61 73 65 5f 3c 4e 3e 28 29 29 3b 20 52 75 6e 5f ase_<N>()); Run_
0f20: 3c 73 69 7a 65 6f 66 28 54 65 73 74 5f 28 43 61 <sizeof(Test_(Ca
0f30: 73 65 5f 3c 4e 2b 31 3e 28 29 29 29 3d 3d 31 20 se_<N+1>()))==1
0f40: 3f 20 2d 31 20 3a 20 4e 2b 31 3e 28 29 3b 20 7d ? -1 : N+1>(); }
0f50: 0d 0a 74 65 6d 70 6c 61 74 65 3c 3e 20 20 20 20 ..template<>
0f60: 20 20 76 6f 69 64 20 52 75 6e 5f 3c 2d 31 3e 28 void Run_<-1>(
0f70: 29 20 7b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 ) {}..int main()
0f80: 20 7b 20 52 75 6e 5f 3c 30 3e 28 29 3b 20 7d 0d { Run_<0>(); }.
0f90: 0a 2f 2f 20 45 4e 44 20 43 55 54 20 48 45 52 45 .// END CUT HERE
0fa0: 0d 0a 0d 0a ....