Artifact a3e0995e3ee0720523b430c2a744300ab64623ed:
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 75 73 69 6e 67 20 6e 61 6d sert>..using nam
0140: 65 73 70 61 63 65 20 73 74 64 3b 0d 0a 74 79 70 espace std;..typ
0150: 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67 20 4c edef long long L
0160: 4c 3b 0d 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 L;..typedef long
0170: 20 64 6f 75 62 6c 65 20 4c 44 3b 0d 0a 74 79 70 double LD;..typ
0180: 65 64 65 66 20 63 6f 6d 70 6c 65 78 3c 4c 44 3e edef complex<LD>
0190: 20 43 4d 50 3b 0d 0a 0d 0a 63 6c 61 73 73 20 4b CMP;....class K
01a0: 6c 65 6f 66 61 73 54 61 69 6c 20 7b 20 70 75 62 leofasTail { pub
01b0: 6c 69 63 3a 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 lic:...long long
01c0: 20 63 6f 75 6e 74 47 6f 6f 64 53 65 71 75 65 6e countGoodSequen
01d0: 63 65 73 28 6c 6f 6e 67 20 6c 6f 6e 67 20 4b 2c ces(long long K,
01e0: 20 6c 6f 6e 67 20 6c 6f 6e 67 20 41 2c 20 6c 6f long long A, lo
01f0: 6e 67 20 6c 6f 6e 67 20 42 29 0d 0a 09 7b 0d 0a ng long B)...{..
0200: 09 09 72 65 74 75 72 6e 20 73 6f 6c 76 65 28 4b ..return solve(K
0210: 2c 42 29 20 2d 20 73 6f 6c 76 65 28 4b 2c 41 2d ,B) - solve(K,A-
0220: 31 29 3b 0d 0a 09 7d 0d 0a 09 4c 4c 20 73 6f 6c 1);...}...LL sol
0230: 76 65 28 4c 4c 20 4b 2c 20 4c 4c 20 55 29 0d 0a ve(LL K, LL U)..
0240: 09 7b 0d 0a 09 09 69 66 28 55 3d 3d 2d 31 29 0d .{....if(U==-1).
0250: 0a 09 09 09 72 65 74 75 72 6e 20 30 3b 0d 0a 09 ....return 0;...
0260: 09 69 66 28 4b 3d 3d 30 29 0d 0a 09 09 09 72 65 .if(K==0).....re
0270: 74 75 72 6e 20 55 2b 31 3b 0d 0a 0d 0a 09 09 4c turn U+1;......L
0280: 4c 20 76 20 3d 20 30 3b 0d 0a 09 09 66 6f 72 28 L v = 0;....for(
0290: 69 6e 74 20 69 3d 30 3b 20 28 4b 3c 3c 69 29 3c int i=0; (K<<i)<
02a0: 3d 55 3b 20 2b 2b 69 29 20 7b 0d 0a 09 09 09 69 =U; ++i) {.....i
02b0: 66 28 20 28 4b 2b 31 3c 3c 69 29 2d 31 20 3c 3d f( (K+1<<i)-1 <=
02c0: 20 55 20 29 0d 0a 09 09 09 09 76 20 2b 3d 20 28 U )......v += (
02d0: 31 4c 4c 3c 3c 69 29 3b 0d 0a 09 09 09 65 6c 73 1LL<<i);.....els
02e0: 65 0d 0a 09 09 09 09 76 20 2b 3d 20 55 20 2d 20 e......v += U -
02f0: 28 4b 3c 3c 69 29 20 2b 20 31 3b 0d 0a 09 09 7d (K<<i) + 1;....}
0300: 0d 0a 09 09 69 66 28 28 4b 26 31 29 3d 3d 30 29 ....if((K&1)==0)
0310: 20 7b 0d 0a 09 09 09 2b 2b 4b 3b 0d 0a 09 09 09 {.....++K;.....
0320: 66 6f 72 28 69 6e 74 20 69 3d 30 3b 20 28 4b 3c for(int i=0; (K<
0330: 3c 69 29 3c 3d 55 3b 20 2b 2b 69 29 20 7b 0d 0a <i)<=U; ++i) {..
0340: 09 09 09 09 69 66 28 20 28 4b 2b 31 3c 3c 69 29 ....if( (K+1<<i)
0350: 2d 31 20 3c 3d 20 55 20 29 0d 0a 09 09 09 09 09 -1 <= U ).......
0360: 76 20 2b 3d 20 28 31 4c 4c 3c 3c 69 29 3b 0d 0a v += (1LL<<i);..
0370: 09 09 09 09 65 6c 73 65 0d 0a 09 09 09 09 09 76 ....else.......v
0380: 20 2b 3d 20 55 20 2d 20 28 4b 3c 3c 69 29 20 2b += U - (K<<i) +
0390: 20 31 3b 0d 0a 09 09 09 7d 0d 0a 09 09 7d 0d 0a 1;.....}....}..
03a0: 09 09 72 65 74 75 72 6e 20 76 3b 0d 0a 09 7d 0d ..return v;...}.
03b0: 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 42 45 47 49 4e 20 .};....// BEGIN
03c0: 43 55 54 20 48 45 52 45 0d 0a 23 69 6e 63 6c 75 CUT HERE..#inclu
03d0: 64 65 20 3c 63 74 69 6d 65 3e 0d 0a 64 6f 75 62 de <ctime>..doub
03e0: 6c 65 20 73 74 61 72 74 5f 74 69 6d 65 3b 20 73 le start_time; s
03f0: 74 72 69 6e 67 20 74 69 6d 65 72 28 29 0d 0a 20 tring timer()..
0400: 7b 20 6f 73 74 72 69 6e 67 73 74 72 65 61 6d 20 { ostringstream
0410: 6f 73 3b 20 6f 73 20 3c 3c 20 22 20 28 22 20 3c os; os << " (" <
0420: 3c 20 69 6e 74 28 28 63 6c 6f 63 6b 28 29 2d 73 < int((clock()-s
0430: 74 61 72 74 5f 74 69 6d 65 29 2f 43 4c 4f 43 4b tart_time)/CLOCK
0440: 53 5f 50 45 52 5f 53 45 43 2a 31 30 30 30 29 20 S_PER_SEC*1000)
0450: 3c 3c 20 22 20 6d 73 65 63 29 22 3b 20 72 65 74 << " msec)"; ret
0460: 75 72 6e 20 6f 73 2e 73 74 72 28 29 3b 20 7d 0d urn os.str(); }.
0470: 0a 74 65 6d 70 6c 61 74 65 3c 74 79 70 65 6e 61 .template<typena
0480: 6d 65 20 54 3e 20 6f 73 74 72 65 61 6d 26 20 6f me T> ostream& o
0490: 70 65 72 61 74 6f 72 3c 3c 28 6f 73 74 72 65 61 perator<<(ostrea
04a0: 6d 26 20 6f 73 2c 20 63 6f 6e 73 74 20 76 65 63 m& os, const vec
04b0: 74 6f 72 3c 54 3e 26 20 76 29 0d 0a 20 7b 20 6f tor<T>& v).. { o
04c0: 73 20 3c 3c 20 22 7b 20 22 3b 0d 0a 20 20 20 66 s << "{ ";.. f
04d0: 6f 72 28 74 79 70 65 6e 61 6d 65 20 76 65 63 74 or(typename vect
04e0: 6f 72 3c 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 or<T>::const_ite
04f0: 72 61 74 6f 72 20 69 74 3d 76 2e 62 65 67 69 6e rator it=v.begin
0500: 28 29 3b 20 69 74 21 3d 76 2e 65 6e 64 28 29 3b (); it!=v.end();
0510: 20 2b 2b 69 74 29 0d 0a 20 20 20 6f 73 20 3c 3c ++it).. os <<
0520: 20 27 5c 22 27 20 3c 3c 20 2a 69 74 20 3c 3c 20 '\"' << *it <<
0530: 27 5c 22 27 20 3c 3c 20 28 69 74 2b 31 3d 3d 76 '\"' << (it+1==v
0540: 2e 65 6e 64 28 29 20 3f 20 22 22 20 3a 20 22 2c .end() ? "" : ",
0550: 20 22 29 3b 20 6f 73 20 3c 3c 20 22 20 7d 22 3b "); os << " }";
0560: 20 72 65 74 75 72 6e 20 6f 73 3b 20 7d 0d 0a 76 return os; }..v
0570: 6f 69 64 20 76 65 72 69 66 79 5f 63 61 73 65 28 oid verify_case(
0580: 63 6f 6e 73 74 20 6c 6f 6e 67 20 6c 6f 6e 67 26 const long long&
0590: 20 45 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 Expected, const
05a0: 20 6c 6f 6e 67 20 6c 6f 6e 67 26 20 52 65 63 65 long long& Rece
05b0: 69 76 65 64 29 20 7b 0d 0a 20 62 6f 6f 6c 20 6f ived) {.. bool o
05c0: 6b 20 3d 20 28 45 78 70 65 63 74 65 64 20 3d 3d k = (Expected ==
05d0: 20 52 65 63 65 69 76 65 64 29 3b 0d 0a 20 69 66 Received);.. if
05e0: 28 6f 6b 29 20 63 65 72 72 20 3c 3c 20 22 50 41 (ok) cerr << "PA
05f0: 53 53 45 44 22 20 3c 3c 20 74 69 6d 65 72 28 29 SSED" << timer()
0600: 20 3c 3c 20 65 6e 64 6c 3b 20 20 65 6c 73 65 20 << endl; else
0610: 7b 20 63 65 72 72 20 3c 3c 20 22 46 41 49 4c 45 { cerr << "FAILE
0620: 44 22 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c D" << timer() <<
0630: 20 65 6e 64 6c 3b 0d 0a 20 63 65 72 72 20 3c 3c endl;.. cerr <<
0640: 20 22 5c 74 6f 3a 20 5c 22 22 20 3c 3c 20 45 78 "\to: \"" << Ex
0650: 70 65 63 74 65 64 20 3c 3c 20 27 5c 22 27 20 3c pected << '\"' <
0660: 3c 20 65 6e 64 6c 20 3c 3c 20 22 5c 74 78 3a 20 < endl << "\tx:
0670: 5c 22 22 20 3c 3c 20 52 65 63 65 69 76 65 64 20 \"" << Received
0680: 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 3b << '\"' << endl;
0690: 20 7d 20 7d 0d 0a 23 64 65 66 69 6e 65 20 43 41 } }..#define CA
06a0: 53 45 28 4e 29 20 7b 63 65 72 72 20 3c 3c 20 22 SE(N) {cerr << "
06b0: 54 65 73 74 20 43 61 73 65 20 23 22 20 3c 3c 20 Test Case #" <<
06c0: 4e 20 3c 3c 20 22 2e 2e 2e 22 20 3c 3c 20 66 6c N << "..." << fl
06d0: 75 73 68 3b 20 73 74 61 72 74 5f 74 69 6d 65 3d ush; start_time=
06e0: 63 6c 6f 63 6b 28 29 3b 0d 0a 23 64 65 66 69 6e clock();..#defin
06f0: 65 20 45 4e 44 09 20 76 65 72 69 66 79 5f 63 61 e END. verify_ca
0700: 73 65 28 5f 2c 20 4b 6c 65 6f 66 61 73 54 61 69 se(_, KleofasTai
0710: 6c 28 29 2e 63 6f 75 6e 74 47 6f 6f 64 53 65 71 l().countGoodSeq
0720: 75 65 6e 63 65 73 28 4b 2c 20 41 2c 20 42 29 29 uences(K, A, B))
0730: 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 7b 0d ;}..int main(){.
0740: 0a 43 41 53 45 28 30 29 0d 0a 09 6c 6f 6e 67 20 .CASE(0)...long
0750: 6c 6f 6e 67 20 4b 20 3d 20 33 4c 4c 3b 20 0d 0a long K = 3LL; ..
0760: 09 6c 6f 6e 67 20 6c 6f 6e 67 20 41 20 3d 20 34 .long long A = 4
0770: 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 LL; ...long long
0780: 20 42 20 3d 20 38 4c 4c 3b 20 0d 0a 09 6c 6f 6e B = 8LL; ...lon
0790: 67 20 6c 6f 6e 67 20 5f 20 3d 20 32 4c 4c 3b 20 g long _ = 2LL;
07a0: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 31 29 0d 0a ..END..CASE(1)..
07b0: 09 6c 6f 6e 67 20 6c 6f 6e 67 20 4b 20 3d 20 31 .long long K = 1
07c0: 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 LL; ...long long
07d0: 20 41 20 3d 20 32 33 34 35 37 4c 4c 3b 20 0d 0a A = 23457LL; ..
07e0: 09 6c 6f 6e 67 20 6c 6f 6e 67 20 42 20 3d 20 31 .long long B = 1
07f0: 32 33 34 35 36 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 23456LL; ...long
0800: 20 6c 6f 6e 67 20 5f 20 3d 20 31 30 30 30 30 30 long _ = 100000
0810: 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 LL; ..END..CASE(
0820: 32 29 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 4b 2)...long long K
0830: 20 3d 20 31 32 33 34 35 36 37 38 39 30 31 32 33 = 1234567890123
0840: 34 35 36 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 456LL; ...long l
0850: 6f 6e 67 20 41 20 3d 20 31 30 4c 4c 3b 20 0d 0a ong A = 10LL; ..
0860: 09 6c 6f 6e 67 20 6c 6f 6e 67 20 42 20 3d 20 31 .long long B = 1
0870: 30 30 30 30 30 30 4c 4c 3b 20 0d 0a 09 6c 6f 6e 000000LL; ...lon
0880: 67 20 6c 6f 6e 67 20 5f 20 3d 20 30 4c 4c 3b 20 g long _ = 0LL;
0890: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 33 29 0d 0a ..END..CASE(3)..
08a0: 09 6c 6f 6e 67 20 6c 6f 6e 67 20 4b 20 3d 20 30 .long long K = 0
08b0: 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 LL; ...long long
08c0: 20 41 20 3d 20 30 4c 4c 3b 20 0d 0a 09 6c 6f 6e A = 0LL; ...lon
08d0: 67 20 6c 6f 6e 67 20 42 20 3d 20 32 4c 4c 3b 20 g long B = 2LL;
08e0: 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 5f 20 3d ...long long _ =
08f0: 20 33 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 3LL; ..END..CAS
0900: 45 28 34 29 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 E(4)...long long
0910: 20 4b 20 3d 20 32 4c 4c 3b 20 0d 0a 09 6c 6f 6e K = 2LL; ...lon
0920: 67 20 6c 6f 6e 67 20 41 20 3d 20 33 4c 4c 3b 20 g long A = 3LL;
0930: 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 42 20 3d ...long long B =
0940: 20 33 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 3LL; ...long lo
0950: 6e 67 20 5f 20 3d 20 31 4c 4c 3b 20 0d 0a 45 4e ng _ = 1LL; ..EN
0960: 44 0d 0a 43 41 53 45 28 35 29 0d 0a 09 6c 6f 6e D..CASE(5)...lon
0970: 67 20 6c 6f 6e 67 20 4b 20 3d 20 31 33 4c 4c 3b g long K = 13LL;
0980: 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 41 20 ...long long A
0990: 3d 20 31 32 33 34 35 4c 4c 3b 20 0d 0a 09 6c 6f = 12345LL; ...lo
09a0: 6e 67 20 6c 6f 6e 67 20 42 20 3d 20 36 37 38 39 ng long B = 6789
09b0: 30 31 32 33 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 0123LL; ...long
09c0: 6c 6f 6e 67 20 5f 20 3d 20 38 33 38 37 35 38 34 long _ = 8387584
09d0: 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 LL; ..END..CASE(
09e0: 36 29 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 4b 6)...long long K
09f0: 20 3d 20 32 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 = 2LL; ...long
0a00: 6c 6f 6e 67 20 41 20 3d 20 30 4c 4c 3b 20 0d 0a long A = 0LL; ..
0a10: 09 6c 6f 6e 67 20 6c 6f 6e 67 20 42 20 3d 20 36 .long long B = 6
0a20: 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 LL; ...long long
0a30: 20 5f 20 3d 20 35 4c 4c 3b 20 0d 0a 45 4e 44 0d _ = 5LL; ..END.
0a40: 0a 43 41 53 45 28 37 29 0d 0a 09 6c 6f 6e 67 20 .CASE(7)...long
0a50: 6c 6f 6e 67 20 4b 20 3d 20 32 4c 4c 3b 20 0d 0a long K = 2LL; ..
0a60: 09 6c 6f 6e 67 20 6c 6f 6e 67 20 41 20 3d 20 30 .long long A = 0
0a70: 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 LL; ...long long
0a80: 20 42 20 3d 20 39 39 39 39 39 39 39 39 39 39 39 B = 99999999999
0a90: 39 39 4c 4c 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 99LL; ...long lo
0aa0: 6e 67 20 5f 20 3d 20 2d 31 4c 4c 3b 20 0d 0a 45 ng _ = -1LL; ..E
0ab0: 4e 44 0d 0a 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 20 ND....}..// END
0ac0: 43 55 54 20 48 45 52 45 0d 0a CUT HERE..