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 74 65 6d 70 6c ong LL;....templ
0140: 61 74 65 3c 74 79 70 65 6e 61 6d 65 20 54 20 3d ate<typename T =
0150: 20 4c 4c 3e 0d 0a 73 74 72 75 63 74 20 46 65 6e LL>..struct Fen
0160: 77 69 63 6b 54 72 65 65 0d 0a 7b 0d 0a 09 76 65 wickTree..{...ve
0170: 63 74 6f 72 3c 54 3e 20 78 3b 0d 0a 09 46 65 6e ctor<T> x;...Fen
0180: 77 69 63 6b 54 72 65 65 28 73 69 7a 65 5f 74 20 wickTree(size_t
0190: 6e 2c 20 63 6f 6e 73 74 20 54 26 20 76 20 3d 20 n, const T& v =
01a0: 54 28 29 29 20 3a 20 78 28 6e 2c 20 76 29 20 7b T()) : x(n, v) {
01b0: 7d 0d 0a 0d 0a 09 76 6f 69 64 20 69 6e 63 72 28 }.....void incr(
01c0: 20 69 6e 74 20 6b 2c 20 63 6f 6e 73 74 20 54 26 int k, const T&
01d0: 20 61 20 29 20 7b 20 2f 2f 20 7a 5b 6b 5d 20 2b a ) { // z[k] +
01e0: 3d 20 61 3b 0d 0a 09 09 66 6f 72 28 3b 20 6b 20 = a;....for(; k
01f0: 3c 20 78 2e 73 69 7a 65 28 29 3b 20 6b 7c 3d 6b < x.size(); k|=k
0200: 2b 31 29 0d 0a 09 09 09 78 5b 6b 5d 20 2b 3d 20 +1).....x[k] +=
0210: 61 3b 0d 0a 09 7d 0d 0a 0d 0a 09 54 20 73 75 6d a;...}.....T sum
0220: 28 69 6e 74 20 69 2c 20 69 6e 74 20 6a 29 20 7b (int i, int j) {
0230: 20 2f 2f 20 7a 5b 69 5d 2b 2e 2e 2e 2b 7a 5b 6a // z[i]+...+z[j
0240: 5d 20 3a 20 69 6e 63 6c 75 73 69 76 65 0d 0a 09 ] : inclusive...
0250: 09 69 66 28 69 29 0d 0a 09 09 09 72 65 74 75 72 .if(i).....retur
0260: 6e 20 73 75 6d 28 30 2c 20 6a 29 20 2d 20 73 75 n sum(0, j) - su
0270: 6d 28 30 2c 20 69 2d 31 29 3b 0d 0a 09 09 65 6c m(0, i-1);....el
0280: 73 65 20 7b 0d 0a 09 09 09 54 20 76 20 3d 20 54 se {.....T v = T
0290: 28 29 3b 0d 0a 09 09 09 66 6f 72 28 3b 20 6a 3e ();.....for(; j>
02a0: 3d 30 3b 20 6a 3d 28 6a 26 28 6a 2b 31 29 29 2d =0; j=(j&(j+1))-
02b0: 31 29 0d 0a 09 09 09 09 76 20 2b 3d 20 78 5b 6a 1)......v += x[j
02c0: 5d 3b 0d 0a 09 09 09 72 65 74 75 72 6e 20 76 3b ];.....return v;
02d0: 0d 0a 09 09 7d 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d ....}...}..};...
02e0: 0a 73 74 61 74 69 63 20 63 6f 6e 73 74 20 69 6e .static const in
02f0: 74 20 4d 4f 44 56 41 4c 20 3d 20 31 30 30 30 30 t MODVAL = 10000
0300: 30 30 30 30 37 3b 0d 0a 0d 0a 63 6c 61 73 73 20 00007;....class
0310: 50 72 6f 64 75 63 74 4f 66 50 72 69 63 65 73 0d ProductOfPrices.
0320: 0a 7b 0d 0a 70 75 62 6c 69 63 3a 0d 0a 09 69 6e .{..public:...in
0330: 74 20 70 72 6f 64 75 63 74 28 69 6e 74 20 4e 2c t product(int N,
0340: 20 69 6e 74 20 4c 2c 20 69 6e 74 20 58 30 2c 20 int L, int X0,
0350: 69 6e 74 20 41 2c 20 69 6e 74 20 42 29 20 0d 0a int A, int B) ..
0360: 09 7b 0d 0a 09 09 4c 4c 20 72 65 73 75 6c 74 20 .{....LL result
0370: 3d 20 31 3b 0d 0a 0d 0a 09 09 46 65 6e 77 69 63 = 1;......Fenwic
0380: 6b 54 72 65 65 3c 3e 20 66 77 28 4c 29 2c 20 6e kTree<> fw(L), n
0390: 6d 28 4c 29 3b 0d 0a 0d 0a 09 09 4c 4c 20 58 20 m(L);......LL X
03a0: 3d 20 58 30 20 25 20 4c 3b 0d 0a 09 09 66 77 2e = X0 % L;....fw.
03b0: 69 6e 63 72 28 58 2c 20 58 29 3b 0d 0a 09 09 6e incr(X, X);....n
03c0: 6d 2e 69 6e 63 72 28 58 2c 20 31 29 3b 0d 0a 09 m.incr(X, 1);...
03d0: 09 66 6f 72 28 69 6e 74 20 69 3d 31 3b 20 69 3c .for(int i=1; i<
03e0: 4e 3b 20 2b 2b 69 29 0d 0a 09 09 7b 0d 0a 09 09 N; ++i)....{....
03f0: 09 58 20 3d 20 28 58 2a 41 2b 42 29 20 25 20 4c .X = (X*A+B) % L
0400: 3b 0d 0a 0d 0a 09 09 09 4c 4c 20 73 75 6d 4c 20 ;.......LL sumL
0410: 3d 20 66 77 2e 73 75 6d 28 30 2c 20 58 2d 31 29 = fw.sum(0, X-1)
0420: 3b 0d 0a 09 09 09 4c 4c 20 73 75 6d 52 20 3d 20 ;.....LL sumR =
0430: 66 77 2e 73 75 6d 28 58 2c 20 4c 2d 31 29 3b 0d fw.sum(X, L-1);.
0440: 0a 09 09 09 4c 4c 20 6e 75 6d 4c 20 3d 20 6e 6d ....LL numL = nm
0450: 2e 73 75 6d 28 30 2c 20 58 2d 31 29 3b 0d 0a 09 .sum(0, X-1);...
0460: 09 09 4c 4c 20 6e 75 6d 52 20 3d 20 6e 6d 2e 73 ..LL numR = nm.s
0470: 75 6d 28 58 2c 20 4c 2d 31 29 3b 0d 0a 0d 0a 09 um(X, L-1);.....
0480: 09 09 4c 4c 20 70 72 69 63 65 20 3d 20 28 28 58 ..LL price = ((X
0490: 2a 6e 75 6d 4c 20 2d 20 73 75 6d 4c 29 20 2b 20 *numL - sumL) +
04a0: 28 73 75 6d 52 20 2d 20 58 2a 6e 75 6d 52 29 29 (sumR - X*numR))
04b0: 20 25 20 4d 4f 44 56 41 4c 3b 0d 0a 09 09 09 72 % MODVAL;.....r
04c0: 65 73 75 6c 74 20 3d 20 28 72 65 73 75 6c 74 20 esult = (result
04d0: 2a 20 70 72 69 63 65 29 20 25 20 4d 4f 44 56 41 * price) % MODVA
04e0: 4c 3b 0d 0a 0d 0a 09 09 09 66 77 2e 69 6e 63 72 L;.......fw.incr
04f0: 28 58 2c 20 58 29 3b 0d 0a 09 09 09 6e 6d 2e 69 (X, X);.....nm.i
0500: 6e 63 72 28 58 2c 20 31 29 3b 0d 0a 09 09 7d 0d ncr(X, 1);....}.
0510: 0a 09 09 72 65 74 75 72 6e 20 72 65 73 75 6c 74 ...return result
0520: 3b 0d 0a 09 7d 0d 0a 0d 0a 2f 2f 20 42 45 47 49 ;...}....// BEGI
0530: 4e 20 43 55 54 20 48 45 52 45 0d 0a 09 70 75 62 N CUT HERE...pub
0540: 6c 69 63 3a 0d 0a 09 76 6f 69 64 20 72 75 6e 5f lic:...void run_
0550: 74 65 73 74 28 69 6e 74 20 43 61 73 65 29 20 7b test(int Case) {
0560: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
0570: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 30 29 ) || (Case == 0)
0580: 29 20 74 65 73 74 5f 63 61 73 65 5f 30 28 29 3b ) test_case_0();
0590: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
05a0: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 31 29 ) || (Case == 1)
05b0: 29 20 74 65 73 74 5f 63 61 73 65 5f 31 28 29 3b ) test_case_1();
05c0: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
05d0: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 32 29 ) || (Case == 2)
05e0: 29 20 74 65 73 74 5f 63 61 73 65 5f 32 28 29 3b ) test_case_2();
05f0: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
0600: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 33 29 ) || (Case == 3)
0610: 29 20 74 65 73 74 5f 63 61 73 65 5f 33 28 29 3b ) test_case_3();
0620: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
0630: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 34 29 ) || (Case == 4)
0640: 29 20 74 65 73 74 5f 63 61 73 65 5f 34 28 29 3b ) test_case_4();
0650: 20 7d 0d 0a 09 70 72 69 76 61 74 65 3a 0d 0a 09 }...private:...
0660: 74 65 6d 70 6c 61 74 65 20 3c 74 79 70 65 6e 61 template <typena
0670: 6d 65 20 54 3e 20 73 74 72 69 6e 67 20 70 72 69 me T> string pri
0680: 6e 74 5f 61 72 72 61 79 28 63 6f 6e 73 74 20 76 nt_array(const v
0690: 65 63 74 6f 72 3c 54 3e 20 26 56 29 20 7b 20 6f ector<T> &V) { o
06a0: 73 74 72 69 6e 67 73 74 72 65 61 6d 20 6f 73 3b stringstream os;
06b0: 20 6f 73 20 3c 3c 20 22 7b 20 22 3b 20 66 6f 72 os << "{ "; for
06c0: 20 28 74 79 70 65 6e 61 6d 65 20 76 65 63 74 6f (typename vecto
06d0: 72 3c 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 r<T>::const_iter
06e0: 61 74 6f 72 20 69 74 65 72 20 3d 20 56 2e 62 65 ator iter = V.be
06f0: 67 69 6e 28 29 3b 20 69 74 65 72 20 21 3d 20 56 gin(); iter != V
0700: 2e 65 6e 64 28 29 3b 20 2b 2b 69 74 65 72 29 20 .end(); ++iter)
0710: 6f 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 os << '\"' << *i
0720: 74 65 72 20 3c 3c 20 22 5c 22 2c 22 3b 20 6f 73 ter << "\","; os
0730: 20 3c 3c 20 22 20 7d 22 3b 20 72 65 74 75 72 6e << " }"; return
0740: 20 6f 73 2e 73 74 72 28 29 3b 20 7d 0d 0a 09 76 os.str(); }...v
0750: 6f 69 64 20 76 65 72 69 66 79 5f 63 61 73 65 28 oid verify_case(
0760: 69 6e 74 20 43 61 73 65 2c 20 63 6f 6e 73 74 20 int Case, const
0770: 69 6e 74 20 26 45 78 70 65 63 74 65 64 2c 20 63 int &Expected, c
0780: 6f 6e 73 74 20 69 6e 74 20 26 52 65 63 65 69 76 onst int &Receiv
0790: 65 64 29 20 7b 20 63 65 72 72 20 3c 3c 20 22 54 ed) { cerr << "T
07a0: 65 73 74 20 43 61 73 65 20 23 22 20 3c 3c 20 43 est Case #" << C
07b0: 61 73 65 20 3c 3c 20 22 2e 2e 2e 22 3b 20 69 66 ase << "..."; if
07c0: 20 28 45 78 70 65 63 74 65 64 20 3d 3d 20 52 65 (Expected == Re
07d0: 63 65 69 76 65 64 29 20 63 65 72 72 20 3c 3c 20 ceived) cerr <<
07e0: 22 50 41 53 53 45 44 22 20 3c 3c 20 65 6e 64 6c "PASSED" << endl
07f0: 3b 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c 3c ; else { cerr <<
0800: 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 65 6e 64 "FAILED" << end
0810: 6c 3b 20 63 65 72 72 20 3c 3c 20 22 5c 74 45 78 l; cerr << "\tEx
0820: 70 65 63 74 65 64 3a 20 5c 22 22 20 3c 3c 20 45 pected: \"" << E
0830: 78 70 65 63 74 65 64 20 3c 3c 20 27 5c 22 27 20 xpected << '\"'
0840: 3c 3c 20 65 6e 64 6c 3b 20 63 65 72 72 20 3c 3c << endl; cerr <<
0850: 20 22 5c 74 52 65 63 65 69 76 65 64 3a 20 5c 22 "\tReceived: \"
0860: 22 20 3c 3c 20 52 65 63 65 69 76 65 64 20 3c 3c " << Received <<
0870: 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d '\"' << endl; }
0880: 20 7d 0d 0a 09 76 6f 69 64 20 74 65 73 74 5f 63 }...void test_c
0890: 61 73 65 5f 30 28 29 20 7b 20 69 6e 74 20 41 72 ase_0() { int Ar
08a0: 67 30 20 3d 20 35 3b 20 69 6e 74 20 41 72 67 31 g0 = 5; int Arg1
08b0: 20 3d 20 31 30 3b 20 69 6e 74 20 41 72 67 32 20 = 10; int Arg2
08c0: 3d 20 33 3b 20 69 6e 74 20 41 72 67 33 20 3d 20 = 3; int Arg3 =
08d0: 31 3b 20 69 6e 74 20 41 72 67 34 20 3d 20 31 3b 1; int Arg4 = 1;
08e0: 20 69 6e 74 20 41 72 67 35 20 3d 20 31 38 30 3b int Arg5 = 180;
08f0: 20 76 65 72 69 66 79 5f 63 61 73 65 28 30 2c 20 verify_case(0,
0900: 41 72 67 35 2c 20 70 72 6f 64 75 63 74 28 41 72 Arg5, product(Ar
0910: 67 30 2c 20 41 72 67 31 2c 20 41 72 67 32 2c 20 g0, Arg1, Arg2,
0920: 41 72 67 33 2c 20 41 72 67 34 29 29 3b 20 7d 0d Arg3, Arg4)); }.
0930: 0a 09 76 6f 69 64 20 74 65 73 74 5f 63 61 73 65 ..void test_case
0940: 5f 31 28 29 20 7b 20 69 6e 74 20 41 72 67 30 20 _1() { int Arg0
0950: 3d 20 33 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 = 3; int Arg1 =
0960: 32 30 3b 20 69 6e 74 20 41 72 67 32 20 3d 20 35 20; int Arg2 = 5
0970: 3b 20 69 6e 74 20 41 72 67 33 20 3d 20 32 3b 20 ; int Arg3 = 2;
0980: 69 6e 74 20 41 72 67 34 20 3d 20 33 3b 20 69 6e int Arg4 = 3; in
0990: 74 20 41 72 67 35 20 3d 20 36 34 3b 20 76 65 72 t Arg5 = 64; ver
09a0: 69 66 79 5f 63 61 73 65 28 31 2c 20 41 72 67 35 ify_case(1, Arg5
09b0: 2c 20 70 72 6f 64 75 63 74 28 41 72 67 30 2c 20 , product(Arg0,
09c0: 41 72 67 31 2c 20 41 72 67 32 2c 20 41 72 67 33 Arg1, Arg2, Arg3
09d0: 2c 20 41 72 67 34 29 29 3b 20 7d 0d 0a 09 76 6f , Arg4)); }...vo
09e0: 69 64 20 74 65 73 74 5f 63 61 73 65 5f 32 28 29 id test_case_2()
09f0: 20 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 34 3b { int Arg0 = 4;
0a00: 20 69 6e 74 20 41 72 67 31 20 3d 20 32 31 3b 20 int Arg1 = 21;
0a10: 69 6e 74 20 41 72 67 32 20 3d 20 31 3b 20 69 6e int Arg2 = 1; in
0a20: 74 20 41 72 67 33 20 3d 20 37 3b 20 69 6e 74 20 t Arg3 = 7; int
0a30: 41 72 67 34 20 3d 20 31 3b 20 69 6e 74 20 41 72 Arg4 = 1; int Ar
0a40: 67 35 20 3d 20 33 30 38 37 3b 20 76 65 72 69 66 g5 = 3087; verif
0a50: 79 5f 63 61 73 65 28 32 2c 20 41 72 67 35 2c 20 y_case(2, Arg5,
0a60: 70 72 6f 64 75 63 74 28 41 72 67 30 2c 20 41 72 product(Arg0, Ar
0a70: 67 31 2c 20 41 72 67 32 2c 20 41 72 67 33 2c 20 g1, Arg2, Arg3,
0a80: 41 72 67 34 29 29 3b 20 7d 0d 0a 09 76 6f 69 64 Arg4)); }...void
0a90: 20 74 65 73 74 5f 63 61 73 65 5f 33 28 29 20 7b test_case_3() {
0aa0: 20 69 6e 74 20 41 72 67 30 20 3d 20 31 30 3b 20 int Arg0 = 10;
0ab0: 69 6e 74 20 41 72 67 31 20 3d 20 31 30 30 3b 20 int Arg1 = 100;
0ac0: 69 6e 74 20 41 72 67 32 20 3d 20 34 3b 20 69 6e int Arg2 = 4; in
0ad0: 74 20 41 72 67 33 20 3d 20 33 37 3b 20 69 6e 74 t Arg3 = 37; int
0ae0: 20 41 72 67 34 20 3d 20 31 31 3b 20 69 6e 74 20 Arg4 = 11; int
0af0: 41 72 67 35 20 3d 20 35 39 31 38 36 30 37 36 37 Arg5 = 591860767
0b00: 3b 20 76 65 72 69 66 79 5f 63 61 73 65 28 33 2c ; verify_case(3,
0b10: 20 41 72 67 35 2c 20 70 72 6f 64 75 63 74 28 41 Arg5, product(A
0b20: 72 67 30 2c 20 41 72 67 31 2c 20 41 72 67 32 2c rg0, Arg1, Arg2,
0b30: 20 41 72 67 33 2c 20 41 72 67 34 29 29 3b 20 7d Arg3, Arg4)); }
0b40: 0d 0a 09 76 6f 69 64 20 74 65 73 74 5f 63 61 73 ...void test_cas
0b50: 65 5f 34 28 29 20 7b 20 69 6e 74 20 41 72 67 30 e_4() { int Arg0
0b60: 20 3d 20 35 3b 20 69 6e 74 20 41 72 67 31 20 3d = 5; int Arg1 =
0b70: 20 32 30 30 30 30 30 3b 20 69 6e 74 20 41 72 67 200000; int Arg
0b80: 32 20 3d 20 39 39 39 39 39 39 39 39 39 3b 20 69 2 = 999999999; i
0b90: 6e 74 20 41 72 67 33 20 3d 20 31 32 33 34 35 36 nt Arg3 = 123456
0ba0: 37 38 39 3b 20 69 6e 74 20 41 72 67 34 20 3d 20 789; int Arg4 =
0bb0: 39 38 37 36 35 34 33 32 31 3b 20 69 6e 74 20 41 987654321; int A
0bc0: 72 67 35 20 3d 20 34 39 39 37 33 39 31 37 35 3b rg5 = 499739175;
0bd0: 20 76 65 72 69 66 79 5f 63 61 73 65 28 34 2c 20 verify_case(4,
0be0: 41 72 67 35 2c 20 70 72 6f 64 75 63 74 28 41 72 Arg5, product(Ar
0bf0: 67 30 2c 20 41 72 67 31 2c 20 41 72 67 32 2c 20 g0, Arg1, Arg2,
0c00: 41 72 67 33 2c 20 41 72 67 34 29 29 3b 20 7d 0d Arg3, Arg4)); }.
0c10: 0a 0d 0a 2f 2f 20 45 4e 44 20 43 55 54 20 48 45 ...// END CUT HE
0c20: 52 45 0d 0a 7d 3b 0d 0a 2f 2f 20 42 45 47 49 4e RE..};..// BEGIN
0c30: 20 43 55 54 20 48 45 52 45 20 0d 0a 69 6e 74 20 CUT HERE ..int
0c40: 6d 61 69 6e 28 29 20 7b 20 50 72 6f 64 75 63 74 main() { Product
0c50: 4f 66 50 72 69 63 65 73 28 29 2e 72 75 6e 5f 74 OfPrices().run_t
0c60: 65 73 74 28 2d 31 29 3b 20 7d 0d 0a 2f 2f 20 45 est(-1); }..// E
0c70: 4e 44 20 43 55 54 20 48 45 52 45 0d 0a ND CUT HERE..