Artifact 6a8dec603d2a8fec9078e09a10469d61c6ba3408:
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 69 6e 74 20 67 63 64 28 CMP;....int gcd(
01a0: 69 6e 74 20 61 2c 20 69 6e 74 20 62 29 20 7b 20 int a, int b) {
01b0: 77 68 69 6c 65 28 61 29 20 73 77 61 70 28 61 2c while(a) swap(a,
01c0: 62 25 3d 61 29 3b 20 72 65 74 75 72 6e 20 62 3b b%=a); return b;
01d0: 20 7d 0d 0a 0d 0a 63 6c 61 73 73 20 53 65 74 74 }....class Sett
01e0: 69 6e 67 54 65 6e 74 73 20 7b 0d 0a 70 75 62 6c ingTents {..publ
01f0: 69 63 3a 0d 0a 09 69 6e 74 20 63 6f 75 6e 74 53 ic:...int countS
0200: 69 74 65 73 28 69 6e 74 20 4e 2c 20 69 6e 74 20 ites(int N, int
0210: 4d 29 20 0d 0a 09 7b 0d 0a 09 09 69 6e 74 20 63 M) ...{....int c
0220: 6e 74 20 3d 20 30 3b 0d 0a 09 09 66 6f 72 28 69 nt = 0;....for(i
0230: 6e 74 20 77 3d 31 3b 20 77 3c 3d 4e 3b 20 2b 2b nt w=1; w<=N; ++
0240: 77 29 20 20 20 20 20 2f 2f 20 66 6f 72 20 65 61 w) // for ea
0250: 63 68 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 68 ch.....for(int h
0260: 3d 31 3b 20 68 3c 3d 4d 3b 20 2b 2b 68 29 20 2f =1; h<=M; ++h) /
0270: 2f 20 62 6f 75 6e 64 69 6e 67 20 62 6f 78 2e 2e / bounding box..
0280: 2e 0d 0a 09 09 09 7b 0d 0a 09 09 09 09 69 6e 74 ......{......int
0290: 20 6e 75 6d 57 48 20 3d 20 28 4e 2b 31 2d 77 29 numWH = (N+1-w)
02a0: 2a 28 4d 2b 31 2d 68 29 3b 0d 0a 0d 0a 09 09 09 *(M+1-h);.......
02b0: 09 2f 2f 20 77 2f 32 20 2d 20 6b 28 68 2d 32 6a .// w/2 - k(h-2j
02c0: 29 20 3d 20 78 0d 0a 09 09 09 09 2f 2f 20 68 2f ) = x......// h/
02d0: 32 20 2b 20 6b 77 20 20 20 20 20 20 3d 20 79 0d 2 + kw = y.
02e0: 0a 09 09 09 09 66 6f 72 28 69 6e 74 20 6a 3d 31 .....for(int j=1
02f0: 3b 20 6a 3c 68 3b 20 2b 2b 6a 29 20 2f 2f 20 74 ; j<h; ++j) // t
0300: 68 65 20 63 61 73 65 20 77 68 65 6e 20 62 6f 74 he case when bot
0310: 68 20 74 77 6f 20 64 69 61 67 73 20 63 6f 6e 74 h two diags cont
0320: 72 69 62 75 74 65 20 74 6f 20 74 68 65 20 62 6f ribute to the bo
0330: 75 6e 64 73 0d 0a 09 09 09 09 09 69 66 28 20 28 unds.......if( (
0340: 77 2a 77 2b 68 2a 28 68 2d 32 2a 6a 29 29 20 25 w*w+h*(h-2*j)) %
0350: 20 28 32 2a 77 29 20 3d 3d 20 30 20 29 0d 0a 09 (2*w) == 0 )...
0360: 09 09 09 09 7b 0d 0a 09 09 09 09 09 09 69 6e 74 ....{........int
0370: 20 78 20 3d 20 28 77 2a 77 2b 68 2a 28 68 2d 32 x = (w*w+h*(h-2
0380: 2a 6a 29 29 20 2f 20 28 32 2a 77 29 3b 0d 0a 09 *j)) / (2*w);...
0390: 09 09 09 09 09 69 66 28 20 30 3c 3d 78 20 26 26 .....if( 0<=x &&
03a0: 20 78 3c 3d 77 20 29 0d 0a 09 09 09 09 09 09 09 x<=w ).........
03b0: 63 6e 74 20 2b 3d 20 6e 75 6d 57 48 3b 0d 0a 09 cnt += numWH;...
03c0: 09 09 09 09 7d 0d 0a 0d 0a 09 09 09 09 66 6f 72 ....}........for
03d0: 28 69 6e 74 20 79 3d 30 3b 20 32 2a 79 3c 68 3b (int y=0; 2*y<h;
03e0: 20 2b 2b 79 29 20 2f 2f 20 74 68 65 20 63 61 73 ++y) // the cas
03f0: 65 20 77 68 65 6e 20 6f 6e 6c 79 20 6f 6e 65 20 e when only one
0400: 64 69 61 67 20 63 6f 6e 74 72 69 62 75 74 20 74 diag contribut t
0410: 6f 20 74 68 65 20 62 6f 75 6e 64 73 0d 0a 09 09 o the bounds....
0420: 09 09 09 69 66 28 20 28 77 2a 77 20 2d 20 28 32 ...if( (w*w - (2
0430: 2a 79 2d 68 29 2a 68 29 20 25 20 28 32 2a 77 29 *y-h)*h) % (2*w)
0440: 20 3d 3d 20 30 20 29 0d 0a 09 09 09 09 09 7b 0d == 0 ).......{.
0450: 0a 09 09 09 09 09 09 69 6e 74 20 78 20 3d 20 28 .......int x = (
0460: 77 2a 77 20 2d 20 28 32 2a 79 2d 68 29 2a 68 29 w*w - (2*y-h)*h)
0470: 20 2f 20 28 32 2a 77 29 3b 0d 0a 09 09 09 09 09 / (2*w);.......
0480: 09 69 66 28 20 78 3c 77 20 29 0d 0a 09 09 09 09 .if( x<w )......
0490: 09 09 09 63 6e 74 20 2b 3d 20 32 2a 6e 75 6d 57 ...cnt += 2*numW
04a0: 48 3b 0d 0a 09 09 09 09 09 7d 0d 0a 0d 0a 09 09 H;.......}......
04b0: 09 09 69 66 28 20 77 3d 3d 68 20 29 20 2f 2f 20 ..if( w==h ) //
04c0: 73 70 65 63 69 61 6c 20 63 61 73 65 0d 0a 09 09 special case....
04d0: 09 09 09 63 6e 74 20 2b 3d 20 6e 75 6d 57 48 3b ...cnt += numWH;
04e0: 0d 0a 09 09 09 7d 0d 0a 09 09 72 65 74 75 72 6e .....}....return
04f0: 20 63 6e 74 3b 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d cnt;...}..};...
0500: 0a 2f 2f 20 42 45 47 49 4e 20 43 55 54 20 48 45 .// BEGIN CUT HE
0510: 52 45 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 63 74 RE..#include <ct
0520: 69 6d 65 3e 0d 0a 64 6f 75 62 6c 65 20 73 74 61 ime>..double sta
0530: 72 74 5f 74 69 6d 65 3b 73 74 72 69 6e 67 20 74 rt_time;string t
0540: 69 6d 65 72 28 29 20 7b 20 6f 73 74 72 69 6e 67 imer() { ostring
0550: 73 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c stream os; os <<
0560: 20 22 20 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c " (" << int((cl
0570: 6f 63 6b 28 29 2d 73 74 61 72 74 5f 74 69 6d 65 ock()-start_time
0580: 29 2f 43 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43 )/CLOCKS_PER_SEC
0590: 2a 31 30 30 30 29 20 3c 3c 20 22 20 6d 73 65 63 *1000) << " msec
05a0: 29 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 )"; return os.st
05b0: 72 28 29 3b 20 7d 0d 0a 0d 0a 74 65 6d 70 6c 61 r(); }....templa
05c0: 74 65 20 3c 74 79 70 65 6e 61 6d 65 20 54 3e 20 te <typename T>
05d0: 73 74 72 69 6e 67 20 70 72 69 6e 74 5f 61 72 72 string print_arr
05e0: 61 79 28 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c ay(const vector<
05f0: 54 3e 20 26 56 29 20 7b 20 6f 73 74 72 69 6e 67 T> &V) { ostring
0600: 73 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c stream os; os <<
0610: 20 22 7b 20 22 3b 20 66 6f 72 20 28 74 79 70 65 "{ "; for (type
0620: 6e 61 6d 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a name vector<T>::
0630: 63 6f 6e 73 74 5f 69 74 65 72 61 74 6f 72 20 69 const_iterator i
0640: 74 65 72 20 3d 20 56 2e 62 65 67 69 6e 28 29 3b ter = V.begin();
0650: 20 69 74 65 72 20 21 3d 20 56 2e 65 6e 64 28 29 iter != V.end()
0660: 3b 20 2b 2b 69 74 65 72 29 20 6f 73 20 3c 3c 20 ; ++iter) os <<
0670: 27 5c 22 27 20 3c 3c 20 2a 69 74 65 72 20 3c 3c '\"' << *iter <<
0680: 20 22 5c 22 2c 22 3b 20 6f 73 20 3c 3c 20 22 20 "\","; os << "
0690: 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 }"; return os.st
06a0: 72 28 29 3b 20 7d 0d 0a 69 6e 74 20 76 65 72 69 r(); }..int veri
06b0: 66 79 5f 63 61 73 65 28 63 6f 6e 73 74 20 69 6e fy_case(const in
06c0: 74 20 26 45 78 70 65 63 74 65 64 2c 20 63 6f 6e t &Expected, con
06d0: 73 74 20 69 6e 74 20 26 52 65 63 65 69 76 65 64 st int &Received
06e0: 29 20 7b 20 69 66 20 28 45 78 70 65 63 74 65 64 ) { if (Expected
06f0: 20 3d 3d 20 52 65 63 65 69 76 65 64 29 20 63 65 == Received) ce
0700: 72 72 20 3c 3c 20 22 50 41 53 53 45 44 22 20 3c rr << "PASSED" <
0710: 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 < timer() << end
0720: 6c 3b 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c l; else { cerr <
0730: 3c 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 69 < "FAILED" << ti
0740: 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 20 63 mer() << endl; c
0750: 65 72 72 20 3c 3c 20 22 5c 74 45 78 70 65 63 74 err << "\tExpect
0760: 65 64 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 ed: \"" << Expec
0770: 74 65 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 ted << '\"' << e
0780: 6e 64 6c 3b 20 63 65 72 72 20 3c 3c 20 22 5c 74 ndl; cerr << "\t
0790: 52 65 63 65 69 76 65 64 3a 20 5c 22 22 20 3c 3c Received: \"" <<
07a0: 20 52 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 Received << '\"
07b0: 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 72 65 74 ' << endl; } ret
07c0: 75 72 6e 20 30 3b 7d 0d 0a 0d 0a 74 65 6d 70 6c urn 0;}....templ
07d0: 61 74 65 3c 69 6e 74 20 4e 3e 20 73 74 72 75 63 ate<int N> struc
07e0: 74 20 43 61 73 65 5f 20 7b 20 43 61 73 65 5f 28 t Case_ { Case_(
07f0: 29 7b 73 74 61 72 74 5f 74 69 6d 65 3d 63 6c 6f ){start_time=clo
0800: 63 6b 28 29 3b 7d 20 7d 3b 0d 0a 63 68 61 72 20 ck();} };..char
0810: 54 65 73 74 5f 28 2e 2e 2e 29 3b 0d 0a 69 6e 74 Test_(...);..int
0820: 20 54 65 73 74 5f 28 43 61 73 65 5f 3c 30 3e 29 Test_(Case_<0>)
0830: 20 7b 0d 0a 09 69 6e 74 20 4e 20 3d 20 32 3b 20 {...int N = 2;
0840: 0d 0a 09 69 6e 74 20 4d 20 3d 20 32 3b 20 0d 0a ...int M = 2; ..
0850: 09 69 6e 74 20 52 65 74 56 61 6c 20 3d 20 36 3b .int RetVal = 6;
0860: 20 0d 0a 09 72 65 74 75 72 6e 20 76 65 72 69 66 ...return verif
0870: 79 5f 63 61 73 65 28 52 65 74 56 61 6c 2c 20 53 y_case(RetVal, S
0880: 65 74 74 69 6e 67 54 65 6e 74 73 28 29 2e 63 6f ettingTents().co
0890: 75 6e 74 53 69 74 65 73 28 4e 2c 20 4d 29 29 3b untSites(N, M));
08a0: 20 7d 0d 0a 69 6e 74 20 54 65 73 74 5f 28 43 61 }..int Test_(Ca
08b0: 73 65 5f 3c 31 3e 29 20 7b 0d 0a 09 69 6e 74 20 se_<1>) {...int
08c0: 4e 20 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 4d 20 N = 1; ...int M
08d0: 3d 20 36 3b 20 0d 0a 09 69 6e 74 20 52 65 74 56 = 6; ...int RetV
08e0: 61 6c 20 3d 20 36 3b 20 0d 0a 09 72 65 74 75 72 al = 6; ...retur
08f0: 6e 20 76 65 72 69 66 79 5f 63 61 73 65 28 52 65 n verify_case(Re
0900: 74 56 61 6c 2c 20 53 65 74 74 69 6e 67 54 65 6e tVal, SettingTen
0910: 74 73 28 29 2e 63 6f 75 6e 74 53 69 74 65 73 28 ts().countSites(
0920: 4e 2c 20 4d 29 29 3b 20 7d 0d 0a 69 6e 74 20 54 N, M)); }..int T
0930: 65 73 74 5f 28 43 61 73 65 5f 3c 32 3e 29 20 7b est_(Case_<2>) {
0940: 0d 0a 09 69 6e 74 20 4e 20 3d 20 36 3b 20 0d 0a ...int N = 6; ..
0950: 09 69 6e 74 20 4d 20 3d 20 38 3b 20 0d 0a 09 69 .int M = 8; ...i
0960: 6e 74 20 52 65 74 56 61 6c 20 3d 20 35 32 37 3b nt RetVal = 527;
0970: 20 0d 0a 09 72 65 74 75 72 6e 20 76 65 72 69 66 ...return verif
0980: 79 5f 63 61 73 65 28 52 65 74 56 61 6c 2c 20 53 y_case(RetVal, S
0990: 65 74 74 69 6e 67 54 65 6e 74 73 28 29 2e 63 6f ettingTents().co
09a0: 75 6e 74 53 69 74 65 73 28 4e 2c 20 4d 29 29 3b untSites(N, M));
09b0: 20 7d 0d 0a 0d 0a 74 65 6d 70 6c 61 74 65 3c 69 }....template<i
09c0: 6e 74 20 4e 3e 20 76 6f 69 64 20 52 75 6e 5f 28 nt N> void Run_(
09d0: 29 20 7b 20 63 65 72 72 20 3c 3c 20 22 54 65 73 ) { cerr << "Tes
09e0: 74 20 43 61 73 65 20 23 22 20 3c 3c 20 4e 20 3c t Case #" << N <
09f0: 3c 20 22 2e 2e 2e 22 20 3c 3c 20 66 6c 75 73 68 < "..." << flush
0a00: 3b 20 54 65 73 74 5f 28 43 61 73 65 5f 3c 4e 3e ; Test_(Case_<N>
0a10: 28 29 29 3b 20 52 75 6e 5f 3c 73 69 7a 65 6f 66 ()); Run_<sizeof
0a20: 28 54 65 73 74 5f 28 43 61 73 65 5f 3c 4e 2b 31 (Test_(Case_<N+1
0a30: 3e 28 29 29 29 3d 3d 31 20 3f 20 2d 31 20 3a 20 >()))==1 ? -1 :
0a40: 4e 2b 31 3e 28 29 3b 20 7d 0d 0a 74 65 6d 70 6c N+1>(); }..templ
0a50: 61 74 65 3c 3e 20 20 20 20 20 20 76 6f 69 64 20 ate<> void
0a60: 52 75 6e 5f 3c 2d 31 3e 28 29 20 7b 7d 0d 0a 69 Run_<-1>() {}..i
0a70: 6e 74 20 6d 61 69 6e 28 29 20 7b 20 52 75 6e 5f nt main() { Run_
0a80: 3c 30 3e 28 29 3b 20 7d 0d 0a 2f 2f 20 45 4e 44 <0>(); }..// END
0a90: 20 43 55 54 20 48 45 52 45 0d 0a 0d 0a CUT HERE....