Artifact 148a1a19572b802a21ab07455141773892dec42f:
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 63 6c 61 73 73 ong LL;....class
0140: 20 53 75 6d 4f 66 50 6f 77 65 72 73 0d 0a 7b 0d SumOfPowers..{.
0150: 0a 70 75 62 6c 69 63 3a 0d 0a 09 69 6e 74 20 76 .public:...int v
0160: 61 6c 75 65 28 69 6e 74 20 6e 2c 20 69 6e 74 20 alue(int n, int
0170: 6b 29 20 0d 0a 09 7b 0d 0a 09 09 2f 2f 20 20 32 k) ...{....// 2
0180: 5e 6d 20 2d 20 31 5e 6d 20 3d 20 28 31 2b 31 29 ^m - 1^m = (1+1)
0190: 5e 6d 20 2d 20 31 5e 6d 20 3d 20 6d 43 31 20 31 ^m - 1^m = mC1 1
01a0: 5e 28 6d 2d 31 29 20 2b 20 2e 2e 2e 20 2b 20 6d ^(m-1) + ... + m
01b0: 43 69 20 31 5e 28 6d 2d 69 29 20 2b 20 2e 2e 2e Ci 1^(m-i) + ...
01c0: 20 2b 20 6d 43 6d 20 31 5e 30 0d 0a 09 09 2f 2f + mCm 1^0....//
01d0: 20 20 33 5e 6d 20 2d 20 32 5e 6d 20 3d 20 28 32 3^m - 2^m = (2
01e0: 2b 31 29 5e 6d 20 2d 20 31 5e 6d 20 3d 20 6d 43 +1)^m - 1^m = mC
01f0: 31 20 32 5e 28 6d 2d 31 29 20 2b 20 2e 2e 2e 20 1 2^(m-1) + ...
0200: 2b 20 6d 43 69 20 32 5e 28 6d 2d 69 29 20 2b 20 + mCi 2^(m-i) +
0210: 2e 2e 2e 20 2b 20 6d 43 6d 20 32 5e 30 0d 0a 09 ... + mCm 2^0...
0220: 09 2f 2f 2b 20 2e 2e 2e 0d 0a 09 09 2f 2f 2d 2d .//+ .......//--
0230: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0240: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0250: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0260: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0270: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 0d 0a --------------..
0280: 09 09 2f 2f 20 20 28 6e 2b 31 29 5e 6d 20 2d 20 ..// (n+1)^m -
0290: 31 20 20 3d 20 6d 43 31 20 5c 53 69 67 6d 61 20 1 = mC1 \Sigma
02a0: 6e 5e 28 6d 2d 31 29 20 2b 20 2e 2e 2e 20 2b 20 n^(m-1) + ... +
02b0: 6d 43 69 20 5c 53 69 67 6d 61 20 6e 5e 28 6d 2d mCi \Sigma n^(m-
02c0: 69 29 20 2b 20 2e 2e 2e 20 2b 20 6d 43 6d 20 5c i) + ... + mCm \
02d0: 53 69 67 6d 61 20 6e 5e 30 0d 0a 0d 0a 09 09 76 Sigma n^0......v
02e0: 65 63 74 6f 72 3c 4c 4c 3e 20 76 61 6c 73 3b 0d ector<LL> vals;.
02f0: 0a 09 09 66 6f 72 28 69 6e 74 20 6d 3d 31 3b 20 ...for(int m=1;
0300: 6d 3c 3d 6b 2b 31 3b 20 2b 2b 6d 29 0d 0a 09 09 m<=k+1; ++m)....
0310: 7b 0d 0a 09 09 09 4c 4c 20 78 20 3d 20 53 55 42 {.....LL x = SUB
0320: 28 50 4f 57 28 6e 2b 31 2c 20 6d 29 2c 20 31 29 (POW(n+1, m), 1)
0330: 3b 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 69 3d ;.....for(int i=
0340: 32 3b 20 69 3c 3d 6d 3b 20 2b 2b 69 29 0d 0a 09 2; i<=m; ++i)...
0350: 09 09 09 78 20 3d 20 53 55 42 28 20 78 2c 20 4d ...x = SUB( x, M
0360: 55 4c 28 43 28 6d 2c 20 69 29 2c 20 76 61 6c 73 UL(C(m, i), vals
0370: 5b 6d 2d 69 5d 29 20 29 3b 0d 0a 09 09 09 78 20 [m-i]) );.....x
0380: 3d 20 44 49 56 28 78 2c 20 6d 29 3b 0d 0a 09 09 = DIV(x, m);....
0390: 09 76 61 6c 73 2e 70 75 73 68 5f 62 61 63 6b 28 .vals.push_back(
03a0: 78 29 3b 0d 0a 09 09 7d 0d 0a 09 09 72 65 74 75 x);....}....retu
03b0: 72 6e 20 28 69 6e 74 29 76 61 6c 73 2e 62 61 63 rn (int)vals.bac
03c0: 6b 28 29 3b 0d 0a 09 7d 0d 0a 09 4c 4c 20 41 44 k();...}...LL AD
03d0: 44 28 4c 4c 20 78 2c 20 4c 4c 20 79 29 20 7b 20 D(LL x, LL y) {
03e0: 72 65 74 75 72 6e 20 28 78 2b 79 29 25 31 30 30 return (x+y)%100
03f0: 30 30 30 30 30 30 37 3b 20 7d 0d 0a 09 4c 4c 20 0000007; }...LL
0400: 53 55 42 28 4c 4c 20 78 2c 20 4c 4c 20 79 29 20 SUB(LL x, LL y)
0410: 7b 20 72 65 74 75 72 6e 20 28 78 2d 79 2b 31 30 { return (x-y+10
0420: 30 30 30 30 30 30 30 37 29 25 31 30 30 30 30 30 00000007)%100000
0430: 30 30 30 37 3b 20 7d 0d 0a 09 4c 4c 20 4d 55 4c 0007; }...LL MUL
0440: 28 4c 4c 20 78 2c 20 4c 4c 20 79 29 20 7b 20 72 (LL x, LL y) { r
0450: 65 74 75 72 6e 20 28 78 2a 79 29 25 31 30 30 30 eturn (x*y)%1000
0460: 30 30 30 30 30 37 3b 20 7d 0d 0a 09 4c 4c 20 43 000007; }...LL C
0470: 28 4c 4c 20 6e 2c 20 4c 4c 20 6b 29 20 7b 0d 0a (LL n, LL k) {..
0480: 09 09 4c 4c 20 76 20 3d 20 31 3b 0d 0a 09 09 66 ..LL v = 1;....f
0490: 6f 72 28 4c 4c 20 69 3d 31 3b 20 69 3c 3d 6b 3b or(LL i=1; i<=k;
04a0: 20 2b 2b 69 29 0d 0a 09 09 09 76 20 3d 20 44 49 ++i).....v = DI
04b0: 56 28 4d 55 4c 28 76 2c 20 6e 2d 69 2b 31 29 2c V(MUL(v, n-i+1),
04c0: 20 69 29 3b 0d 0a 09 09 72 65 74 75 72 6e 20 76 i);....return v
04d0: 3b 0d 0a 09 7d 0d 0a 09 4c 4c 20 50 4f 57 28 4c ;...}...LL POW(L
04e0: 4c 20 78 2c 20 4c 4c 20 65 29 20 7b 0d 0a 09 09 L x, LL e) {....
04f0: 4c 4c 20 76 20 3d 20 31 3b 0d 0a 09 09 66 6f 72 LL v = 1;....for
0500: 28 3b 65 3b 78 3d 4d 55 4c 28 78 2c 78 29 2c 65 (;e;x=MUL(x,x),e
0510: 3e 3e 3d 31 29 0d 0a 09 09 09 69 66 28 65 26 31 >>=1).....if(e&1
0520: 29 0d 0a 09 09 09 09 76 20 3d 20 4d 55 4c 28 76 )......v = MUL(v
0530: 2c 20 78 29 3b 0d 0a 09 09 72 65 74 75 72 6e 20 , x);....return
0540: 76 3b 0d 0a 09 7d 0d 0a 09 4c 4c 20 44 49 56 28 v;...}...LL DIV(
0550: 4c 4c 20 78 2c 20 4c 4c 20 79 29 20 7b 0d 0a 09 LL x, LL y) {...
0560: 09 4c 4c 20 69 79 2c 20 5f 3b 0d 0a 09 09 78 67 .LL iy, _;....xg
0570: 63 64 28 79 2c 20 31 30 30 30 30 30 30 30 30 37 cd(y, 1000000007
0580: 2c 20 26 69 79 2c 20 26 5f 29 3b 0d 0a 09 09 72 , &iy, &_);....r
0590: 65 74 75 72 6e 20 4d 55 4c 28 78 2c 20 28 69 79 eturn MUL(x, (iy
05a0: 2b 31 30 30 30 30 30 30 30 30 37 29 25 31 30 30 +1000000007)%100
05b0: 30 30 30 30 30 30 37 29 3b 0d 0a 09 7d 0d 0a 09 0000007);...}...
05c0: 4c 4c 20 78 67 63 64 28 4c 4c 20 61 2c 20 4c 4c LL xgcd(LL a, LL
05d0: 20 62 2c 20 4c 4c 2a 20 78 2c 20 4c 4c 2a 20 79 b, LL* x, LL* y
05e0: 29 20 7b 20 2f 2f 20 61 78 2b 62 79 3d 67 0d 0a ) { // ax+by=g..
05f0: 09 09 69 66 28 62 29 20 7b 0d 0a 09 09 09 4c 4c ..if(b) {.....LL
0600: 20 79 79 2c 20 67 20 3d 20 78 67 63 64 28 62 2c yy, g = xgcd(b,
0610: 61 25 62 2c 26 79 79 2c 78 29 3b 0d 0a 09 09 09 a%b,&yy,x);.....
0620: 2a 79 20 3d 20 79 79 20 2d 20 61 2f 62 2a 2a 78 *y = yy - a/b**x
0630: 3b 0d 0a 09 09 09 72 65 74 75 72 6e 20 67 3b 0d ;.....return g;.
0640: 0a 09 09 7d 0d 0a 09 09 65 6c 73 65 20 7b 0d 0a ...}....else {..
0650: 09 09 09 2a 78 3d 31 2c 20 2a 79 3d 30 3b 0d 0a ...*x=1, *y=0;..
0660: 09 09 09 72 65 74 75 72 6e 20 61 3b 0d 0a 09 09 ...return a;....
0670: 7d 0d 0a 09 7d 0d 0a 0d 0a 2f 2f 20 42 45 47 49 }...}....// BEGI
0680: 4e 20 43 55 54 20 48 45 52 45 0d 0a 09 70 75 62 N CUT HERE...pub
0690: 6c 69 63 3a 0d 0a 09 76 6f 69 64 20 72 75 6e 5f lic:...void run_
06a0: 74 65 73 74 28 69 6e 74 20 43 61 73 65 29 20 7b test(int Case) {
06b0: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
06c0: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 30 29 ) || (Case == 0)
06d0: 29 20 74 65 73 74 5f 63 61 73 65 5f 30 28 29 3b ) test_case_0();
06e0: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
06f0: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 31 29 ) || (Case == 1)
0700: 29 20 74 65 73 74 5f 63 61 73 65 5f 31 28 29 3b ) test_case_1();
0710: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
0720: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 32 29 ) || (Case == 2)
0730: 29 20 74 65 73 74 5f 63 61 73 65 5f 32 28 29 3b ) test_case_2();
0740: 20 69 66 20 28 28 43 61 73 65 20 3d 3d 20 2d 31 if ((Case == -1
0750: 29 20 7c 7c 20 28 43 61 73 65 20 3d 3d 20 33 29 ) || (Case == 3)
0760: 29 20 74 65 73 74 5f 63 61 73 65 5f 33 28 29 3b ) test_case_3();
0770: 20 7d 0d 0a 09 70 72 69 76 61 74 65 3a 0d 0a 09 }...private:...
0780: 74 65 6d 70 6c 61 74 65 20 3c 74 79 70 65 6e 61 template <typena
0790: 6d 65 20 54 3e 20 73 74 72 69 6e 67 20 70 72 69 me T> string pri
07a0: 6e 74 5f 61 72 72 61 79 28 63 6f 6e 73 74 20 76 nt_array(const v
07b0: 65 63 74 6f 72 3c 54 3e 20 26 56 29 20 7b 20 6f ector<T> &V) { o
07c0: 73 74 72 69 6e 67 73 74 72 65 61 6d 20 6f 73 3b stringstream os;
07d0: 20 6f 73 20 3c 3c 20 22 7b 20 22 3b 20 66 6f 72 os << "{ "; for
07e0: 20 28 74 79 70 65 6e 61 6d 65 20 76 65 63 74 6f (typename vecto
07f0: 72 3c 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 r<T>::const_iter
0800: 61 74 6f 72 20 69 74 65 72 20 3d 20 56 2e 62 65 ator iter = V.be
0810: 67 69 6e 28 29 3b 20 69 74 65 72 20 21 3d 20 56 gin(); iter != V
0820: 2e 65 6e 64 28 29 3b 20 2b 2b 69 74 65 72 29 20 .end(); ++iter)
0830: 6f 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 os << '\"' << *i
0840: 74 65 72 20 3c 3c 20 22 5c 22 2c 22 3b 20 6f 73 ter << "\","; os
0850: 20 3c 3c 20 22 20 7d 22 3b 20 72 65 74 75 72 6e << " }"; return
0860: 20 6f 73 2e 73 74 72 28 29 3b 20 7d 0d 0a 09 76 os.str(); }...v
0870: 6f 69 64 20 76 65 72 69 66 79 5f 63 61 73 65 28 oid verify_case(
0880: 69 6e 74 20 43 61 73 65 2c 20 63 6f 6e 73 74 20 int Case, const
0890: 69 6e 74 20 26 45 78 70 65 63 74 65 64 2c 20 63 int &Expected, c
08a0: 6f 6e 73 74 20 69 6e 74 20 26 52 65 63 65 69 76 onst int &Receiv
08b0: 65 64 29 20 7b 20 63 65 72 72 20 3c 3c 20 22 54 ed) { cerr << "T
08c0: 65 73 74 20 43 61 73 65 20 23 22 20 3c 3c 20 43 est Case #" << C
08d0: 61 73 65 20 3c 3c 20 22 2e 2e 2e 22 3b 20 69 66 ase << "..."; if
08e0: 20 28 45 78 70 65 63 74 65 64 20 3d 3d 20 52 65 (Expected == Re
08f0: 63 65 69 76 65 64 29 20 63 65 72 72 20 3c 3c 20 ceived) cerr <<
0900: 22 50 41 53 53 45 44 22 20 3c 3c 20 65 6e 64 6c "PASSED" << endl
0910: 3b 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c 3c ; else { cerr <<
0920: 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 65 6e 64 "FAILED" << end
0930: 6c 3b 20 63 65 72 72 20 3c 3c 20 22 5c 74 45 78 l; cerr << "\tEx
0940: 70 65 63 74 65 64 3a 20 5c 22 22 20 3c 3c 20 45 pected: \"" << E
0950: 78 70 65 63 74 65 64 20 3c 3c 20 27 5c 22 27 20 xpected << '\"'
0960: 3c 3c 20 65 6e 64 6c 3b 20 63 65 72 72 20 3c 3c << endl; cerr <<
0970: 20 22 5c 74 52 65 63 65 69 76 65 64 3a 20 5c 22 "\tReceived: \"
0980: 22 20 3c 3c 20 52 65 63 65 69 76 65 64 20 3c 3c " << Received <<
0990: 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d '\"' << endl; }
09a0: 20 7d 0d 0a 09 76 6f 69 64 20 74 65 73 74 5f 63 }...void test_c
09b0: 61 73 65 5f 30 28 29 20 7b 20 69 6e 74 20 41 72 ase_0() { int Ar
09c0: 67 30 20 3d 20 35 3b 20 69 6e 74 20 41 72 67 31 g0 = 5; int Arg1
09d0: 20 3d 20 31 3b 20 69 6e 74 20 41 72 67 32 20 3d = 1; int Arg2 =
09e0: 20 31 35 3b 20 76 65 72 69 66 79 5f 63 61 73 65 15; verify_case
09f0: 28 30 2c 20 41 72 67 32 2c 20 76 61 6c 75 65 28 (0, Arg2, value(
0a00: 41 72 67 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d Arg0, Arg1)); }.
0a10: 0a 09 76 6f 69 64 20 74 65 73 74 5f 63 61 73 65 ..void test_case
0a20: 5f 31 28 29 20 7b 20 69 6e 74 20 41 72 67 30 20 _1() { int Arg0
0a30: 3d 20 34 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 = 4; int Arg1 =
0a40: 32 3b 20 69 6e 74 20 41 72 67 32 20 3d 20 33 30 2; int Arg2 = 30
0a50: 3b 20 76 65 72 69 66 79 5f 63 61 73 65 28 31 2c ; verify_case(1,
0a60: 20 41 72 67 32 2c 20 76 61 6c 75 65 28 41 72 67 Arg2, value(Arg
0a70: 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76 0, Arg1)); }...v
0a80: 6f 69 64 20 74 65 73 74 5f 63 61 73 65 5f 32 28 oid test_case_2(
0a90: 29 20 7b 20 69 6e 74 20 41 72 67 30 20 3d 20 31 ) { int Arg0 = 1
0aa0: 33 3b 20 69 6e 74 20 41 72 67 31 20 3d 20 35 3b 3; int Arg1 = 5;
0ab0: 20 69 6e 74 20 41 72 67 32 20 3d 20 31 30 30 32 int Arg2 = 1002
0ac0: 30 30 31 3b 20 76 65 72 69 66 79 5f 63 61 73 65 001; verify_case
0ad0: 28 32 2c 20 41 72 67 32 2c 20 76 61 6c 75 65 28 (2, Arg2, value(
0ae0: 41 72 67 30 2c 20 41 72 67 31 29 29 3b 20 7d 0d Arg0, Arg1)); }.
0af0: 0a 09 76 6f 69 64 20 74 65 73 74 5f 63 61 73 65 ..void test_case
0b00: 5f 33 28 29 20 7b 20 69 6e 74 20 41 72 67 30 20 _3() { int Arg0
0b10: 3d 20 31 32 33 34 35 36 37 38 39 3b 20 69 6e 74 = 123456789; int
0b20: 20 41 72 67 31 20 3d 20 31 3b 20 69 6e 74 20 41 Arg1 = 1; int A
0b30: 72 67 32 20 3d 20 33 38 33 34 37 38 31 33 32 3b rg2 = 383478132;
0b40: 20 76 65 72 69 66 79 5f 63 61 73 65 28 33 2c 20 verify_case(3,
0b50: 41 72 67 32 2c 20 76 61 6c 75 65 28 41 72 67 30 Arg2, value(Arg0
0b60: 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a 0d 0a 2f , Arg1)); }..../
0b70: 2f 20 45 4e 44 20 43 55 54 20 48 45 52 45 0d 0a / END CUT HERE..
0b80: 7d 3b 0d 0a 2f 2f 20 42 45 47 49 4e 20 43 55 54 };..// BEGIN CUT
0b90: 20 48 45 52 45 20 0d 0a 69 6e 74 20 6d 61 69 6e HERE ..int main
0ba0: 28 29 20 7b 0d 0a 09 53 75 6d 4f 66 50 6f 77 65 () {...SumOfPowe
0bb0: 72 73 28 29 2e 72 75 6e 5f 74 65 73 74 28 2d 31 rs().run_test(-1
0bc0: 29 3b 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 20 43 55 );..}..// END CU
0bd0: 54 20 48 45 52 45 0d 0a T HERE..