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 23 69 66 64 65 <cstring>..#ifde
0150: 66 20 5f 5f 47 4e 55 43 5f 5f 0d 0a 23 69 6e 63 f __GNUC__..#inc
0160: 6c 75 64 65 20 3c 65 78 74 2f 68 61 73 68 5f 6d lude <ext/hash_m
0170: 61 70 3e 0d 0a 23 64 65 66 69 6e 65 20 75 6e 6f ap>..#define uno
0180: 72 64 65 72 65 64 5f 6d 61 70 20 5f 5f 67 6e 75 rdered_map __gnu
0190: 5f 63 78 78 3a 3a 68 61 73 68 5f 6d 61 70 0d 0a _cxx::hash_map..
01a0: 23 65 6c 73 65 0d 0a 23 69 6e 63 6c 75 64 65 20 #else..#include
01b0: 3c 75 6e 6f 72 64 65 72 65 64 5f 6d 61 70 3e 0d <unordered_map>.
01c0: 0a 23 65 6e 64 69 66 0d 0a 75 73 69 6e 67 20 6e .#endif..using n
01d0: 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d 0a 74 amespace std;..t
01e0: 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67 ypedef long long
01f0: 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 63 6f LL;..typedef co
0200: 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 43 4d mplex<double> CM
0210: 50 3b 0d 0a 0d 0a 63 6c 61 73 73 20 52 61 6e 64 P;....class Rand
0220: 6f 6d 43 6f 6c 6f 72 69 6e 67 20 7b 20 70 75 62 omColoring { pub
0230: 6c 69 63 3a 0d 0a 09 64 6f 75 62 6c 65 20 67 65 lic:...double ge
0240: 74 50 72 6f 62 61 62 69 6c 69 74 79 28 69 6e 74 tProbability(int
0250: 20 4e 2c 20 69 6e 74 20 6d 61 78 52 2c 20 69 6e N, int maxR, in
0260: 74 20 6d 61 78 47 2c 20 69 6e 74 20 6d 61 78 42 t maxG, int maxB
0270: 2c 20 69 6e 74 20 73 74 61 72 74 52 2c 20 69 6e , int startR, in
0280: 74 20 73 74 61 72 74 47 2c 20 69 6e 74 20 73 74 t startG, int st
0290: 61 72 74 42 2c 20 69 6e 74 20 64 31 2c 20 69 6e artB, int d1, in
02a0: 74 20 64 32 29 0d 0a 09 7b 0d 0a 09 09 63 6f 6e t d2)...{....con
02b0: 73 74 20 69 6e 74 20 64 78 20 3d 20 64 31 2d 31 st int dx = d1-1
02c0: 3b 0d 0a 0d 0a 09 09 76 65 63 74 6f 72 3c 20 76 ;......vector< v
02d0: 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c 64 6f ector< vector<do
02e0: 75 62 6c 65 3e 20 3e 20 3e 20 70 28 6d 61 78 52 uble> > > p(maxR
02f0: 2c 20 76 65 63 74 6f 72 3c 76 65 63 74 6f 72 3c , vector<vector<
0300: 64 6f 75 62 6c 65 3e 20 3e 28 6d 61 78 47 2c 20 double> >(maxG,
0310: 76 65 63 74 6f 72 3c 64 6f 75 62 6c 65 3e 28 6d vector<double>(m
0320: 61 78 42 29 29 29 3b 0d 0a 09 09 70 5b 73 74 61 axB)));....p[sta
0330: 72 74 52 5d 5b 73 74 61 72 74 47 5d 5b 73 74 61 rtR][startG][sta
0340: 72 74 42 5d 20 3d 20 31 2e 30 3b 0d 0a 0d 0a 09 rtB] = 1.0;.....
0350: 09 66 6f 72 28 69 6e 74 20 69 3d 31 3b 20 69 3c .for(int i=1; i<
0360: 4e 3b 20 2b 2b 69 29 0d 0a 09 09 7b 0d 0a 09 09 N; ++i)....{....
0370: 09 66 6f 72 28 69 6e 74 20 72 3d 30 3b 20 72 3c .for(int r=0; r<
0380: 6d 61 78 52 3b 20 2b 2b 72 29 0d 0a 09 09 09 66 maxR; ++r).....f
0390: 6f 72 28 69 6e 74 20 67 3d 30 3b 20 67 3c 6d 61 or(int g=0; g<ma
03a0: 78 47 3b 20 2b 2b 67 29 0d 0a 09 09 09 66 6f 72 xG; ++g).....for
03b0: 28 69 6e 74 20 62 3d 30 3b 20 62 3c 6d 61 78 42 (int b=0; b<maxB
03c0: 3b 20 2b 2b 62 29 20 7b 0d 0a 09 09 09 09 69 6e ; ++b) {......in
03d0: 74 20 63 20 3d 20 63 6e 74 28 6d 61 78 52 2c 20 t c = cnt(maxR,
03e0: 6d 61 78 47 2c 20 6d 61 78 42 2c 20 72 2d 64 32 maxG, maxB, r-d2
03f0: 2c 20 72 2b 64 32 2c 20 67 2d 64 32 2c 20 67 2b , r+d2, g-d2, g+
0400: 64 32 2c 20 62 2d 64 32 2c 20 62 2b 64 32 29 0d d2, b-d2, b+d2).
0410: 0a 09 09 09 09 2d 20 28 64 31 3d 3d 30 20 3f 20 .....- (d1==0 ?
0420: 30 20 3a 20 63 6e 74 28 6d 61 78 52 2c 20 6d 61 0 : cnt(maxR, ma
0430: 78 47 2c 20 6d 61 78 42 2c 20 72 2d 64 78 2c 20 xG, maxB, r-dx,
0440: 72 2b 64 78 2c 20 67 2d 64 78 2c 20 67 2b 64 78 r+dx, g-dx, g+dx
0450: 2c 20 62 2d 64 78 2c 20 62 2b 64 78 29 29 3b 0d , b-dx, b+dx));.
0460: 0a 09 09 09 09 69 66 28 20 63 20 29 0d 0a 09 09 .....if( c )....
0470: 09 09 09 70 5b 72 5d 5b 67 5d 5b 62 5d 20 2f 3d ...p[r][g][b] /=
0480: 20 63 3b 0d 0a 09 09 09 09 65 6c 73 65 0d 0a 09 c;......else...
0490: 09 09 09 09 70 5b 72 5d 5b 67 5d 5b 62 5d 20 3d ....p[r][g][b] =
04a0: 20 30 3b 0d 0a 09 09 09 7d 0d 0a 0d 0a 09 09 09 0;.....}.......
04b0: 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c 20 vector< vector<
04c0: 76 65 63 74 6f 72 3c 64 6f 75 62 6c 65 3e 20 3e vector<double> >
04d0: 20 3e 20 6c 6f 77 20 3d 20 70 3b 0d 0a 09 09 09 > low = p;.....
04e0: 66 6f 72 28 69 6e 74 20 72 67 62 3d 30 3b 20 72 for(int rgb=0; r
04f0: 67 62 3c 3d 6d 61 78 52 2b 6d 61 78 47 2b 6d 61 gb<=maxR+maxG+ma
0500: 78 42 2d 33 3b 20 2b 2b 72 67 62 29 0d 0a 09 09 xB-3; ++rgb)....
0510: 09 66 6f 72 28 69 6e 74 20 72 3d 30 3b 20 72 3c .for(int r=0; r<
0520: 6d 61 78 52 3b 20 2b 2b 72 29 0d 0a 09 09 09 66 maxR; ++r).....f
0530: 6f 72 28 69 6e 74 20 67 3d 30 3b 20 67 3c 6d 61 or(int g=0; g<ma
0540: 78 47 3b 20 2b 2b 67 29 0d 0a 09 09 09 7b 0d 0a xG; ++g).....{..
0550: 09 09 09 09 69 6e 74 20 62 20 3d 20 72 67 62 2d ....int b = rgb-
0560: 72 2d 67 3b 0d 0a 09 09 09 09 69 66 28 20 30 3c r-g;......if( 0<
0570: 3d 62 20 26 26 20 62 3c 6d 61 78 42 20 29 20 7b =b && b<maxB ) {
0580: 0d 0a 09 09 09 09 09 6c 6f 77 5b 72 5d 5b 67 5d .......low[r][g]
0590: 5b 62 5d 20 3d 0d 0a 09 09 09 09 09 09 70 5b 72 [b] =........p[r
05a0: 5d 5b 67 5d 5b 62 5d 0d 0a 09 09 09 09 09 09 2b ][g][b]........+
05b0: 20 28 72 20 3f 20 6c 6f 77 5b 72 2d 31 5d 5b 67 (r ? low[r-1][g
05c0: 5d 5b 62 5d 20 3a 20 30 29 0d 0a 09 09 09 09 09 ][b] : 0).......
05d0: 09 2b 20 28 67 20 3f 20 6c 6f 77 5b 72 5d 5b 67 .+ (g ? low[r][g
05e0: 2d 31 5d 5b 62 5d 20 3a 20 30 29 0d 0a 09 09 09 -1][b] : 0).....
05f0: 09 09 09 2b 20 28 62 20 3f 20 6c 6f 77 5b 72 5d ...+ (b ? low[r]
0600: 5b 67 5d 5b 62 2d 31 5d 20 3a 20 30 29 0d 0a 09 [g][b-1] : 0)...
0610: 09 09 09 09 09 2d 20 28 72 26 26 67 20 3f 20 6c .....- (r&&g ? l
0620: 6f 77 5b 72 2d 31 5d 5b 67 2d 31 5d 5b 62 5d 20 ow[r-1][g-1][b]
0630: 3a 20 30 29 0d 0a 09 09 09 09 09 09 2d 20 28 67 : 0)........- (g
0640: 26 26 62 20 3f 20 6c 6f 77 5b 72 5d 5b 67 2d 31 &&b ? low[r][g-1
0650: 5d 5b 62 2d 31 5d 20 3a 20 30 29 0d 0a 09 09 09 ][b-1] : 0).....
0660: 09 09 09 2d 20 28 62 26 26 72 20 3f 20 6c 6f 77 ...- (b&&r ? low
0670: 5b 72 2d 31 5d 5b 67 5d 5b 62 2d 31 5d 20 3a 20 [r-1][g][b-1] :
0680: 30 29 0d 0a 09 09 09 09 09 09 2b 20 28 72 26 26 0)........+ (r&&
0690: 67 26 26 62 20 3f 20 6c 6f 77 5b 72 2d 31 5d 5b g&&b ? low[r-1][
06a0: 67 2d 31 5d 5b 62 2d 31 5d 20 3a 20 30 29 3b 0d g-1][b-1] : 0);.
06b0: 0a 09 09 09 09 7d 0d 0a 09 09 09 7d 0d 0a 0d 0a .....}.....}....
06c0: 09 09 09 76 65 63 74 6f 72 3c 20 76 65 63 74 6f ...vector< vecto
06d0: 72 3c 20 76 65 63 74 6f 72 3c 64 6f 75 62 6c 65 r< vector<double
06e0: 3e 20 3e 20 3e 20 71 20 3d 20 70 3b 0d 0a 09 09 > > > q = p;....
06f0: 09 66 6f 72 28 69 6e 74 20 72 3d 30 3b 20 72 3c .for(int r=0; r<
0700: 6d 61 78 52 3b 20 2b 2b 72 29 0d 0a 09 09 09 66 maxR; ++r).....f
0710: 6f 72 28 69 6e 74 20 67 3d 30 3b 20 67 3c 6d 61 or(int g=0; g<ma
0720: 78 47 3b 20 2b 2b 67 29 0d 0a 09 09 09 66 6f 72 xG; ++g).....for
0730: 28 69 6e 74 20 62 3d 30 3b 20 62 3c 6d 61 78 42 (int b=0; b<maxB
0740: 3b 20 2b 2b 62 29 0d 0a 09 09 09 7b 0d 0a 09 09 ; ++b).....{....
0750: 09 09 71 5b 72 5d 5b 67 5d 5b 62 5d 20 3d 0d 0a ..q[r][g][b] =..
0760: 09 09 09 09 09 20 20 7a 6f 6e 65 28 6c 6f 77 2c ..... zone(low,
0770: 20 72 2d 64 32 2c 20 72 2b 64 32 2c 20 67 2d 64 r-d2, r+d2, g-d
0780: 32 2c 20 67 2b 64 32 2c 20 62 2d 64 32 2c 20 62 2, g+d2, b-d2, b
0790: 2b 64 32 29 0d 0a 09 09 09 09 09 2d 20 28 64 31 +d2).......- (d1
07a0: 3d 3d 30 20 3f 20 30 20 3a 20 7a 6f 6e 65 28 6c ==0 ? 0 : zone(l
07b0: 6f 77 2c 20 72 2d 64 78 2c 20 72 2b 64 78 2c 20 ow, r-dx, r+dx,
07c0: 67 2d 64 78 2c 20 67 2b 64 78 2c 20 62 2d 64 78 g-dx, g+dx, b-dx
07d0: 2c 20 62 2b 64 78 29 29 3b 0d 0a 09 09 09 7d 0d , b+dx));.....}.
07e0: 0a 09 09 09 70 2e 73 77 61 70 28 71 29 3b 0d 0a ....p.swap(q);..
07f0: 09 09 7d 0d 0a 0d 0a 09 09 76 65 63 74 6f 72 3c ..}......vector<
0800: 20 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c vector< vector<
0810: 64 6f 75 62 6c 65 3e 20 3e 20 3e 20 6c 6f 77 20 double> > > low
0820: 3d 20 70 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 = p;....for(int
0830: 72 67 62 3d 30 3b 20 72 67 62 3c 3d 6d 61 78 52 rgb=0; rgb<=maxR
0840: 2b 6d 61 78 47 2b 6d 61 78 42 2d 33 3b 20 2b 2b +maxG+maxB-3; ++
0850: 72 67 62 29 0d 0a 09 09 66 6f 72 28 69 6e 74 20 rgb)....for(int
0860: 72 3d 30 3b 20 72 3c 6d 61 78 52 3b 20 2b 2b 72 r=0; r<maxR; ++r
0870: 29 0d 0a 09 09 66 6f 72 28 69 6e 74 20 67 3d 30 )....for(int g=0
0880: 3b 20 67 3c 6d 61 78 47 3b 20 2b 2b 67 29 0d 0a ; g<maxG; ++g)..
0890: 09 09 7b 0d 0a 09 09 09 69 6e 74 20 62 20 3d 20 ..{.....int b =
08a0: 72 67 62 2d 72 2d 67 3b 0d 0a 09 09 09 69 66 28 rgb-r-g;.....if(
08b0: 20 30 3c 3d 62 20 26 26 20 62 3c 6d 61 78 42 20 0<=b && b<maxB
08c0: 29 20 7b 0d 0a 09 09 09 09 6c 6f 77 5b 72 5d 5b ) {......low[r][
08d0: 67 5d 5b 62 5d 20 3d 0d 0a 09 09 09 09 09 70 5b g][b] =.......p[
08e0: 72 5d 5b 67 5d 5b 62 5d 0d 0a 09 09 09 09 09 2b r][g][b].......+
08f0: 20 28 72 20 3f 20 6c 6f 77 5b 72 2d 31 5d 5b 67 (r ? low[r-1][g
0900: 5d 5b 62 5d 20 3a 20 30 29 0d 0a 09 09 09 09 09 ][b] : 0).......
0910: 2b 20 28 67 20 3f 20 6c 6f 77 5b 72 5d 5b 67 2d + (g ? low[r][g-
0920: 31 5d 5b 62 5d 20 3a 20 30 29 0d 0a 09 09 09 09 1][b] : 0)......
0930: 09 2b 20 28 62 20 3f 20 6c 6f 77 5b 72 5d 5b 67 .+ (b ? low[r][g
0940: 5d 5b 62 2d 31 5d 20 3a 20 30 29 0d 0a 09 09 09 ][b-1] : 0).....
0950: 09 09 2d 20 28 72 26 26 67 20 3f 20 6c 6f 77 5b ..- (r&&g ? low[
0960: 72 2d 31 5d 5b 67 2d 31 5d 5b 62 5d 20 3a 20 30 r-1][g-1][b] : 0
0970: 29 0d 0a 09 09 09 09 09 2d 20 28 67 26 26 62 20 ).......- (g&&b
0980: 3f 20 6c 6f 77 5b 72 5d 5b 67 2d 31 5d 5b 62 2d ? low[r][g-1][b-
0990: 31 5d 20 3a 20 30 29 0d 0a 09 09 09 09 09 2d 20 1] : 0).......-
09a0: 28 62 26 26 72 20 3f 20 6c 6f 77 5b 72 2d 31 5d (b&&r ? low[r-1]
09b0: 5b 67 5d 5b 62 2d 31 5d 20 3a 20 30 29 0d 0a 09 [g][b-1] : 0)...
09c0: 09 09 09 09 2b 20 28 72 26 26 67 26 26 62 20 3f ....+ (r&&g&&b ?
09d0: 20 6c 6f 77 5b 72 2d 31 5d 5b 67 2d 31 5d 5b 62 low[r-1][g-1][b
09e0: 2d 31 5d 20 3a 20 30 29 3b 0d 0a 09 09 09 7d 0d -1] : 0);.....}.
09f0: 0a 09 09 7d 0d 0a 0d 0a 09 09 72 65 74 75 72 6e ...}......return
0a00: 20 31 2e 30 20 2d 20 28 0d 0a 09 09 09 7a 6f 6e 1.0 - (.....zon
0a10: 65 28 6c 6f 77 2c 20 73 74 61 72 74 52 2d 64 32 e(low, startR-d2
0a20: 2c 20 73 74 61 72 74 52 2b 64 32 2c 20 73 74 61 , startR+d2, sta
0a30: 72 74 47 2d 64 32 2c 20 73 74 61 72 74 47 2b 64 rtG-d2, startG+d
0a40: 32 2c 20 73 74 61 72 74 42 2d 64 32 2c 20 73 74 2, startB-d2, st
0a50: 61 72 74 42 2b 64 32 29 0d 0a 09 09 09 2d 20 28 artB+d2).....- (
0a60: 64 31 3d 3d 30 20 3f 20 30 20 3a 20 7a 6f 6e 65 d1==0 ? 0 : zone
0a70: 28 6c 6f 77 2c 20 73 74 61 72 74 52 2d 64 78 2c (low, startR-dx,
0a80: 20 73 74 61 72 74 52 2b 64 78 2c 20 73 74 61 72 startR+dx, star
0a90: 74 47 2d 64 78 2c 20 73 74 61 72 74 47 2b 64 78 tG-dx, startG+dx
0aa0: 2c 20 73 74 61 72 74 42 2d 64 78 2c 20 73 74 61 , startB-dx, sta
0ab0: 72 74 42 2b 64 78 29 29 0d 0a 09 09 29 3b 0d 0a rtB+dx))....);..
0ac0: 09 7d 0d 0a 0d 0a 09 64 6f 75 62 6c 65 20 7a 6f .}.....double zo
0ad0: 6e 65 28 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c ne(const vector<
0ae0: 20 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c vector< vector<
0af0: 64 6f 75 62 6c 65 3e 20 3e 20 3e 26 20 6c 6f 77 double> > >& low
0b00: 2c 0d 0a 09 09 69 6e 74 20 72 2c 20 69 6e 74 20 ,....int r, int
0b10: 52 2c 20 69 6e 74 20 67 2c 20 69 6e 74 20 47 2c R, int g, int G,
0b20: 20 69 6e 74 20 62 2c 20 69 6e 74 20 42 29 20 7b int b, int B) {
0b30: 0d 0a 09 09 72 20 3d 20 6d 61 78 28 30 2c 20 6d ....r = max(0, m
0b40: 69 6e 3c 69 6e 74 3e 28 6c 6f 77 2e 73 69 7a 65 in<int>(low.size
0b50: 28 29 2d 31 2c 20 72 29 29 3b 0d 0a 09 09 52 20 ()-1, r));....R
0b60: 3d 20 6d 61 78 28 30 2c 20 6d 69 6e 3c 69 6e 74 = max(0, min<int
0b70: 3e 28 6c 6f 77 2e 73 69 7a 65 28 29 2d 31 2c 20 >(low.size()-1,
0b80: 52 29 29 3b 0d 0a 09 09 67 20 3d 20 6d 61 78 28 R));....g = max(
0b90: 30 2c 20 6d 69 6e 3c 69 6e 74 3e 28 6c 6f 77 5b 0, min<int>(low[
0ba0: 30 5d 2e 73 69 7a 65 28 29 2d 31 2c 20 67 29 29 0].size()-1, g))
0bb0: 3b 0d 0a 09 09 47 20 3d 20 6d 61 78 28 30 2c 20 ;....G = max(0,
0bc0: 6d 69 6e 3c 69 6e 74 3e 28 6c 6f 77 5b 30 5d 2e min<int>(low[0].
0bd0: 73 69 7a 65 28 29 2d 31 2c 20 47 29 29 3b 0d 0a size()-1, G));..
0be0: 09 09 62 20 3d 20 6d 61 78 28 30 2c 20 6d 69 6e ..b = max(0, min
0bf0: 3c 69 6e 74 3e 28 6c 6f 77 5b 30 5d 5b 30 5d 2e <int>(low[0][0].
0c00: 73 69 7a 65 28 29 2d 31 2c 20 62 29 29 3b 0d 0a size()-1, b));..
0c10: 09 09 42 20 3d 20 6d 61 78 28 30 2c 20 6d 69 6e ..B = max(0, min
0c20: 3c 69 6e 74 3e 28 6c 6f 77 5b 30 5d 5b 30 5d 2e <int>(low[0][0].
0c30: 73 69 7a 65 28 29 2d 31 2c 20 42 29 29 3b 0d 0a size()-1, B));..
0c40: 09 09 72 65 74 75 72 6e 20 6c 6f 77 5b 52 5d 5b ..return low[R][
0c50: 47 5d 5b 42 5d 0d 0a 09 09 09 09 2d 20 28 72 20 G][B]......- (r
0c60: 3f 20 6c 6f 77 5b 72 2d 31 5d 5b 47 5d 5b 42 5d ? low[r-1][G][B]
0c70: 20 3a 20 30 29 0d 0a 09 09 09 09 2d 20 28 67 20 : 0)......- (g
0c80: 3f 20 6c 6f 77 5b 52 5d 5b 67 2d 31 5d 5b 42 5d ? low[R][g-1][B]
0c90: 20 3a 20 30 29 0d 0a 09 09 09 09 2d 20 28 62 20 : 0)......- (b
0ca0: 3f 20 6c 6f 77 5b 52 5d 5b 47 5d 5b 62 2d 31 5d ? low[R][G][b-1]
0cb0: 20 3a 20 30 29 0d 0a 09 09 09 09 2b 20 28 72 26 : 0)......+ (r&
0cc0: 26 67 20 3f 20 6c 6f 77 5b 72 2d 31 5d 5b 67 2d &g ? low[r-1][g-
0cd0: 31 5d 5b 42 5d 20 3a 20 30 29 0d 0a 09 09 09 09 1][B] : 0)......
0ce0: 2b 20 28 67 26 26 62 20 3f 20 6c 6f 77 5b 52 5d + (g&&b ? low[R]
0cf0: 5b 67 2d 31 5d 5b 62 2d 31 5d 20 3a 20 30 29 0d [g-1][b-1] : 0).
0d00: 0a 09 09 09 09 2b 20 28 62 26 26 72 20 3f 20 6c .....+ (b&&r ? l
0d10: 6f 77 5b 72 2d 31 5d 5b 47 5d 5b 62 2d 31 5d 20 ow[r-1][G][b-1]
0d20: 3a 20 30 29 0d 0a 09 09 09 09 2d 20 28 72 26 26 : 0)......- (r&&
0d30: 67 26 26 62 20 3f 20 6c 6f 77 5b 72 2d 31 5d 5b g&&b ? low[r-1][
0d40: 67 2d 31 5d 5b 62 2d 31 5d 20 3a 20 30 29 3b 0d g-1][b-1] : 0);.
0d50: 0a 09 7d 0d 0a 0d 0a 09 69 6e 74 20 63 6e 74 28 ..}.....int cnt(
0d60: 69 6e 74 20 6d 61 78 52 2c 20 69 6e 74 20 6d 61 int maxR, int ma
0d70: 78 47 2c 20 69 6e 74 20 6d 61 78 42 2c 0d 0a 09 xG, int maxB,...
0d80: 09 69 6e 74 20 72 2c 20 69 6e 74 20 52 2c 20 69 .int r, int R, i
0d90: 6e 74 20 67 2c 20 69 6e 74 20 47 2c 20 69 6e 74 nt g, int G, int
0da0: 20 62 2c 20 69 6e 74 20 42 29 20 7b 0d 0a 09 09 b, int B) {....
0db0: 72 20 3d 20 6d 61 78 28 30 2c 20 6d 69 6e 3c 69 r = max(0, min<i
0dc0: 6e 74 3e 28 6d 61 78 52 2d 31 2c 20 72 29 29 3b nt>(maxR-1, r));
0dd0: 0d 0a 09 09 52 20 3d 20 6d 61 78 28 30 2c 20 6d ....R = max(0, m
0de0: 69 6e 3c 69 6e 74 3e 28 6d 61 78 52 2d 31 2c 20 in<int>(maxR-1,
0df0: 52 29 29 3b 0d 0a 09 09 67 20 3d 20 6d 61 78 28 R));....g = max(
0e00: 30 2c 20 6d 69 6e 3c 69 6e 74 3e 28 6d 61 78 47 0, min<int>(maxG
0e10: 2d 31 2c 20 67 29 29 3b 0d 0a 09 09 47 20 3d 20 -1, g));....G =
0e20: 6d 61 78 28 30 2c 20 6d 69 6e 3c 69 6e 74 3e 28 max(0, min<int>(
0e30: 6d 61 78 47 2d 31 2c 20 47 29 29 3b 0d 0a 09 09 maxG-1, G));....
0e40: 62 20 3d 20 6d 61 78 28 30 2c 20 6d 69 6e 3c 69 b = max(0, min<i
0e50: 6e 74 3e 28 6d 61 78 42 2d 31 2c 20 62 29 29 3b nt>(maxB-1, b));
0e60: 0d 0a 09 09 42 20 3d 20 6d 61 78 28 30 2c 20 6d ....B = max(0, m
0e70: 69 6e 3c 69 6e 74 3e 28 6d 61 78 42 2d 31 2c 20 in<int>(maxB-1,
0e80: 42 29 29 3b 0d 0a 09 09 72 65 74 75 72 6e 20 28 B));....return (
0e90: 52 2d 72 2b 31 29 2a 28 47 2d 67 2b 31 29 2a 28 R-r+1)*(G-g+1)*(
0ea0: 42 2d 62 2b 31 29 3b 0d 0a 09 7d 0d 0a 7d 3b 0d B-b+1);...}..};.
0eb0: 0a 0d 0a 2f 2f 20 42 45 47 49 4e 20 43 55 54 20 ...// BEGIN CUT
0ec0: 48 45 52 45 0d 0a 23 69 6e 63 6c 75 64 65 20 3c HERE..#include <
0ed0: 63 74 69 6d 65 3e 0d 0a 64 6f 75 62 6c 65 20 73 ctime>..double s
0ee0: 74 61 72 74 5f 74 69 6d 65 3b 20 73 74 72 69 6e tart_time; strin
0ef0: 67 20 74 69 6d 65 72 28 29 0d 0a 20 7b 20 6f 73 g timer().. { os
0f00: 74 72 69 6e 67 73 74 72 65 61 6d 20 6f 73 3b 20 tringstream os;
0f10: 6f 73 20 3c 3c 20 22 20 28 22 20 3c 3c 20 69 6e os << " (" << in
0f20: 74 28 28 63 6c 6f 63 6b 28 29 2d 73 74 61 72 74 t((clock()-start
0f30: 5f 74 69 6d 65 29 2f 43 4c 4f 43 4b 53 5f 50 45 _time)/CLOCKS_PE
0f40: 52 5f 53 45 43 2a 31 30 30 30 29 20 3c 3c 20 22 R_SEC*1000) << "
0f50: 20 6d 73 65 63 29 22 3b 20 72 65 74 75 72 6e 20 msec)"; return
0f60: 6f 73 2e 73 74 72 28 29 3b 20 7d 0d 0a 74 65 6d os.str(); }..tem
0f70: 70 6c 61 74 65 3c 74 79 70 65 6e 61 6d 65 20 54 plate<typename T
0f80: 3e 20 6f 73 74 72 65 61 6d 26 20 6f 70 65 72 61 > ostream& opera
0f90: 74 6f 72 3c 3c 28 6f 73 74 72 65 61 6d 26 20 6f tor<<(ostream& o
0fa0: 73 2c 20 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c s, const vector<
0fb0: 54 3e 26 20 76 29 0d 0a 20 7b 20 6f 73 20 3c 3c T>& v).. { os <<
0fc0: 20 22 7b 20 22 3b 0d 0a 20 20 20 66 6f 72 28 74 "{ ";.. for(t
0fd0: 79 70 65 6e 61 6d 65 20 76 65 63 74 6f 72 3c 54 ypename vector<T
0fe0: 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 61 74 6f >::const_iterato
0ff0: 72 20 69 74 3d 76 2e 62 65 67 69 6e 28 29 3b 20 r it=v.begin();
1000: 69 74 21 3d 76 2e 65 6e 64 28 29 3b 20 2b 2b 69 it!=v.end(); ++i
1010: 74 29 0d 0a 20 20 20 6f 73 20 3c 3c 20 27 5c 22 t).. os << '\"
1020: 27 20 3c 3c 20 2a 69 74 20 3c 3c 20 27 5c 22 27 ' << *it << '\"'
1030: 20 3c 3c 20 28 69 74 2b 31 3d 3d 76 2e 65 6e 64 << (it+1==v.end
1040: 28 29 20 3f 20 22 22 20 3a 20 22 2c 20 22 29 3b () ? "" : ", ");
1050: 20 6f 73 20 3c 3c 20 22 20 7d 22 3b 20 72 65 74 os << " }"; ret
1060: 75 72 6e 20 6f 73 3b 20 7d 0d 0a 76 6f 69 64 20 urn os; }..void
1070: 76 65 72 69 66 79 5f 63 61 73 65 28 63 6f 6e 73 verify_case(cons
1080: 74 20 64 6f 75 62 6c 65 26 20 45 78 70 65 63 74 t double& Expect
1090: 65 64 2c 20 63 6f 6e 73 74 20 64 6f 75 62 6c 65 ed, const double
10a0: 26 20 52 65 63 65 69 76 65 64 29 20 7b 0d 0a 20 & Received) {..
10b0: 62 6f 6f 6c 20 6f 6b 20 3d 20 28 61 62 73 28 45 bool ok = (abs(E
10c0: 78 70 65 63 74 65 64 20 2d 20 52 65 63 65 69 76 xpected - Receiv
10d0: 65 64 29 20 3c 20 31 65 2d 39 29 3b 0d 0a 20 69 ed) < 1e-9);.. i
10e0: 66 28 6f 6b 29 20 63 65 72 72 20 3c 3c 20 22 50 f(ok) cerr << "P
10f0: 41 53 53 45 44 22 20 3c 3c 20 74 69 6d 65 72 28 ASSED" << timer(
1100: 29 20 3c 3c 20 65 6e 64 6c 3b 20 20 65 6c 73 65 ) << endl; else
1110: 20 7b 20 63 65 72 72 20 3c 3c 20 22 46 41 49 4c { cerr << "FAIL
1120: 45 44 22 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c ED" << timer() <
1130: 3c 20 65 6e 64 6c 3b 0d 0a 20 63 65 72 72 20 3c < endl;.. cerr <
1140: 3c 20 22 5c 74 6f 3a 20 5c 22 22 20 3c 3c 20 45 < "\to: \"" << E
1150: 78 70 65 63 74 65 64 20 3c 3c 20 27 5c 22 27 20 xpected << '\"'
1160: 3c 3c 20 65 6e 64 6c 20 3c 3c 20 22 5c 74 78 3a << endl << "\tx:
1170: 20 5c 22 22 20 3c 3c 20 52 65 63 65 69 76 65 64 \"" << Received
1180: 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c << '\"' << endl
1190: 3b 20 7d 20 7d 0d 0a 23 64 65 66 69 6e 65 20 43 ; } }..#define C
11a0: 41 53 45 28 4e 29 20 7b 63 65 72 72 20 3c 3c 20 ASE(N) {cerr <<
11b0: 22 54 65 73 74 20 43 61 73 65 20 23 22 20 3c 3c "Test Case #" <<
11c0: 20 4e 20 3c 3c 20 22 2e 2e 2e 22 20 3c 3c 20 66 N << "..." << f
11d0: 6c 75 73 68 3b 20 73 74 61 72 74 5f 74 69 6d 65 lush; start_time
11e0: 3d 63 6c 6f 63 6b 28 29 3b 0d 0a 23 64 65 66 69 =clock();..#defi
11f0: 6e 65 20 45 4e 44 09 20 76 65 72 69 66 79 5f 63 ne END. verify_c
1200: 61 73 65 28 5f 2c 20 52 61 6e 64 6f 6d 43 6f 6c ase(_, RandomCol
1210: 6f 72 69 6e 67 28 29 2e 67 65 74 50 72 6f 62 61 oring().getProba
1220: 62 69 6c 69 74 79 28 4e 2c 20 6d 61 78 52 2c 20 bility(N, maxR,
1230: 6d 61 78 47 2c 20 6d 61 78 42 2c 20 73 74 61 72 maxG, maxB, star
1240: 74 52 2c 20 73 74 61 72 74 47 2c 20 73 74 61 72 tR, startG, star
1250: 74 42 2c 20 64 31 2c 20 64 32 29 29 3b 7d 0d 0a tB, d1, d2));}..
1260: 69 6e 74 20 6d 61 69 6e 28 29 7b 0d 0a 0d 0a 43 int main(){....C
1270: 41 53 45 28 30 29 0d 0a 09 69 6e 74 20 4e 20 3d ASE(0)...int N =
1280: 20 32 3b 20 0d 0a 09 69 6e 74 20 6d 61 78 52 20 2; ...int maxR
1290: 3d 20 35 3b 20 0d 0a 09 69 6e 74 20 6d 61 78 47 = 5; ...int maxG
12a0: 20 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 6d 61 78 = 1; ...int max
12b0: 42 20 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 73 74 B = 1; ...int st
12c0: 61 72 74 52 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 artR = 2; ...int
12d0: 20 73 74 61 72 74 47 20 3d 20 30 3b 20 0d 0a 09 startG = 0; ...
12e0: 69 6e 74 20 73 74 61 72 74 42 20 3d 20 30 3b 20 int startB = 0;
12f0: 0d 0a 09 69 6e 74 20 64 31 20 3d 20 30 3b 20 0d ...int d1 = 0; .
1300: 0a 09 69 6e 74 20 64 32 20 3d 20 31 3b 20 0d 0a ..int d2 = 1; ..
1310: 09 64 6f 75 62 6c 65 20 5f 20 3d 20 30 2e 30 3b .double _ = 0.0;
1320: 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 31 29 0d ..END..CASE(1).
1330: 0a 09 69 6e 74 20 4e 20 3d 20 33 3b 20 0d 0a 09 ..int N = 3; ...
1340: 69 6e 74 20 6d 61 78 52 20 3d 20 35 3b 20 0d 0a int maxR = 5; ..
1350: 09 69 6e 74 20 6d 61 78 47 20 3d 20 31 3b 20 0d .int maxG = 1; .
1360: 0a 09 69 6e 74 20 6d 61 78 42 20 3d 20 31 3b 20 ..int maxB = 1;
1370: 0d 0a 09 69 6e 74 20 73 74 61 72 74 52 20 3d 20 ...int startR =
1380: 32 3b 20 0d 0a 09 69 6e 74 20 73 74 61 72 74 47 2; ...int startG
1390: 20 3d 20 30 3b 20 0d 0a 09 69 6e 74 20 73 74 61 = 0; ...int sta
13a0: 72 74 42 20 3d 20 30 3b 20 0d 0a 09 69 6e 74 20 rtB = 0; ...int
13b0: 64 31 20 3d 20 30 3b 20 0d 0a 09 69 6e 74 20 64 d1 = 0; ...int d
13c0: 32 20 3d 20 31 3b 20 0d 0a 09 64 6f 75 62 6c 65 2 = 1; ...double
13d0: 20 5f 20 3d 20 30 2e 32 32 32 32 32 32 32 32 32 _ = 0.222222222
13e0: 32 32 32 32 32 32 32 37 3b 20 0d 0a 45 4e 44 0d 22222227; ..END.
13f0: 0a 43 41 53 45 28 32 29 0d 0a 09 69 6e 74 20 4e .CASE(2)...int N
1400: 20 3d 20 37 3b 20 0d 0a 09 69 6e 74 20 6d 61 78 = 7; ...int max
1410: 52 20 3d 20 34 3b 20 0d 0a 09 69 6e 74 20 6d 61 R = 4; ...int ma
1420: 78 47 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 6d xG = 2; ...int m
1430: 61 78 42 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 axB = 2; ...int
1440: 73 74 61 72 74 52 20 3d 20 30 3b 20 0d 0a 09 69 startR = 0; ...i
1450: 6e 74 20 73 74 61 72 74 47 20 3d 20 30 3b 20 0d nt startG = 0; .
1460: 0a 09 69 6e 74 20 73 74 61 72 74 42 20 3d 20 30 ..int startB = 0
1470: 3b 20 0d 0a 09 69 6e 74 20 64 31 20 3d 20 33 3b ; ...int d1 = 3;
1480: 20 0d 0a 09 69 6e 74 20 64 32 20 3d 20 33 3b 20 ...int d2 = 3;
1490: 0d 0a 09 64 6f 75 62 6c 65 20 5f 20 3d 20 31 2e ...double _ = 1.
14a0: 30 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 33 0; ..END..CASE(3
14b0: 29 0d 0a 09 69 6e 74 20 4e 20 3d 20 31 30 3b 20 )...int N = 10;
14c0: 0d 0a 09 69 6e 74 20 6d 61 78 52 20 3d 20 37 3b ...int maxR = 7;
14d0: 20 0d 0a 09 69 6e 74 20 6d 61 78 47 20 3d 20 38 ...int maxG = 8
14e0: 3b 20 0d 0a 09 69 6e 74 20 6d 61 78 42 20 3d 20 ; ...int maxB =
14f0: 39 3b 20 0d 0a 09 69 6e 74 20 73 74 61 72 74 52 9; ...int startR
1500: 20 3d 20 31 3b 20 0d 0a 09 69 6e 74 20 73 74 61 = 1; ...int sta
1510: 72 74 47 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 rtG = 2; ...int
1520: 73 74 61 72 74 42 20 3d 20 33 3b 20 0d 0a 09 69 startB = 3; ...i
1530: 6e 74 20 64 31 20 3d 20 30 3b 20 0d 0a 09 69 6e nt d1 = 0; ...in
1540: 74 20 64 32 20 3d 20 31 30 3b 20 0d 0a 09 64 6f t d2 = 10; ...do
1550: 75 62 6c 65 20 5f 20 3d 20 30 2e 30 3b 20 0d 0a uble _ = 0.0; ..
1560: 45 4e 44 0d 0a 43 41 53 45 28 34 29 0d 0a 09 69 END..CASE(4)...i
1570: 6e 74 20 4e 20 3d 20 31 30 3b 20 0d 0a 09 69 6e nt N = 10; ...in
1580: 74 20 6d 61 78 52 20 3d 20 37 3b 20 0d 0a 09 69 t maxR = 7; ...i
1590: 6e 74 20 6d 61 78 47 20 3d 20 38 3b 20 0d 0a 09 nt maxG = 8; ...
15a0: 69 6e 74 20 6d 61 78 42 20 3d 20 39 3b 20 0d 0a int maxB = 9; ..
15b0: 09 69 6e 74 20 73 74 61 72 74 52 20 3d 20 31 3b .int startR = 1;
15c0: 20 0d 0a 09 69 6e 74 20 73 74 61 72 74 47 20 3d ...int startG =
15d0: 20 32 3b 20 0d 0a 09 69 6e 74 20 73 74 61 72 74 2; ...int start
15e0: 42 20 3d 20 33 3b 20 0d 0a 09 69 6e 74 20 64 31 B = 3; ...int d1
15f0: 20 3d 20 34 3b 20 0d 0a 09 69 6e 74 20 64 32 20 = 4; ...int d2
1600: 3d 20 31 30 3b 20 0d 0a 09 64 6f 75 62 6c 65 20 = 10; ...double
1610: 5f 20 3d 20 30 2e 33 37 38 32 36 32 34 35 39 34 _ = 0.3782624594
1620: 33 39 36 37 33 39 36 3b 20 0d 0a 45 4e 44 0d 0a 3967396; ..END..
1630: 43 41 53 45 28 35 29 0d 0a 09 69 6e 74 20 4e 20 CASE(5)...int N
1640: 3d 20 33 3b 20 0d 0a 09 69 6e 74 20 6d 61 78 52 = 3; ...int maxR
1650: 20 3d 20 33 3b 20 0d 0a 09 69 6e 74 20 6d 61 78 = 3; ...int max
1660: 47 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 6d 61 G = 2; ...int ma
1670: 78 42 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 73 xB = 2; ...int s
1680: 74 61 72 74 52 20 3d 20 31 3b 20 0d 0a 09 69 6e tartR = 1; ...in
1690: 74 20 73 74 61 72 74 47 20 3d 20 30 3b 20 0d 0a t startG = 0; ..
16a0: 09 69 6e 74 20 73 74 61 72 74 42 20 3d 20 30 3b .int startB = 0;
16b0: 20 0d 0a 09 69 6e 74 20 64 31 20 3d 20 31 3b 20 ...int d1 = 1;
16c0: 0d 0a 09 69 6e 74 20 64 32 20 3d 20 32 3b 20 0d ...int d2 = 2; .
16d0: 0a 09 64 6f 75 62 6c 65 20 5f 20 3d 20 30 2e 30 ..double _ = 0.0
16e0: 39 30 39 30 39 30 39 30 39 30 39 30 39 31 34 38 9090909090909148
16f0: 3b 20 0d 0a 45 4e 44 0d 0a 2f 2a 0d 0a 43 41 53 ; ..END../*..CAS
1700: 45 28 36 29 0d 0a 09 69 6e 74 20 4e 20 3d 20 3b E(6)...int N = ;
1710: 20 0d 0a 09 69 6e 74 20 6d 61 78 52 20 3d 20 3b ...int maxR = ;
1720: 20 0d 0a 09 69 6e 74 20 6d 61 78 47 20 3d 20 3b ...int maxG = ;
1730: 20 0d 0a 09 69 6e 74 20 6d 61 78 42 20 3d 20 3b ...int maxB = ;
1740: 20 0d 0a 09 69 6e 74 20 73 74 61 72 74 52 20 3d ...int startR =
1750: 20 3b 20 0d 0a 09 69 6e 74 20 73 74 61 72 74 47 ; ...int startG
1760: 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 73 74 61 72 = ; ...int star
1770: 74 42 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 64 31 tB = ; ...int d1
1780: 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 64 32 20 3d = ; ...int d2 =
1790: 20 3b 20 0d 0a 09 64 6f 75 62 6c 65 20 5f 20 3d ; ...double _ =
17a0: 20 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 37 ; ..END..CASE(7
17b0: 29 0d 0a 09 69 6e 74 20 4e 20 3d 20 3b 20 0d 0a )...int N = ; ..
17c0: 09 69 6e 74 20 6d 61 78 52 20 3d 20 3b 20 0d 0a .int maxR = ; ..
17d0: 09 69 6e 74 20 6d 61 78 47 20 3d 20 3b 20 0d 0a .int maxG = ; ..
17e0: 09 69 6e 74 20 6d 61 78 42 20 3d 20 3b 20 0d 0a .int maxB = ; ..
17f0: 09 69 6e 74 20 73 74 61 72 74 52 20 3d 20 3b 20 .int startR = ;
1800: 0d 0a 09 69 6e 74 20 73 74 61 72 74 47 20 3d 20 ...int startG =
1810: 3b 20 0d 0a 09 69 6e 74 20 73 74 61 72 74 42 20 ; ...int startB
1820: 3d 20 3b 20 0d 0a 09 69 6e 74 20 64 31 20 3d 20 = ; ...int d1 =
1830: 3b 20 0d 0a 09 69 6e 74 20 64 32 20 3d 20 3b 20 ; ...int d2 = ;
1840: 0d 0a 09 64 6f 75 62 6c 65 20 5f 20 3d 20 3b 20 ...double _ = ;
1850: 0d 0a 45 4e 44 0d 0a 2a 2f 0d 0a 7d 0d 0a 2f 2f ..END..*/..}..//
1860: 20 45 4e 44 20 43 55 54 20 48 45 52 45 0d 0a END CUT HERE..