Artifact e5662c451b84a1852b7ab4e897c1baae11af01e7:
0000: 0a 2f 2f 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d .//-------------
0010: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0020: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0030: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0040: 0a 2f 2f 20 4d 6f 64 75 6c 6f 20 41 72 69 74 68 .// Modulo Arith
0050: 6d 65 74 69 63 73 0a 2f 2f 0a 2f 2f 20 56 65 72 metics.//.// Ver
0060: 69 66 69 65 64 20 62 79 0a 2f 2f 20 20 20 2d 20 ified by.// -
0070: 54 43 4f 31 30 20 52 33 20 4c 56 33 0a 2f 2f 20 TCO10 R3 LV3.//
0080: 20 20 2d 20 53 52 4d 20 35 34 35 20 4c 56 32 0a - SRM 545 LV2.
0090: 2f 2f 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d //--------------
00a0: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
00b0: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
00c0: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 0a ---------------.
00d0: 0a 73 74 61 74 69 63 20 63 6f 6e 73 74 20 69 6e .static const in
00e0: 74 20 4d 4f 44 56 41 4c 20 3d 20 31 30 30 30 30 t MODVAL = 10000
00f0: 30 30 30 30 37 3b 0a 73 74 72 75 63 74 20 6d 69 00007;.struct mi
0100: 6e 74 0a 7b 0a 09 69 6e 74 20 76 61 6c 3b 0a 09 nt.{..int val;..
0110: 6d 69 6e 74 28 29 3a 76 61 6c 28 30 29 7b 7d 0a mint():val(0){}.
0120: 09 6d 69 6e 74 28 69 6e 74 20 20 20 20 78 29 3a .mint(int x):
0130: 76 61 6c 28 78 25 4d 4f 44 56 41 4c 29 20 7b 7d val(x%MODVAL) {}
0140: 0a 09 6d 69 6e 74 28 73 69 7a 65 5f 74 20 78 29 ..mint(size_t x)
0150: 3a 76 61 6c 28 78 25 4d 4f 44 56 41 4c 29 20 7b :val(x%MODVAL) {
0160: 7d 0a 09 6d 69 6e 74 28 4c 4c 20 20 20 20 20 78 }..mint(LL x
0170: 29 3a 76 61 6c 28 78 25 4d 4f 44 56 41 4c 29 20 ):val(x%MODVAL)
0180: 7b 7d 0a 7d 3b 0a 6d 69 6e 74 26 20 6f 70 65 72 {}.};.mint& oper
0190: 61 74 6f 72 2b 3d 28 6d 69 6e 74 26 20 78 2c 20 ator+=(mint& x,
01a0: 6d 69 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e mint y) { return
01b0: 20 78 20 3d 20 78 2e 76 61 6c 2b 79 2e 76 61 6c x = x.val+y.val
01c0: 3b 20 7d 0a 6d 69 6e 74 26 20 6f 70 65 72 61 74 ; }.mint& operat
01d0: 6f 72 2d 3d 28 6d 69 6e 74 26 20 78 2c 20 6d 69 or-=(mint& x, mi
01e0: 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78 nt y) { return x
01f0: 20 3d 20 78 2e 76 61 6c 2d 79 2e 76 61 6c 2b 4d = x.val-y.val+M
0200: 4f 44 56 41 4c 3b 20 7d 0a 6d 69 6e 74 26 20 6f ODVAL; }.mint& o
0210: 70 65 72 61 74 6f 72 2a 3d 28 6d 69 6e 74 26 20 perator*=(mint&
0220: 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 72 65 74 x, mint y) { ret
0230: 75 72 6e 20 78 20 3d 20 4c 4c 28 78 2e 76 61 6c urn x = LL(x.val
0240: 29 2a 79 2e 76 61 6c 3b 20 7d 0a 6d 69 6e 74 20 )*y.val; }.mint
0250: 50 4f 57 28 6d 69 6e 74 20 78 2c 20 4c 4c 20 65 POW(mint x, LL e
0260: 29 20 7b 20 6d 69 6e 74 20 76 3d 31 3b 20 66 6f ) { mint v=1; fo
0270: 72 28 3b 65 3b 78 2a 3d 78 2c 65 3e 3e 3d 31 29 r(;e;x*=x,e>>=1)
0280: 20 69 66 28 65 26 31 29 20 76 2a 3d 78 3b 20 72 if(e&1) v*=x; r
0290: 65 74 75 72 6e 20 76 3b 20 7d 0a 6d 69 6e 74 26 eturn v; }.mint&
02a0: 20 6f 70 65 72 61 74 6f 72 2f 3d 28 6d 69 6e 74 operator/=(mint
02b0: 26 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 72 & x, mint y) { r
02c0: 65 74 75 72 6e 20 78 20 2a 3d 20 50 4f 57 28 79 eturn x *= POW(y
02d0: 2c 20 4d 4f 44 56 41 4c 2d 32 29 3b 20 7d 0a 6d , MODVAL-2); }.m
02e0: 69 6e 74 20 6f 70 65 72 61 74 6f 72 2b 28 6d 69 int operator+(mi
02f0: 6e 74 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 nt x, mint y) {
0300: 72 65 74 75 72 6e 20 78 2b 3d 79 3b 20 7d 0a 6d return x+=y; }.m
0310: 69 6e 74 20 6f 70 65 72 61 74 6f 72 2d 28 6d 69 int operator-(mi
0320: 6e 74 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 nt x, mint y) {
0330: 72 65 74 75 72 6e 20 78 2d 3d 79 3b 20 7d 0a 6d return x-=y; }.m
0340: 69 6e 74 20 6f 70 65 72 61 74 6f 72 2a 28 6d 69 int operator*(mi
0350: 6e 74 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 nt x, mint y) {
0360: 72 65 74 75 72 6e 20 78 2a 3d 79 3b 20 7d 0a 6d return x*=y; }.m
0370: 69 6e 74 20 6f 70 65 72 61 74 6f 72 2f 28 6d 69 int operator/(mi
0380: 6e 74 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 nt x, mint y) {
0390: 72 65 74 75 72 6e 20 78 2f 3d 79 3b 20 7d 0a 76 return x/=y; }.v
03a0: 65 63 74 6f 72 3c 6d 69 6e 74 3e 20 46 41 43 5f ector<mint> FAC_
03b0: 28 31 2c 31 29 3b 0a 6d 69 6e 74 20 46 41 43 28 (1,1);.mint FAC(
03c0: 4c 4c 20 6e 29 20 7b 20 77 68 69 6c 65 28 20 46 LL n) { while( F
03d0: 41 43 5f 2e 73 69 7a 65 28 29 3c 3d 6e 20 29 20 AC_.size()<=n )
03e0: 46 41 43 5f 2e 70 75 73 68 5f 62 61 63 6b 28 20 FAC_.push_back(
03f0: 46 41 43 5f 2e 62 61 63 6b 28 29 2a 46 41 43 5f FAC_.back()*FAC_
0400: 2e 73 69 7a 65 28 29 20 29 3b 20 72 65 74 75 72 .size() ); retur
0410: 6e 20 46 41 43 5f 5b 6e 5d 3b 20 7d 0a 6d 69 6e n FAC_[n]; }.min
0420: 74 20 43 28 4c 4c 20 6e 2c 20 4c 4c 20 6b 29 20 t C(LL n, LL k)
0430: 7b 20 72 65 74 75 72 6e 20 6b 3c 30 20 7c 7c 20 { return k<0 ||
0440: 6e 3c 6b 20 3f 20 30 20 3a 20 46 41 43 28 6e 29 n<k ? 0 : FAC(n)
0450: 20 2f 20 28 46 41 43 28 6b 29 20 2a 20 46 41 43 / (FAC(k) * FAC
0460: 28 6e 2d 6b 29 29 3b 20 7d 0a 76 65 63 74 6f 72 (n-k)); }.vector
0470: 3c 20 76 65 63 74 6f 72 3c 6d 69 6e 74 3e 20 3e < vector<mint> >
0480: 20 43 50 5f 3b 20 2f 2f 20 50 61 73 63 61 6c 27 CP_; // Pascal'
0490: 73 20 74 72 69 61 6e 67 6c 65 3a 20 69 66 20 4f s triangle: if O
04a0: 28 31 29 20 6e 43 6b 20 69 73 20 6e 65 65 64 65 (1) nCk is neede
04b0: 64 2e 0a 6d 69 6e 74 20 43 28 4c 4c 20 6e 2c 20 d..mint C(LL n,
04c0: 4c 4c 20 6b 29 20 7b 0a 09 77 68 69 6c 65 28 20 LL k) {..while(
04d0: 43 50 5f 2e 73 69 7a 65 28 29 20 3c 3d 20 6e 20 CP_.size() <= n
04e0: 29 20 7b 0a 09 09 69 6e 74 20 6e 6e 20 3d 20 43 ) {...int nn = C
04f0: 50 5f 2e 73 69 7a 65 28 29 3b 0a 09 09 43 50 5f P_.size();...CP_
0500: 2e 70 75 73 68 5f 62 61 63 6b 28 76 65 63 74 6f .push_back(vecto
0510: 72 3c 6d 69 6e 74 3e 28 6e 6e 2b 31 2c 31 29 29 r<mint>(nn+1,1))
0520: 3b 0a 09 09 66 6f 72 28 69 6e 74 20 6b 6b 3d 31 ;...for(int kk=1
0530: 3b 20 6b 6b 3c 6e 6e 3b 20 2b 2b 6b 6b 29 0a 09 ; kk<nn; ++kk)..
0540: 09 09 43 50 5f 5b 6e 6e 5d 5b 6b 6b 5d 20 3d 20 ..CP_[nn][kk] =
0550: 43 50 5f 5b 6e 6e 2d 31 5d 5b 6b 6b 2d 31 5d 20 CP_[nn-1][kk-1]
0560: 2b 20 43 50 5f 5b 6e 6e 2d 31 5d 5b 6b 6b 5d 3b + CP_[nn-1][kk];
0570: 0a 09 7d 0a 09 72 65 74 75 72 6e 20 6b 3c 30 20 ..}..return k<0
0580: 7c 7c 20 6e 3c 6b 20 3f 20 30 20 3a 20 43 50 5f || n<k ? 0 : CP_
0590: 5b 6e 5d 5b 6b 5d 3b 0a 7d 0a 0a 2f 2a 0a 2f 2f [n][k];.}../*.//
05a0: 20 4d 4f 44 56 41 4c 20 6d 75 73 74 20 62 65 20 MODVAL must be
05b0: 61 20 70 72 69 6d 65 21 21 0a 4c 4c 20 47 53 53 a prime!!.LL GSS
05c0: 28 4c 4c 20 6b 2c 20 4c 4c 20 62 2c 20 4c 4c 20 (LL k, LL b, LL
05d0: 65 29 20 2f 2f 20 6b 5e 62 20 2b 20 6b 5e 62 2b e) // k^b + k^b+
05e0: 31 20 2b 20 2e 2e 2e 20 2b 20 6b 5e 65 0a 7b 0a 1 + ... + k^e.{.
05f0: 09 69 66 28 20 62 20 3e 20 20 65 20 29 20 72 65 .if( b > e ) re
0600: 74 75 72 6e 20 30 3b 0a 09 69 66 28 20 6b 20 3c turn 0;..if( k <
0610: 3d 20 31 20 29 20 72 65 74 75 72 6e 20 6b 2a 28 = 1 ) return k*(
0620: 65 2d 62 2b 31 29 3b 0a 09 72 65 74 75 72 6e 20 e-b+1);..return
0630: 44 49 56 28 53 55 42 28 50 4f 57 28 6b 2c 20 65 DIV(SUB(POW(k, e
0640: 2b 31 29 2c 20 50 4f 57 28 6b 2c 62 29 29 2c 20 +1), POW(k,b)),
0650: 6b 2d 31 29 3b 0a 7d 0a 0a 4c 4c 20 43 70 61 73 k-1);.}..LL Cpas
0660: 63 61 6c 28 4c 4c 20 6e 2c 20 4c 4c 20 6b 29 0a cal(LL n, LL k).
0670: 7b 0a 09 76 65 63 74 6f 72 3c 20 76 65 63 74 6f {..vector< vecto
0680: 72 3c 4c 4c 3e 20 3e 20 63 28 6e 2b 31 2c 20 76 r<LL> > c(n+1, v
0690: 65 63 74 6f 72 3c 4c 4c 3e 28 6b 2b 31 29 29 3b ector<LL>(k+1));
06a0: 0a 09 66 6f 72 28 4c 4c 20 6e 6e 3d 31 3b 20 6e ..for(LL nn=1; n
06b0: 6e 3c 3d 6e 3b 20 2b 2b 6e 6e 29 0a 09 09 66 6f n<=n; ++nn)...fo
06c0: 72 28 4c 4c 20 6b 6b 3d 30 3b 20 6b 6b 3c 3d 6d r(LL kk=0; kk<=m
06d0: 69 6e 28 6e 6e 2c 6b 29 3b 20 2b 2b 6b 6b 29 0a in(nn,k); ++kk).
06e0: 09 09 09 63 5b 6e 6e 5d 5b 6b 6b 5d 20 3d 20 6b ...c[nn][kk] = k
06f0: 6b 3d 3d 30 20 7c 7c 20 6b 6b 3d 3d 6e 6e 20 3f k==0 || kk==nn ?
0700: 20 31 20 3a 20 41 44 44 28 63 5b 6e 6e 2d 31 5d 1 : ADD(c[nn-1]
0710: 5b 6b 6b 2d 31 5d 2c 20 63 5b 6e 6e 2d 31 5d 5b [kk-1], c[nn-1][
0720: 6b 6b 5d 29 3b 0a 09 72 65 74 75 72 6e 20 63 5b kk]);..return c[
0730: 6e 5d 5b 6b 5d 3b 0a 7d 0a 0a 76 65 63 74 6f 72 n][k];.}..vector
0740: 3c 20 76 65 63 74 6f 72 3c 4c 4c 3e 20 3e 20 4d < vector<LL> > M
0750: 41 54 4d 55 4c 28 76 65 63 74 6f 72 3c 20 76 65 ATMUL(vector< ve
0760: 63 74 6f 72 3c 4c 4c 3e 20 3e 26 20 61 2c 20 76 ctor<LL> >& a, v
0770: 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c 4c 4c ector< vector<LL
0780: 3e 20 3e 26 20 62 29 0a 7b 0a 20 20 20 69 6e 74 > >& b).{. int
0790: 20 4e 20 3d 20 61 2e 73 69 7a 65 28 29 3b 0a 20 N = a.size();.
07a0: 20 20 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 vector< vector
07b0: 3c 4c 4c 3e 20 3e 20 63 28 4e 2c 20 76 65 63 74 <LL> > c(N, vect
07c0: 6f 72 3c 4c 4c 3e 28 4e 29 29 3b 0a 20 20 20 66 or<LL>(N));. f
07d0: 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 3c 4e 3b or(int i=0; i<N;
07e0: 20 2b 2b 69 29 0a 20 20 20 66 6f 72 28 69 6e 74 ++i). for(int
07f0: 20 6a 3d 30 3b 20 6a 3c 4e 3b 20 2b 2b 6a 29 0a j=0; j<N; ++j).
0800: 20 20 20 66 6f 72 28 69 6e 74 20 6b 3d 30 3b 20 for(int k=0;
0810: 6b 3c 4e 3b 20 2b 2b 6b 29 0a 20 20 20 20 20 20 k<N; ++k).
0820: 63 5b 69 5d 5b 6a 5d 20 3d 20 41 44 44 28 63 5b c[i][j] = ADD(c[
0830: 69 5d 5b 6a 5d 2c 20 4d 55 4c 28 61 5b 69 5d 5b i][j], MUL(a[i][
0840: 6b 5d 2c 62 5b 6b 5d 5b 6a 5d 29 29 3b 0a 20 20 k],b[k][j]));.
0850: 20 72 65 74 75 72 6e 20 63 3b 0a 7d 0a 0a 2f 2f return c;.}..//
0860: 20 77 6f 72 6b 73 20 66 6f 72 20 6e 6f 6e 2d 70 works for non-p
0870: 72 69 6d 65 20 4d 4f 44 56 41 4c 0a 4c 4c 20 47 rime MODVAL.LL G
0880: 45 4f 28 4c 4c 20 78 5f 2c 20 4c 4c 20 65 29 20 EO(LL x_, LL e)
0890: 2f 2f 20 78 5e 30 20 2b 20 78 5e 31 20 2b 20 2e // x^0 + x^1 + .
08a0: 2e 2e 20 2b 20 78 5e 65 2d 31 0a 7b 0a 20 20 20 .. + x^e-1.{.
08b0: 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c 4c vector< vector<L
08c0: 4c 3e 20 3e 20 76 28 32 2c 20 76 65 63 74 6f 72 L> > v(2, vector
08d0: 3c 4c 4c 3e 28 32 29 29 3b 0a 20 20 20 76 65 63 <LL>(2));. vec
08e0: 74 6f 72 3c 20 76 65 63 74 6f 72 3c 4c 4c 3e 20 tor< vector<LL>
08f0: 3e 20 78 28 32 2c 20 76 65 63 74 6f 72 3c 4c 4c > x(2, vector<LL
0900: 3e 28 32 29 29 3b 0a 20 20 20 76 5b 30 5d 5b 30 >(2));. v[0][0
0910: 5d 20 3d 20 76 5b 31 5d 5b 31 5d 20 3d 20 31 3b ] = v[1][1] = 1;
0920: 0a 20 20 20 78 5b 30 5d 5b 30 5d 20 3d 20 78 5f . x[0][0] = x_
0930: 3b 20 78 5b 30 5d 5b 31 5d 20 3d 20 30 3b 0a 20 ; x[0][1] = 0;.
0940: 20 20 78 5b 31 5d 5b 30 5d 20 3d 20 31 20 3b 20 x[1][0] = 1 ;
0950: 78 5b 31 5d 5b 31 5d 20 3d 20 31 3b 0a 20 20 20 x[1][1] = 1;.
0960: 66 6f 72 28 3b 65 3b 78 3d 4d 41 54 4d 55 4c 28 for(;e;x=MATMUL(
0970: 78 2c 78 29 2c 65 3e 3e 3d 31 29 0a 20 20 20 20 x,x),e>>=1).
0980: 20 20 69 66 28 65 26 31 29 0a 20 20 20 20 20 20 if(e&1).
0990: 20 20 20 76 20 3d 20 4d 41 54 4d 55 4c 28 76 2c v = MATMUL(v,
09a0: 20 78 29 3b 0a 20 20 20 72 65 74 75 72 6e 20 76 x);. return v
09b0: 5b 31 5d 5b 30 5d 3b 0a 7d 0a 0a 2f 2f 20 77 6f [1][0];.}..// wo
09c0: 72 6b 73 20 66 6f 72 20 6e 6f 6e 2d 70 72 69 6d rks for non-prim
09d0: 65 20 4d 4f 44 56 41 4c 0a 4c 4c 20 48 59 50 28 e MODVAL.LL HYP(
09e0: 4c 4c 20 78 5f 2c 20 4c 4c 20 65 29 20 2f 2f 20 LL x_, LL e) //
09f0: 65 20 78 5e 30 20 2b 20 28 65 2d 31 29 20 78 5e e x^0 + (e-1) x^
0a00: 31 20 2b 20 2e 2e 2e 20 2b 20 31 20 78 5e 65 2d 1 + ... + 1 x^e-
0a10: 31 20 3d 20 47 45 4f 28 78 2c 31 29 2b 47 45 4f 1 = GEO(x,1)+GEO
0a20: 28 78 2c 32 29 2b 2e 2e 2e 2b 47 45 4f 28 78 2c (x,2)+...+GEO(x,
0a30: 65 29 0a 7b 0a 20 20 20 76 65 63 74 6f 72 3c 20 e).{. vector<
0a40: 76 65 63 74 6f 72 3c 4c 4c 3e 20 3e 20 76 28 33 vector<LL> > v(3
0a50: 2c 20 76 65 63 74 6f 72 3c 4c 4c 3e 28 33 29 29 , vector<LL>(3))
0a60: 3b 0a 20 20 20 76 65 63 74 6f 72 3c 20 76 65 63 ;. vector< vec
0a70: 74 6f 72 3c 4c 4c 3e 20 3e 20 78 28 33 2c 20 76 tor<LL> > x(3, v
0a80: 65 63 74 6f 72 3c 4c 4c 3e 28 33 29 29 3b 0a 20 ector<LL>(3));.
0a90: 20 20 76 5b 30 5d 5b 30 5d 20 3d 20 76 5b 31 5d v[0][0] = v[1]
0aa0: 5b 31 5d 20 3d 20 76 5b 32 5d 5b 32 5d 20 3d 20 [1] = v[2][2] =
0ab0: 31 3b 0a 20 20 20 78 5b 30 5d 5b 30 5d 20 3d 20 1;. x[0][0] =
0ac0: 78 5f 3b 20 78 5b 30 5d 5b 31 5d 20 3d 20 30 3b x_; x[0][1] = 0;
0ad0: 20 78 5b 30 5d 5b 32 5d 20 3d 20 30 3b 0a 20 20 x[0][2] = 0;.
0ae0: 20 78 5b 31 5d 5b 30 5d 20 3d 20 31 20 3b 20 78 x[1][0] = 1 ; x
0af0: 5b 31 5d 5b 31 5d 20 3d 20 31 3b 20 78 5b 31 5d [1][1] = 1; x[1]
0b00: 5b 32 5d 20 3d 20 30 3b 0a 20 20 20 78 5b 32 5d [2] = 0;. x[2]
0b10: 5b 30 5d 20 3d 20 30 20 3b 20 78 5b 32 5d 5b 31 [0] = 0 ; x[2][1
0b20: 5d 20 3d 20 31 3b 20 78 5b 32 5d 5b 32 5d 20 3d ] = 1; x[2][2] =
0b30: 20 31 3b 0a 20 20 20 65 2b 2b 3b 0a 20 20 20 66 1;. e++;. f
0b40: 6f 72 28 3b 65 3b 78 3d 4d 41 54 4d 55 4c 28 78 or(;e;x=MATMUL(x
0b50: 2c 78 29 2c 65 3e 3e 3d 31 29 0a 20 20 20 20 20 ,x),e>>=1).
0b60: 20 69 66 28 65 26 31 29 0a 20 20 20 20 20 20 20 if(e&1).
0b70: 20 20 76 20 3d 20 4d 41 54 4d 55 4c 28 76 2c 20 v = MATMUL(v,
0b80: 78 29 3b 0a 20 20 20 72 65 74 75 72 6e 20 76 5b x);. return v[
0b90: 32 5d 5b 30 5d 3b 0a 7d 0a 2a 2f 0a 2][0];.}.*/.