Artifact b348fa4dbce4653ddbc0d87a269b0de4f6fb164e:
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 2d TCO10 R3 LV3.//-
0080: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0090: 2d 2d 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 0a 0a 73 74 ------------..st
00c0: 61 74 69 63 20 63 6f 6e 73 74 20 69 6e 74 20 4d atic const int M
00d0: 4f 44 56 41 4c 20 3d 20 31 30 30 30 30 30 30 30 ODVAL = 10000000
00e0: 30 37 3b 20 2f 2f 20 6f 70 2f 20 64 65 70 65 6e 07; // op/ depen
00f0: 64 73 20 6f 6e 20 70 72 69 6d 61 72 69 74 79 20 ds on primarity
0100: 6f 66 20 74 68 65 20 4d 4f 44 56 41 4c 0a 73 74 of the MODVAL.st
0110: 72 75 63 74 20 6d 69 6e 74 0a 7b 0a 09 69 6e 74 ruct mint.{..int
0120: 20 76 61 6c 3b 0a 09 6d 69 6e 74 28 29 3a 76 61 val;..mint():va
0130: 6c 28 30 29 7b 7d 0a 09 6d 69 6e 74 28 69 6e 74 l(0){}..mint(int
0140: 20 78 29 3a 76 61 6c 28 78 25 4d 4f 44 56 41 4c x):val(x%MODVAL
0150: 29 20 7b 7d 0a 09 6d 69 6e 74 28 4c 4c 20 20 78 ) {}..mint(LL x
0160: 29 3a 76 61 6c 28 78 25 4d 4f 44 56 41 4c 29 20 ):val(x%MODVAL)
0170: 7b 7d 0a 7d 3b 0a 0a 6d 69 6e 74 20 6f 70 65 72 {}.};..mint oper
0180: 61 74 6f 72 2b 28 6d 69 6e 74 20 78 2c 20 6d 69 ator+(mint x, mi
0190: 6e 74 20 79 29 20 7b 20 72 65 74 75 72 6e 20 78 nt y) { return x
01a0: 2e 76 61 6c 2b 79 2e 76 61 6c 3b 20 7d 0a 6d 69 .val+y.val; }.mi
01b0: 6e 74 20 6f 70 65 72 61 74 6f 72 2d 28 6d 69 6e nt operator-(min
01c0: 74 20 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 72 t x, mint y) { r
01d0: 65 74 75 72 6e 20 78 2e 76 61 6c 2d 79 2e 76 61 eturn x.val-y.va
01e0: 6c 2b 4d 4f 44 56 41 4c 3b 20 7d 0a 6d 69 6e 74 l+MODVAL; }.mint
01f0: 20 6f 70 65 72 61 74 6f 72 2a 28 6d 69 6e 74 20 operator*(mint
0200: 78 2c 20 6d 69 6e 74 20 79 29 20 7b 20 72 65 74 x, mint y) { ret
0210: 75 72 6e 20 4c 4c 28 78 2e 76 61 6c 29 2a 79 2e urn LL(x.val)*y.
0220: 76 61 6c 3b 20 7d 0a 6d 69 6e 74 20 50 4f 57 28 val; }.mint POW(
0230: 6d 69 6e 74 20 78 2c 20 69 6e 74 20 65 29 20 7b mint x, int e) {
0240: 0a 09 6d 69 6e 74 20 76 20 3d 20 31 3b 0a 09 66 ..mint v = 1;..f
0250: 6f 72 28 3b 65 3b 78 3d 78 2a 78 2c 65 3e 3e 3d or(;e;x=x*x,e>>=
0260: 31 29 0a 09 09 69 66 28 65 26 31 29 0a 09 09 09 1)...if(e&1)....
0270: 76 3d 76 2a 78 3b 0a 09 72 65 74 75 72 6e 20 76 v=v*x;..return v
0280: 3b 0a 7d 0a 6d 69 6e 74 20 6f 70 65 72 61 74 6f ;.}.mint operato
0290: 72 2f 28 6d 69 6e 74 20 78 2c 20 6d 69 6e 74 20 r/(mint x, mint
02a0: 79 29 20 7b 20 72 65 74 75 72 6e 20 78 20 2a 20 y) { return x *
02b0: 50 4f 57 28 79 2c 20 4d 4f 44 56 41 4c 2d 32 29 POW(y, MODVAL-2)
02c0: 3b 20 7d 0a 0a 76 65 63 74 6f 72 3c 6d 69 6e 74 ; }..vector<mint
02d0: 3e 20 46 41 43 5f 28 31 2c 31 29 3b 0a 76 6f 69 > FAC_(1,1);.voi
02e0: 64 20 46 41 43 5f 49 4e 49 54 28 69 6e 74 20 6e d FAC_INIT(int n
02f0: 29 20 7b 20 66 6f 72 28 69 6e 74 20 69 3d 31 3b ) { for(int i=1;
0300: 20 69 3c 3d 6e 3b 20 2b 2b 69 29 20 46 41 43 5f i<=n; ++i) FAC_
0310: 2e 70 75 73 68 5f 62 61 63 6b 28 20 46 41 43 5f .push_back( FAC_
0320: 2e 62 61 63 6b 28 29 2a 69 20 29 3b 20 7d 0a 6d .back()*i ); }.m
0330: 69 6e 74 20 46 41 43 28 6d 69 6e 74 20 78 29 20 int FAC(mint x)
0340: 20 20 20 20 20 20 7b 20 72 65 74 75 72 6e 20 46 { return F
0350: 41 43 5f 5b 78 2e 76 61 6c 5d 3b 20 7d 0a 6d 69 AC_[x.val]; }.mi
0360: 6e 74 20 43 28 6d 69 6e 74 20 6e 2c 20 6d 69 6e nt C(mint n, min
0370: 74 20 6b 29 20 7b 20 72 65 74 75 72 6e 20 6b 2e t k) { return k.
0380: 76 61 6c 3c 30 20 7c 7c 20 6e 2e 76 61 6c 3c 6b val<0 || n.val<k
0390: 2e 76 61 6c 20 3f 20 30 20 3a 20 46 41 43 28 6e .val ? 0 : FAC(n
03a0: 29 20 2f 20 28 46 41 43 28 6b 29 20 2a 20 46 41 ) / (FAC(k) * FA
03b0: 43 28 6e 2d 6b 29 29 3b 20 7d 0a 0a 2f 2a 0a 2f C(n-k)); }../*./
03c0: 2f 20 4d 4f 44 56 41 4c 20 6d 75 73 74 20 62 65 / MODVAL must be
03d0: 20 61 20 70 72 69 6d 65 21 21 0a 4c 4c 20 47 53 a prime!!.LL GS
03e0: 53 28 4c 4c 20 6b 2c 20 4c 4c 20 62 2c 20 4c 4c S(LL k, LL b, LL
03f0: 20 65 29 20 2f 2f 20 6b 5e 62 20 2b 20 6b 5e 62 e) // k^b + k^b
0400: 2b 31 20 2b 20 2e 2e 2e 20 2b 20 6b 5e 65 0a 7b +1 + ... + k^e.{
0410: 0a 09 69 66 28 20 62 20 3e 20 20 65 20 29 20 72 ..if( b > e ) r
0420: 65 74 75 72 6e 20 30 3b 0a 09 69 66 28 20 6b 20 eturn 0;..if( k
0430: 3c 3d 20 31 20 29 20 72 65 74 75 72 6e 20 6b 2a <= 1 ) return k*
0440: 28 65 2d 62 2b 31 29 3b 0a 09 72 65 74 75 72 6e (e-b+1);..return
0450: 20 44 49 56 28 53 55 42 28 50 4f 57 28 6b 2c 20 DIV(SUB(POW(k,
0460: 65 2b 31 29 2c 20 50 4f 57 28 6b 2c 62 29 29 2c e+1), POW(k,b)),
0470: 20 6b 2d 31 29 3b 0a 7d 0a 0a 4c 4c 20 43 70 61 k-1);.}..LL Cpa
0480: 73 63 61 6c 28 4c 4c 20 6e 2c 20 4c 4c 20 6b 29 scal(LL n, LL k)
0490: 0a 7b 0a 09 76 65 63 74 6f 72 3c 20 76 65 63 74 .{..vector< vect
04a0: 6f 72 3c 4c 4c 3e 20 3e 20 63 28 6e 2b 31 2c 20 or<LL> > c(n+1,
04b0: 76 65 63 74 6f 72 3c 4c 4c 3e 28 6b 2b 31 29 29 vector<LL>(k+1))
04c0: 3b 0a 09 66 6f 72 28 4c 4c 20 6e 6e 3d 31 3b 20 ;..for(LL nn=1;
04d0: 6e 6e 3c 3d 6e 3b 20 2b 2b 6e 6e 29 0a 09 09 66 nn<=n; ++nn)...f
04e0: 6f 72 28 4c 4c 20 6b 6b 3d 30 3b 20 6b 6b 3c 3d or(LL kk=0; kk<=
04f0: 6d 69 6e 28 6e 6e 2c 6b 29 3b 20 2b 2b 6b 6b 29 min(nn,k); ++kk)
0500: 0a 09 09 09 63 5b 6e 6e 5d 5b 6b 6b 5d 20 3d 20 ....c[nn][kk] =
0510: 6b 6b 3d 3d 30 20 7c 7c 20 6b 6b 3d 3d 6e 6e 20 kk==0 || kk==nn
0520: 3f 20 31 20 3a 20 41 44 44 28 63 5b 6e 6e 2d 31 ? 1 : ADD(c[nn-1
0530: 5d 5b 6b 6b 2d 31 5d 2c 20 63 5b 6e 6e 2d 31 5d ][kk-1], c[nn-1]
0540: 5b 6b 6b 5d 29 3b 0a 09 72 65 74 75 72 6e 20 63 [kk]);..return c
0550: 5b 6e 5d 5b 6b 5d 3b 0a 7d 0a 0a 76 65 63 74 6f [n][k];.}..vecto
0560: 72 3c 20 76 65 63 74 6f 72 3c 4c 4c 3e 20 3e 20 r< vector<LL> >
0570: 4d 41 54 4d 55 4c 28 76 65 63 74 6f 72 3c 20 76 MATMUL(vector< v
0580: 65 63 74 6f 72 3c 4c 4c 3e 20 3e 26 20 61 2c 20 ector<LL> >& a,
0590: 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c 4c vector< vector<L
05a0: 4c 3e 20 3e 26 20 62 29 0a 7b 0a 20 20 20 69 6e L> >& b).{. in
05b0: 74 20 4e 20 3d 20 61 2e 73 69 7a 65 28 29 3b 0a t N = a.size();.
05c0: 20 20 20 76 65 63 74 6f 72 3c 20 76 65 63 74 6f vector< vecto
05d0: 72 3c 4c 4c 3e 20 3e 20 63 28 4e 2c 20 76 65 63 r<LL> > c(N, vec
05e0: 74 6f 72 3c 4c 4c 3e 28 4e 29 29 3b 0a 20 20 20 tor<LL>(N));.
05f0: 66 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 3c 4e for(int i=0; i<N
0600: 3b 20 2b 2b 69 29 0a 20 20 20 66 6f 72 28 69 6e ; ++i). for(in
0610: 74 20 6a 3d 30 3b 20 6a 3c 4e 3b 20 2b 2b 6a 29 t j=0; j<N; ++j)
0620: 0a 20 20 20 66 6f 72 28 69 6e 74 20 6b 3d 30 3b . for(int k=0;
0630: 20 6b 3c 4e 3b 20 2b 2b 6b 29 0a 20 20 20 20 20 k<N; ++k).
0640: 20 63 5b 69 5d 5b 6a 5d 20 3d 20 41 44 44 28 63 c[i][j] = ADD(c
0650: 5b 69 5d 5b 6a 5d 2c 20 4d 55 4c 28 61 5b 69 5d [i][j], MUL(a[i]
0660: 5b 6b 5d 2c 62 5b 6b 5d 5b 6a 5d 29 29 3b 0a 20 [k],b[k][j]));.
0670: 20 20 72 65 74 75 72 6e 20 63 3b 0a 7d 0a 0a 2f return c;.}../
0680: 2f 20 77 6f 72 6b 73 20 66 6f 72 20 6e 6f 6e 2d / works for non-
0690: 70 72 69 6d 65 20 4d 4f 44 56 41 4c 0a 4c 4c 20 prime MODVAL.LL
06a0: 47 45 4f 28 4c 4c 20 78 5f 2c 20 4c 4c 20 65 29 GEO(LL x_, LL e)
06b0: 20 2f 2f 20 78 5e 30 20 2b 20 78 5e 31 20 2b 20 // x^0 + x^1 +
06c0: 2e 2e 2e 20 2b 20 78 5e 65 2d 31 0a 7b 0a 20 20 ... + x^e-1.{.
06d0: 20 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c vector< vector<
06e0: 4c 4c 3e 20 3e 20 76 28 32 2c 20 76 65 63 74 6f LL> > v(2, vecto
06f0: 72 3c 4c 4c 3e 28 32 29 29 3b 0a 20 20 20 76 65 r<LL>(2));. ve
0700: 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c 4c 4c 3e ctor< vector<LL>
0710: 20 3e 20 78 28 32 2c 20 76 65 63 74 6f 72 3c 4c > x(2, vector<L
0720: 4c 3e 28 32 29 29 3b 0a 20 20 20 76 5b 30 5d 5b L>(2));. v[0][
0730: 30 5d 20 3d 20 76 5b 31 5d 5b 31 5d 20 3d 20 31 0] = v[1][1] = 1
0740: 3b 0a 20 20 20 78 5b 30 5d 5b 30 5d 20 3d 20 78 ;. x[0][0] = x
0750: 5f 3b 20 78 5b 30 5d 5b 31 5d 20 3d 20 30 3b 0a _; x[0][1] = 0;.
0760: 20 20 20 78 5b 31 5d 5b 30 5d 20 3d 20 31 20 3b x[1][0] = 1 ;
0770: 20 78 5b 31 5d 5b 31 5d 20 3d 20 31 3b 0a 20 20 x[1][1] = 1;.
0780: 20 66 6f 72 28 3b 65 3b 78 3d 4d 41 54 4d 55 4c for(;e;x=MATMUL
0790: 28 78 2c 78 29 2c 65 3e 3e 3d 31 29 0a 20 20 20 (x,x),e>>=1).
07a0: 20 20 20 69 66 28 65 26 31 29 0a 20 20 20 20 20 if(e&1).
07b0: 20 20 20 20 76 20 3d 20 4d 41 54 4d 55 4c 28 76 v = MATMUL(v
07c0: 2c 20 78 29 3b 0a 20 20 20 72 65 74 75 72 6e 20 , x);. return
07d0: 76 5b 31 5d 5b 30 5d 3b 0a 7d 0a 0a 2f 2f 20 77 v[1][0];.}..// w
07e0: 6f 72 6b 73 20 66 6f 72 20 6e 6f 6e 2d 70 72 69 orks for non-pri
07f0: 6d 65 20 4d 4f 44 56 41 4c 0a 4c 4c 20 48 59 50 me MODVAL.LL HYP
0800: 28 4c 4c 20 78 5f 2c 20 4c 4c 20 65 29 20 2f 2f (LL x_, LL e) //
0810: 20 65 20 78 5e 30 20 2b 20 28 65 2d 31 29 20 78 e x^0 + (e-1) x
0820: 5e 31 20 2b 20 2e 2e 2e 20 2b 20 31 20 78 5e 65 ^1 + ... + 1 x^e
0830: 2d 31 20 3d 20 47 45 4f 28 78 2c 31 29 2b 47 45 -1 = GEO(x,1)+GE
0840: 4f 28 78 2c 32 29 2b 2e 2e 2e 2b 47 45 4f 28 78 O(x,2)+...+GEO(x
0850: 2c 65 29 0a 7b 0a 20 20 20 76 65 63 74 6f 72 3c ,e).{. vector<
0860: 20 76 65 63 74 6f 72 3c 4c 4c 3e 20 3e 20 76 28 vector<LL> > v(
0870: 33 2c 20 76 65 63 74 6f 72 3c 4c 4c 3e 28 33 29 3, vector<LL>(3)
0880: 29 3b 0a 20 20 20 76 65 63 74 6f 72 3c 20 76 65 );. vector< ve
0890: 63 74 6f 72 3c 4c 4c 3e 20 3e 20 78 28 33 2c 20 ctor<LL> > x(3,
08a0: 76 65 63 74 6f 72 3c 4c 4c 3e 28 33 29 29 3b 0a vector<LL>(3));.
08b0: 20 20 20 76 5b 30 5d 5b 30 5d 20 3d 20 76 5b 31 v[0][0] = v[1
08c0: 5d 5b 31 5d 20 3d 20 76 5b 32 5d 5b 32 5d 20 3d ][1] = v[2][2] =
08d0: 20 31 3b 0a 20 20 20 78 5b 30 5d 5b 30 5d 20 3d 1;. x[0][0] =
08e0: 20 78 5f 3b 20 78 5b 30 5d 5b 31 5d 20 3d 20 30 x_; x[0][1] = 0
08f0: 3b 20 78 5b 30 5d 5b 32 5d 20 3d 20 30 3b 0a 20 ; x[0][2] = 0;.
0900: 20 20 78 5b 31 5d 5b 30 5d 20 3d 20 31 20 3b 20 x[1][0] = 1 ;
0910: 78 5b 31 5d 5b 31 5d 20 3d 20 31 3b 20 78 5b 31 x[1][1] = 1; x[1
0920: 5d 5b 32 5d 20 3d 20 30 3b 0a 20 20 20 78 5b 32 ][2] = 0;. x[2
0930: 5d 5b 30 5d 20 3d 20 30 20 3b 20 78 5b 32 5d 5b ][0] = 0 ; x[2][
0940: 31 5d 20 3d 20 31 3b 20 78 5b 32 5d 5b 32 5d 20 1] = 1; x[2][2]
0950: 3d 20 31 3b 0a 20 20 20 65 2b 2b 3b 0a 20 20 20 = 1;. e++;.
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 32 5d 5b 30 5d 3b 0a 7d 0a 2a 2f 0a [2][0];.}.*/.