Artifact e67bc0ae357b08b2c29f6e866adf0d7ca2f401dd:
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 3a 76 61 6c 28 30 29 ...mint():val(0)
01f0: 7b 7d 0d 0a 09 6d 69 6e 74 28 69 6e 74 20 20 20 {}...mint(int
0200: 20 20 20 78 29 3a 76 61 6c 28 78 25 4d 4f 44 56 x):val(x%MODV
0210: 41 4c 29 20 7b 7d 0d 0a 09 6d 69 6e 74 28 75 6e AL) {}...mint(un
0220: 73 69 67 6e 65 64 20 78 29 3a 76 61 6c 28 78 25 signed x):val(x%
0230: 4d 4f 44 56 41 4c 29 20 7b 7d 0d 0a 09 6d 69 6e MODVAL) {}...min
0240: 74 28 4c 4c 20 20 20 20 20 20 20 78 29 3a 76 61 t(LL x):va
0250: 6c 28 78 25 4d 4f 44 56 41 4c 29 20 7b 7d 0d 0a l(x%MODVAL) {}..
0260: 7d 3b 0d 0a 6d 69 6e 74 26 20 6f 70 65 72 61 74 };..mint& operat
0270: 6f 72 2b 3d 28 6d 69 6e 74 26 20 78 2c 20 6d 69 or+=(mint& x, mi
0280: 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78 nt y) { return x
0290: 20 3d 20 78 2e 76 61 6c 2b 79 2e 76 61 6c 3b 20 = x.val+y.val;
02a0: 7d 0d 0a 6d 69 6e 74 26 20 6f 70 65 72 61 74 6f }..mint& operato
02b0: 72 2a 3d 28 6d 69 6e 74 26 20 78 2c 20 6d 69 6e r*=(mint& x, min
02c0: 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78 20 t y) { return x
02d0: 3d 20 4c 4c 28 78 2e 76 61 6c 29 2a 79 2e 76 61 = LL(x.val)*y.va
02e0: 6c 3b 20 7d 0d 0a 6d 69 6e 74 20 6f 70 65 72 61 l; }..mint opera
02f0: 74 6f 72 2b 28 6d 69 6e 74 20 78 2c 20 6d 69 6e tor+(mint x, min
0300: 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78 2b t y) { return x+
0310: 3d 79 3b 20 7d 0d 0a 6d 69 6e 74 20 6f 70 65 72 =y; }..mint oper
0320: 61 74 6f 72 2a 28 6d 69 6e 74 20 78 2c 20 6d 69 ator*(mint x, mi
0330: 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78 nt y) { return x
0340: 2a 3d 79 3b 20 7d 0d 0a 0d 0a 69 6e 74 20 67 63 *=y; }....int gc
0350: 64 28 69 6e 74 20 61 2c 20 69 6e 74 20 62 29 0d d(int a, int b).
0360: 0a 7b 0d 0a 09 77 68 69 6c 65 28 61 29 0d 0a 09 .{...while(a)...
0370: 09 73 77 61 70 28 61 2c 20 62 25 3d 61 29 3b 0d .swap(a, b%=a);.
0380: 0a 09 72 65 74 75 72 6e 20 62 3b 0d 0a 7d 0d 0a ..return b;..}..
0390: 0d 0a 63 6c 61 73 73 20 52 65 66 6c 65 63 74 69 ..class Reflecti
03a0: 76 65 52 65 63 74 61 6e 67 6c 65 20 7b 20 70 75 veRectangle { pu
03b0: 62 6c 69 63 3a 0d 0a 09 69 6e 74 20 66 69 6e 64 blic:...int find
03c0: 53 75 6d 28 69 6e 74 20 41 2c 20 69 6e 74 20 42 Sum(int A, int B
03d0: 2c 20 69 6e 74 20 62 6f 75 6e 63 65 73 29 0d 0a , int bounces)..
03e0: 09 7b 0d 0a 09 09 69 66 28 62 6f 75 6e 63 65 73 .{....if(bounces
03f0: 25 32 20 3d 3d 20 31 29 0d 0a 09 09 09 72 65 74 %2 == 1).....ret
0400: 75 72 6e 20 30 3b 0d 0a 09 09 69 66 28 62 6f 75 urn 0;....if(bou
0410: 6e 63 65 73 20 3d 3d 20 30 29 0d 0a 09 09 09 72 nces == 0).....r
0420: 65 74 75 72 6e 20 28 6d 69 6e 74 28 41 29 2a 41 eturn (mint(A)*A
0430: 2b 6d 69 6e 74 28 42 29 2a 42 29 2e 76 61 6c 3b +mint(B)*B).val;
0440: 0d 0a 0d 0a 09 09 6d 69 6e 74 20 73 68 3b 0d 0a ......mint sh;..
0450: 09 09 66 6f 72 28 69 6e 74 20 68 3d 31 3b 20 68 ..for(int h=1; h
0460: 3c 62 6f 75 6e 63 65 73 2b 32 3b 20 68 2b 3d 32 <bounces+2; h+=2
0470: 29 20 7b 0d 0a 09 09 09 69 66 28 67 63 64 28 68 ) {.....if(gcd(h
0480: 2c 20 62 6f 75 6e 63 65 73 2b 32 29 20 3d 3d 20 , bounces+2) ==
0490: 31 29 0d 0a 09 09 09 09 73 68 20 2b 3d 20 4c 4c 1)......sh += LL
04a0: 28 68 29 2a 68 3b 0d 0a 09 09 7d 0d 0a 09 09 2f (h)*h;....}..../
04b0: 2f 20 73 6f 6c 75 74 69 6f 6e 3a 20 53 69 67 6d / solution: Sigm
04c0: 61 28 68 2a 68 29 20 2d 20 53 69 67 6d 61 5f 64 a(h*h) - Sigma_d
04d0: 69 76 69 73 6f 72 73 28 68 2a 68 29 20 3f 0d 0a ivisors(h*h) ?..
04e0: 09 09 2f 2f 20 62 75 74 20 6e 6f 20 74 69 6d 65 ..// but no time
04f0: 20 74 6f 20 63 6f 64 65 2e 2e 2e 2e 2e 2e 0d 0a to code........
0500: 0d 0a 09 09 72 65 74 75 72 6e 20 28 73 68 2a 41 ....return (sh*A
0510: 2a 41 2b 73 68 2a 42 2a 42 29 2e 76 61 6c 3b 0d *A+sh*B*B).val;.
0520: 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 42 45 ..}..};....// BE
0530: 47 49 4e 20 43 55 54 20 48 45 52 45 0d 0a 23 69 GIN CUT HERE..#i
0540: 6e 63 6c 75 64 65 20 3c 63 74 69 6d 65 3e 0d 0a nclude <ctime>..
0550: 64 6f 75 62 6c 65 20 73 74 61 72 74 5f 74 69 6d double start_tim
0560: 65 3b 20 73 74 72 69 6e 67 20 74 69 6d 65 72 28 e; string timer(
0570: 29 0d 0a 20 7b 20 6f 73 74 72 69 6e 67 73 74 72 ).. { ostringstr
0580: 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 22 20 eam os; os << "
0590: 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f 63 6b (" << int((clock
05a0: 28 29 2d 73 74 61 72 74 5f 74 69 6d 65 29 2f 43 ()-start_time)/C
05b0: 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a 31 30 LOCKS_PER_SEC*10
05c0: 30 30 29 20 3c 3c 20 22 20 6d 73 65 63 29 22 3b 00) << " msec)";
05d0: 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 28 29 return os.str()
05e0: 3b 20 7d 0d 0a 74 65 6d 70 6c 61 74 65 3c 74 79 ; }..template<ty
05f0: 70 65 6e 61 6d 65 20 54 3e 20 6f 73 74 72 65 61 pename T> ostrea
0600: 6d 26 20 6f 70 65 72 61 74 6f 72 3c 3c 28 6f 73 m& operator<<(os
0610: 74 72 65 61 6d 26 20 6f 73 2c 20 63 6f 6e 73 74 tream& os, const
0620: 20 76 65 63 74 6f 72 3c 54 3e 26 20 76 29 0d 0a vector<T>& v)..
0630: 20 7b 20 6f 73 20 3c 3c 20 22 7b 20 22 3b 0d 0a { os << "{ ";..
0640: 20 20 20 66 6f 72 28 74 79 70 65 6e 61 6d 65 20 for(typename
0650: 76 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f 6e 73 74 vector<T>::const
0660: 5f 69 74 65 72 61 74 6f 72 20 69 74 3d 76 2e 62 _iterator it=v.b
0670: 65 67 69 6e 28 29 3b 20 69 74 21 3d 76 2e 65 6e egin(); it!=v.en
0680: 64 28 29 3b 20 2b 2b 69 74 29 0d 0a 20 20 20 6f d(); ++it).. o
0690: 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 74 s << '\"' << *it
06a0: 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 28 69 74 2b << '\"' << (it+
06b0: 31 3d 3d 76 2e 65 6e 64 28 29 20 3f 20 22 22 20 1==v.end() ? ""
06c0: 3a 20 22 2c 20 22 29 3b 20 6f 73 20 3c 3c 20 22 : ", "); os << "
06d0: 20 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 3b 20 }"; return os;
06e0: 7d 0d 0a 76 6f 69 64 20 76 65 72 69 66 79 5f 63 }..void verify_c
06f0: 61 73 65 28 63 6f 6e 73 74 20 69 6e 74 26 20 45 ase(const int& E
0700: 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 20 69 xpected, const i
0710: 6e 74 26 20 52 65 63 65 69 76 65 64 29 20 7b 0d nt& Received) {.
0720: 0a 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28 45 78 70 . bool ok = (Exp
0730: 65 63 74 65 64 20 3d 3d 20 52 65 63 65 69 76 65 ected == Receive
0740: 64 29 3b 0d 0a 20 69 66 28 6f 6b 29 20 63 65 72 d);.. if(ok) cer
0750: 72 20 3c 3c 20 22 50 41 53 53 45 44 22 20 3c 3c r << "PASSED" <<
0760: 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c timer() << endl
0770: 3b 20 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c ; else { cerr <
0780: 3c 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 69 < "FAILED" << ti
0790: 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 0d 0a mer() << endl;..
07a0: 20 63 65 72 72 20 3c 3c 20 22 5c 74 6f 3a 20 5c cerr << "\to: \
07b0: 22 22 20 3c 3c 20 45 78 70 65 63 74 65 64 20 3c "" << Expected <
07c0: 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 20 3c < '\"' << endl <
07d0: 3c 20 22 5c 74 78 3a 20 5c 22 22 20 3c 3c 20 52 < "\tx: \"" << R
07e0: 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 20 eceived << '\"'
07f0: 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a 23 64 << endl; } }..#d
0800: 65 66 69 6e 65 20 43 41 53 45 28 4e 29 20 7b 63 efine CASE(N) {c
0810: 65 72 72 20 3c 3c 20 22 54 65 73 74 20 43 61 73 err << "Test Cas
0820: 65 20 23 22 20 3c 3c 20 4e 20 3c 3c 20 22 2e 2e e #" << N << "..
0830: 2e 22 20 3c 3c 20 66 6c 75 73 68 3b 20 73 74 61 ." << flush; sta
0840: 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b 28 29 3b rt_time=clock();
0850: 0d 0a 23 64 65 66 69 6e 65 20 45 4e 44 09 20 76 ..#define END. v
0860: 65 72 69 66 79 5f 63 61 73 65 28 5f 2c 20 52 65 erify_case(_, Re
0870: 66 6c 65 63 74 69 76 65 52 65 63 74 61 6e 67 6c flectiveRectangl
0880: 65 28 29 2e 66 69 6e 64 53 75 6d 28 73 69 64 65 e().findSum(side
0890: 41 2c 20 73 69 64 65 42 2c 20 62 6f 75 6e 63 65 A, sideB, bounce
08a0: 73 29 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 s));}..int main(
08b0: 29 7b 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 ){....CASE(0)...
08c0: 69 6e 74 20 73 69 64 65 41 20 3d 20 33 3b 20 0d int sideA = 3; .
08d0: 0a 09 69 6e 74 20 73 69 64 65 42 20 3d 20 34 3b ..int sideB = 4;
08e0: 20 0d 0a 09 69 6e 74 20 62 6f 75 6e 63 65 73 20 ...int bounces
08f0: 3d 20 30 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 = 0; ...int _ =
0900: 32 35 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 25; ..END..CASE(
0910: 31 29 0d 0a 09 69 6e 74 20 73 69 64 65 41 20 3d 1)...int sideA =
0920: 20 33 3b 20 0d 0a 09 69 6e 74 20 73 69 64 65 42 3; ...int sideB
0930: 20 3d 20 33 3b 20 0d 0a 09 69 6e 74 20 62 6f 75 = 3; ...int bou
0940: 6e 63 65 73 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 nces = 2; ...int
0950: 20 5f 20 3d 20 31 38 30 3b 20 0d 0a 45 4e 44 0d _ = 180; ..END.
0960: 0a 43 41 53 45 28 32 29 0d 0a 09 69 6e 74 20 73 .CASE(2)...int s
0970: 69 64 65 41 20 3d 20 31 33 3b 20 0d 0a 09 69 6e ideA = 13; ...in
0980: 74 20 73 69 64 65 42 20 3d 20 31 37 3b 20 0d 0a t sideB = 17; ..
0990: 09 69 6e 74 20 62 6f 75 6e 63 65 73 20 3d 20 31 .int bounces = 1
09a0: 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 30 3b 20 ; ...int _ = 0;
09b0: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 33 29 0d 0a ..END..CASE(3)..
09c0: 09 69 6e 74 20 73 69 64 65 41 20 3d 20 35 39 33 .int sideA = 593
09d0: 32 35 3b 20 0d 0a 09 69 6e 74 20 73 69 64 65 42 25; ...int sideB
09e0: 20 3d 20 33 31 37 38 35 3b 20 0d 0a 09 69 6e 74 = 31785; ...int
09f0: 20 62 6f 75 6e 63 65 73 20 3d 20 32 36 32 31 34 bounces = 26214
0a00: 32 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 34 38 2; ...int _ = 48
0a10: 30 33 32 38 35 30 3b 20 0d 0a 45 4e 44 0d 0a 43 032850; ..END..C
0a20: 41 53 45 28 34 29 0d 0a 09 69 6e 74 20 73 69 64 ASE(4)...int sid
0a30: 65 41 20 3d 20 31 30 30 30 30 30 30 3b 20 0d 0a eA = 1000000; ..
0a40: 09 69 6e 74 20 73 69 64 65 42 20 3d 20 31 30 30 .int sideB = 100
0a50: 30 30 30 30 3b 20 0d 0a 09 69 6e 74 20 62 6f 75 0000; ...int bou
0a60: 6e 63 65 73 20 3d 20 31 30 30 30 30 30 30 30 30 nces = 100000000
0a70: 30 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 31 34 0; ...int _ = 14
0a80: 35 39 37 32 31 31 30 3b 20 0d 0a 45 4e 44 0d 0a 5972110; ..END..
0a90: 2f 2a 0d 0a 43 41 53 45 28 35 29 0d 0a 09 69 6e /*..CASE(5)...in
0aa0: 74 20 73 69 64 65 41 20 3d 20 3b 20 0d 0a 09 69 t sideA = ; ...i
0ab0: 6e 74 20 73 69 64 65 42 20 3d 20 3b 20 0d 0a 09 nt sideB = ; ...
0ac0: 69 6e 74 20 62 6f 75 6e 63 65 73 20 3d 20 3b 20 int bounces = ;
0ad0: 0d 0a 09 69 6e 74 20 5f 20 3d 20 3b 20 0d 0a 45 ...int _ = ; ..E
0ae0: 4e 44 0d 0a 43 41 53 45 28 36 29 0d 0a 09 69 6e ND..CASE(6)...in
0af0: 74 20 73 69 64 65 41 20 3d 20 3b 20 0d 0a 09 69 t sideA = ; ...i
0b00: 6e 74 20 73 69 64 65 42 20 3d 20 3b 20 0d 0a 09 nt sideB = ; ...
0b10: 69 6e 74 20 62 6f 75 6e 63 65 73 20 3d 20 3b 20 int bounces = ;
0b20: 0d 0a 09 69 6e 74 20 5f 20 3d 20 3b 20 0d 0a 45 ...int _ = ; ..E
0b30: 4e 44 0d 0a 2a 2f 0d 0a 7d 0d 0a 2f 2f 20 45 4e ND..*/..}..// EN
0b40: 44 20 43 55 54 20 48 45 52 45 0d 0a D CUT HERE..