Artifact dc438b5d9c04f458abdc399801c576db7c61617c:
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 23 69 6e 63 6c 75 64 65 20 sert>..#include
0140: 3c 74 75 70 6c 65 3e 0d 0a 75 73 69 6e 67 20 6e <tuple>..using n
0150: 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d 0a 74 amespace std;..t
0160: 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67 ypedef long long
0170: 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 63 6f LL;..typedef co
0180: 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 43 4d mplex<double> CM
0190: 50 3b 0d 0a 0d 0a 63 6c 61 73 73 20 46 69 6e 64 P;....class Find
01a0: 69 6e 67 4b 69 64 73 20 7b 20 70 75 62 6c 69 63 ingKids { public
01b0: 3a 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 67 65 :...long long ge
01c0: 74 53 75 6d 28 69 6e 74 20 6e 2c 20 69 6e 74 20 tSum(int n, int
01d0: 71 2c 20 69 6e 74 20 41 5f 2c 20 63 6f 6e 73 74 q, int A_, const
01e0: 20 69 6e 74 20 42 2c 20 63 6f 6e 73 74 20 69 6e int B, const in
01f0: 74 20 43 29 0d 0a 09 7b 0d 0a 09 09 73 65 74 3c t C)...{....set<
0200: 69 6e 74 3e 20 70 6f 73 5f 73 65 74 3b 0d 0a 09 int> pos_set;...
0210: 09 76 65 63 74 6f 72 3c 69 6e 74 3e 20 70 6f 73 .vector<int> pos
0220: 28 6e 29 2c 20 64 69 72 28 6e 29 3b 0d 0a 0d 0a (n), dir(n);....
0230: 09 09 63 6f 6e 73 74 20 69 6e 74 20 4d 20 3d 20 ..const int M =
0240: 31 30 30 30 30 30 30 30 30 37 3b 0d 0a 09 09 4c 1000000007;....L
0250: 4c 20 41 20 3d 20 41 5f 3b 0d 0a 09 09 66 6f 72 L A = A_;....for
0260: 28 69 6e 74 20 69 3d 30 3b 20 69 3c 6e 3b 20 2b (int i=0; i<n; +
0270: 2b 69 29 20 7b 0d 0a 09 09 20 20 20 20 41 20 3d +i) {.... A =
0280: 20 28 41 20 2a 20 42 20 2b 20 43 29 20 25 20 4d (A * B + C) % M
0290: 3b 0d 0a 09 09 09 69 6e 74 20 70 20 3d 20 69 6e ;.....int p = in
02a0: 74 28 41 20 25 20 28 4d 20 2d 20 6e 20 2b 20 69 t(A % (M - n + i
02b0: 20 2b 20 31 29 29 3b 0d 0a 09 09 20 20 20 20 69 + 1));.... i
02c0: 66 28 70 6f 73 5f 73 65 74 2e 63 6f 75 6e 74 28 f(pos_set.count(
02d0: 70 29 29 0d 0a 09 09 09 09 70 20 3d 20 4d 20 2d p))......p = M -
02e0: 20 6e 20 2b 20 69 3b 0d 0a 09 09 20 20 20 20 70 n + i;.... p
02f0: 6f 73 5b 69 5d 20 3d 20 70 3b 0d 0a 09 09 09 70 os[i] = p;.....p
0300: 6f 73 5f 73 65 74 2e 69 6e 73 65 72 74 28 70 29 os_set.insert(p)
0310: 3b 0d 0a 09 09 20 20 20 20 69 66 28 70 25 32 20 ;.... if(p%2
0320: 3d 3d 20 30 29 0d 0a 09 09 20 20 20 20 20 20 20 == 0)....
0330: 20 64 69 72 5b 69 5d 20 3d 20 2b 31 3b 0d 0a 09 dir[i] = +1;...
0340: 09 20 20 20 20 65 6c 73 65 0d 0a 09 09 09 20 20 . else.....
0350: 20 20 64 69 72 5b 69 5d 20 3d 20 2d 31 3b 0d 0a dir[i] = -1;..
0360: 09 09 7d 0d 0a 0d 0a 09 09 76 65 63 74 6f 72 3c ..}......vector<
0370: 69 6e 74 3e 20 6b 69 64 28 71 29 2c 20 74 69 6d int> kid(q), tim
0380: 65 28 71 29 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 e(q);....for(int
0390: 20 69 3d 30 3b 20 69 3c 71 3b 20 2b 2b 69 29 20 i=0; i<q; ++i)
03a0: 7b 0d 0a 09 09 20 20 20 20 41 20 3d 20 28 41 20 {.... A = (A
03b0: 2a 20 42 20 2b 20 43 29 20 25 20 4d 3b 0d 0a 09 * B + C) % M;...
03c0: 09 20 20 20 20 6b 69 64 5b 69 5d 20 3d 20 69 6e . kid[i] = in
03d0: 74 28 41 25 6e 29 3b 0d 0a 09 09 20 20 20 20 41 t(A%n);.... A
03e0: 20 3d 20 28 41 20 2a 20 42 20 2b 20 43 29 20 25 = (A * B + C) %
03f0: 20 4d 3b 0d 0a 09 09 09 74 69 6d 65 5b 69 5d 20 M;.....time[i]
0400: 3d 20 69 6e 74 28 41 29 3b 0d 0a 09 09 7d 0d 0a = int(A);....}..
0410: 0d 0a 09 09 72 65 74 75 72 6e 20 73 6f 6c 76 65 ....return solve
0420: 28 6e 2c 20 70 6f 73 2c 20 64 69 72 2c 20 71 2c (n, pos, dir, q,
0430: 20 6b 69 64 2c 20 74 69 6d 65 29 3b 0d 0a 09 7d kid, time);...}
0440: 0d 0a 0d 0a 09 4c 4c 20 73 6f 6c 76 65 28 0d 0a .....LL solve(..
0450: 09 09 63 6f 6e 73 74 20 69 6e 74 20 4e 2c 20 63 ..const int N, c
0460: 6f 6e 73 74 20 76 65 63 74 6f 72 3c 69 6e 74 3e onst vector<int>
0470: 26 20 50 2c 20 63 6f 6e 73 74 20 76 65 63 74 6f & P, const vecto
0480: 72 3c 69 6e 74 3e 26 20 44 2c 0d 0a 09 09 63 6f r<int>& D,....co
0490: 6e 73 74 20 69 6e 74 20 51 2c 20 63 6f 6e 73 74 nst int Q, const
04a0: 20 76 65 63 74 6f 72 3c 69 6e 74 3e 26 20 4b 2c vector<int>& K,
04b0: 20 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c 69 6e const vector<in
04c0: 74 3e 26 20 54 29 20 7b 0d 0a 0d 0a 09 09 76 65 t>& T) {......ve
04d0: 63 74 6f 72 3c 69 6e 74 3e 20 6b 69 64 5f 69 64 ctor<int> kid_id
04e0: 5f 74 6f 5f 70 6f 73 5f 6f 72 64 65 72 28 4e 29 _to_pos_order(N)
04f0: 3b 20 7b 0d 0a 09 09 09 76 65 63 74 6f 72 3c 69 ; {.....vector<i
0500: 6e 74 3e 20 6b 69 64 73 5f 73 6f 72 74 65 64 5f nt> kids_sorted_
0510: 62 79 5f 70 6f 73 28 4e 29 3b 0d 0a 09 09 09 66 by_pos(N);.....f
0520: 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 3c 4e 3b or(int i=0; i<N;
0530: 20 2b 2b 69 29 20 6b 69 64 73 5f 73 6f 72 74 65 ++i) kids_sorte
0540: 64 5f 62 79 5f 70 6f 73 5b 69 5d 20 3d 20 69 3b d_by_pos[i] = i;
0550: 0d 0a 09 09 09 73 6f 72 74 28 6b 69 64 73 5f 73 .....sort(kids_s
0560: 6f 72 74 65 64 5f 62 79 5f 70 6f 73 2e 62 65 67 orted_by_pos.beg
0570: 69 6e 28 29 2c 20 6b 69 64 73 5f 73 6f 72 74 65 in(), kids_sorte
0580: 64 5f 62 79 5f 70 6f 73 2e 65 6e 64 28 29 2c 20 d_by_pos.end(),
0590: 5b 26 5d 28 69 6e 74 20 61 2c 20 69 6e 74 20 62 [&](int a, int b
05a0: 29 7b 0d 0a 09 09 09 09 72 65 74 75 72 6e 20 50 ){......return P
05b0: 5b 61 5d 20 3c 20 50 5b 62 5d 3b 0d 0a 09 09 09 [a] < P[b];.....
05c0: 7d 29 3b 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 });.....for(int
05d0: 69 3d 30 3b 20 69 3c 4e 3b 20 2b 2b 69 29 0d 0a i=0; i<N; ++i)..
05e0: 09 09 09 09 6b 69 64 5f 69 64 5f 74 6f 5f 70 6f ....kid_id_to_po
05f0: 73 5f 6f 72 64 65 72 5b 6b 69 64 73 5f 73 6f 72 s_order[kids_sor
0600: 74 65 64 5f 62 79 5f 70 6f 73 5b 69 5d 5d 20 3d ted_by_pos[i]] =
0610: 20 69 3b 0d 0a 09 09 7d 0d 0a 0d 0a 09 09 76 65 i;....}......ve
0620: 63 74 6f 72 3c 4c 4c 3e 20 6d 6f 76 65 5f 6b 69 ctor<LL> move_ki
0630: 64 73 2c 20 73 74 61 79 5f 6b 69 64 73 3b 0d 0a ds, stay_kids;..
0640: 09 09 66 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 ..for(int i=0; i
0650: 3c 4e 3b 20 2b 2b 69 29 0d 0a 09 09 09 28 44 5b <N; ++i).....(D[
0660: 69 5d 20 3d 3d 20 2b 31 20 3f 20 6d 6f 76 65 5f i] == +1 ? move_
0670: 6b 69 64 73 20 3a 20 73 74 61 79 5f 6b 69 64 73 kids : stay_kids
0680: 29 2e 65 6d 70 6c 61 63 65 5f 62 61 63 6b 28 50 ).emplace_back(P
0690: 5b 69 5d 29 3b 0d 0a 0d 0a 09 09 73 6f 72 74 28 [i]);......sort(
06a0: 6d 6f 76 65 5f 6b 69 64 73 2e 62 65 67 69 6e 28 move_kids.begin(
06b0: 29 2c 20 6d 6f 76 65 5f 6b 69 64 73 2e 65 6e 64 ), move_kids.end
06c0: 28 29 29 3b 0d 0a 09 09 73 6f 72 74 28 73 74 61 ());....sort(sta
06d0: 79 5f 6b 69 64 73 2e 62 65 67 69 6e 28 29 2c 20 y_kids.begin(),
06e0: 73 74 61 79 5f 6b 69 64 73 2e 65 6e 64 28 29 29 stay_kids.end())
06f0: 3b 0d 0a 0d 0a 09 09 4c 4c 20 74 6f 74 61 6c 20 ;......LL total
0700: 3d 20 30 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 = 0;....for(int
0710: 69 3d 30 3b 20 69 3c 51 3b 20 2b 2b 69 29 20 7b i=0; i<Q; ++i) {
0720: 0d 0a 09 09 09 4c 4c 20 70 20 3d 20 66 69 6e 64 .....LL p = find
0730: 5f 6b 74 68 5f 6c 65 66 74 28 6b 69 64 5f 69 64 _kth_left(kid_id
0740: 5f 74 6f 5f 70 6f 73 5f 6f 72 64 65 72 5b 4b 5b _to_pos_order[K[
0750: 69 5d 5d 2c 20 73 74 61 79 5f 6b 69 64 73 2c 20 i]], stay_kids,
0760: 6d 6f 76 65 5f 6b 69 64 73 2c 20 32 2a 54 5b 69 move_kids, 2*T[i
0770: 5d 29 3b 0d 0a 09 09 09 74 6f 74 61 6c 20 2b 3d ]);.....total +=
0780: 20 61 62 73 28 70 20 2d 20 54 5b 69 5d 29 3b 0d abs(p - T[i]);.
0790: 0a 09 09 7d 0d 0a 09 09 72 65 74 75 72 6e 20 74 ...}....return t
07a0: 6f 74 61 6c 3b 0d 0a 09 7d 0d 0a 0d 0a 09 2f 2f otal;...}.....//
07b0: 20 46 69 6e 64 20 74 68 65 20 24 6b 24 2d 74 68 Find the $k$-th
07c0: 20 6c 65 66 74 20 6b 69 64 20 69 6e 20 7b 73 74 left kid in {st
07d0: 61 79 5f 6b 69 64 73 7d 20 55 20 7b 6d 6f 76 65 ay_kids} U {move
07e0: 5f 6b 69 64 73 20 2b 3d 20 74 7d 2e 0d 0a 09 4c _kids += t}....L
07f0: 4c 20 66 69 6e 64 5f 6b 74 68 5f 6c 65 66 74 28 L find_kth_left(
0800: 63 6f 6e 73 74 20 69 6e 74 20 6b 2c 20 63 6f 6e const int k, con
0810: 73 74 20 76 65 63 74 6f 72 3c 4c 4c 3e 26 20 73 st vector<LL>& s
0820: 74 61 79 5f 6b 69 64 73 2c 20 63 6f 6e 73 74 20 tay_kids, const
0830: 76 65 63 74 6f 72 3c 4c 4c 3e 26 20 6d 6f 76 65 vector<LL>& move
0840: 5f 6b 69 64 73 2c 20 63 6f 6e 73 74 20 4c 4c 20 _kids, const LL
0850: 74 29 20 7b 0d 0a 09 09 69 66 28 73 74 61 79 5f t) {....if(stay_
0860: 6b 69 64 73 2e 65 6d 70 74 79 28 29 29 20 72 65 kids.empty()) re
0870: 74 75 72 6e 20 6d 6f 76 65 5f 6b 69 64 73 5b 6b turn move_kids[k
0880: 5d 2b 74 3b 0d 0a 09 09 69 66 28 6d 6f 76 65 5f ]+t;....if(move_
0890: 6b 69 64 73 2e 65 6d 70 74 79 28 29 29 20 72 65 kids.empty()) re
08a0: 74 75 72 6e 20 73 74 61 79 5f 6b 69 64 73 5b 6b turn stay_kids[k
08b0: 5d 3b 0d 0a 09 09 69 66 28 73 74 61 79 5f 6b 69 ];....if(stay_ki
08c0: 64 73 2e 62 61 63 6b 28 29 20 3c 20 6d 6f 76 65 ds.back() < move
08d0: 5f 6b 69 64 73 2e 66 72 6f 6e 74 28 29 2b 74 29 _kids.front()+t)
08e0: 20 7b 0d 0a 09 09 09 69 66 28 6b 20 3c 20 73 74 {.....if(k < st
08f0: 61 79 5f 6b 69 64 73 2e 73 69 7a 65 28 29 29 0d ay_kids.size()).
0900: 0a 09 09 09 09 72 65 74 75 72 6e 20 73 74 61 79 .....return stay
0910: 5f 6b 69 64 73 5b 6b 5d 3b 0d 0a 09 09 09 65 6c _kids[k];.....el
0920: 73 65 0d 0a 09 09 09 09 72 65 74 75 72 6e 20 6d se......return m
0930: 6f 76 65 5f 6b 69 64 73 5b 6b 2d 73 74 61 79 5f ove_kids[k-stay_
0940: 6b 69 64 73 2e 73 69 7a 65 28 29 5d 2b 74 3b 0d kids.size()]+t;.
0950: 0a 09 09 7d 0d 0a 09 09 69 66 28 6d 6f 76 65 5f ...}....if(move_
0960: 6b 69 64 73 2e 62 61 63 6b 28 29 2b 74 20 3c 20 kids.back()+t <
0970: 73 74 61 79 5f 6b 69 64 73 2e 66 72 6f 6e 74 28 stay_kids.front(
0980: 29 29 20 7b 0d 0a 09 09 09 69 66 28 6b 20 3c 20 )) {.....if(k <
0990: 6d 6f 76 65 5f 6b 69 64 73 2e 73 69 7a 65 28 29 move_kids.size()
09a0: 29 0d 0a 09 09 09 09 72 65 74 75 72 6e 20 6d 6f )......return mo
09b0: 76 65 5f 6b 69 64 73 5b 6b 5d 2b 74 3b 0d 0a 09 ve_kids[k]+t;...
09c0: 09 09 65 6c 73 65 0d 0a 09 09 09 09 72 65 74 75 ..else......retu
09d0: 72 6e 20 73 74 61 79 5f 6b 69 64 73 5b 6b 2d 6d rn stay_kids[k-m
09e0: 6f 76 65 5f 6b 69 64 73 2e 73 69 7a 65 28 29 5d ove_kids.size()]
09f0: 2b 74 3b 0d 0a 09 09 7d 0d 0a 0d 0a 09 09 69 6e +t;....}......in
0a00: 74 20 4c 3d 30 2c 20 52 3d 73 74 61 79 5f 6b 69 t L=0, R=stay_ki
0a10: 64 73 2e 73 69 7a 65 28 29 3b 0d 0a 09 09 77 68 ds.size();....wh
0a20: 69 6c 65 28 52 2d 4c 3e 31 29 20 7b 0d 0a 09 09 ile(R-L>1) {....
0a30: 09 69 6e 74 20 43 20 3d 20 28 4c 2b 52 29 2f 32 .int C = (L+R)/2
0a40: 3b 0d 0a 09 09 09 69 6e 74 20 6c 65 66 74 5f 69 ;.....int left_i
0a50: 6e 5f 73 74 61 79 20 3d 20 43 3b 0d 0a 09 09 09 n_stay = C;.....
0a60: 69 6e 74 20 6c 65 66 74 5f 69 6e 5f 6d 6f 76 65 int left_in_move
0a70: 20 3d 20 6c 6f 77 65 72 5f 62 6f 75 6e 64 28 6d = lower_bound(m
0a80: 6f 76 65 5f 6b 69 64 73 2e 62 65 67 69 6e 28 29 ove_kids.begin()
0a90: 2c 20 6d 6f 76 65 5f 6b 69 64 73 2e 65 6e 64 28 , move_kids.end(
0aa0: 29 2c 20 73 74 61 79 5f 6b 69 64 73 5b 43 5d 2d ), stay_kids[C]-
0ab0: 74 29 20 2d 20 6d 6f 76 65 5f 6b 69 64 73 2e 62 t) - move_kids.b
0ac0: 65 67 69 6e 28 29 3b 0d 0a 09 09 09 69 6e 74 20 egin();.....int
0ad0: 6c 65 66 74 20 3d 20 6c 65 66 74 5f 69 6e 5f 73 left = left_in_s
0ae0: 74 61 79 20 2b 20 6c 65 66 74 5f 69 6e 5f 6d 6f tay + left_in_mo
0af0: 76 65 3b 0d 0a 09 09 09 69 66 28 6c 65 66 74 20 ve;.....if(left
0b00: 3d 3d 20 6b 29 0d 0a 09 09 09 09 72 65 74 75 72 == k)......retur
0b10: 6e 20 73 74 61 79 5f 6b 69 64 73 5b 43 5d 3b 0d n stay_kids[C];.
0b20: 0a 09 09 09 28 6c 65 66 74 20 3e 3d 20 6b 20 3f ....(left >= k ?
0b30: 20 52 20 3a 20 4c 29 20 3d 20 43 3b 0d 0a 09 09 R : L) = C;....
0b40: 7d 0d 0a 0d 0a 09 09 4c 3d 30 2c 20 52 3d 6d 6f }......L=0, R=mo
0b50: 76 65 5f 6b 69 64 73 2e 73 69 7a 65 28 29 3b 0d ve_kids.size();.
0b60: 0a 09 09 77 68 69 6c 65 28 52 2d 4c 3e 31 29 20 ...while(R-L>1)
0b70: 7b 0d 0a 09 09 09 69 6e 74 20 43 20 3d 20 28 4c {.....int C = (L
0b80: 2b 52 29 2f 32 3b 0d 0a 09 09 09 69 6e 74 20 6c +R)/2;.....int l
0b90: 65 66 74 5f 69 6e 5f 6d 6f 76 65 20 3d 20 43 3b eft_in_move = C;
0ba0: 0d 0a 09 09 09 69 6e 74 20 6c 65 66 74 5f 69 6e .....int left_in
0bb0: 5f 73 74 61 79 20 3d 20 6c 6f 77 65 72 5f 62 6f _stay = lower_bo
0bc0: 75 6e 64 28 73 74 61 79 5f 6b 69 64 73 2e 62 65 und(stay_kids.be
0bd0: 67 69 6e 28 29 2c 20 73 74 61 79 5f 6b 69 64 73 gin(), stay_kids
0be0: 2e 65 6e 64 28 29 2c 20 6d 6f 76 65 5f 6b 69 64 .end(), move_kid
0bf0: 73 5b 43 5d 2b 74 29 20 2d 20 73 74 61 79 5f 6b s[C]+t) - stay_k
0c00: 69 64 73 2e 62 65 67 69 6e 28 29 3b 0d 0a 09 09 ids.begin();....
0c10: 09 69 6e 74 20 6c 65 66 74 20 3d 20 6c 65 66 74 .int left = left
0c20: 5f 69 6e 5f 73 74 61 79 20 2b 20 6c 65 66 74 5f _in_stay + left_
0c30: 69 6e 5f 6d 6f 76 65 3b 0d 0a 09 09 09 69 66 28 in_move;.....if(
0c40: 6c 65 66 74 20 3d 3d 20 6b 29 0d 0a 09 09 09 09 left == k)......
0c50: 72 65 74 75 72 6e 20 6d 6f 76 65 5f 6b 69 64 73 return move_kids
0c60: 5b 43 5d 2b 74 3b 0d 0a 09 09 09 28 6c 65 66 74 [C]+t;.....(left
0c70: 20 3e 3d 20 6b 20 3f 20 52 20 3a 20 4c 29 20 3d >= k ? R : L) =
0c80: 20 43 3b 0d 0a 09 09 7d 0d 0a 0d 0a 09 09 76 65 C;....}......ve
0c90: 63 74 6f 72 3c 4c 4c 3e 20 6e 6f 77 20 3d 20 73 ctor<LL> now = s
0ca0: 74 61 79 5f 6b 69 64 73 3b 0d 0a 09 09 66 6f 72 tay_kids;....for
0cb0: 28 4c 4c 20 70 3a 20 6d 6f 76 65 5f 6b 69 64 73 (LL p: move_kids
0cc0: 29 20 6e 6f 77 2e 70 75 73 68 5f 62 61 63 6b 28 ) now.push_back(
0cd0: 70 20 2b 20 74 29 3b 0d 0a 09 09 73 6f 72 74 28 p + t);....sort(
0ce0: 6e 6f 77 2e 62 65 67 69 6e 28 29 2c 20 6e 6f 77 now.begin(), now
0cf0: 2e 65 6e 64 28 29 29 3b 0d 0a 09 09 72 65 74 75 .end());....retu
0d00: 72 6e 20 6e 6f 77 5b 6b 5d 3b 0d 0a 09 7d 0d 0a rn now[k];...}..
0d10: 7d 3b 0d 0a 0d 0a 2f 2f 20 42 45 47 49 4e 20 43 };....// BEGIN C
0d20: 55 54 20 48 45 52 45 0d 0a 23 69 6e 63 6c 75 64 UT HERE..#includ
0d30: 65 20 3c 63 74 69 6d 65 3e 0d 0a 64 6f 75 62 6c e <ctime>..doubl
0d40: 65 20 73 74 61 72 74 5f 74 69 6d 65 3b 20 73 74 e start_time; st
0d50: 72 69 6e 67 20 74 69 6d 65 72 28 29 0d 0a 20 7b ring timer().. {
0d60: 20 6f 73 74 72 69 6e 67 73 74 72 65 61 6d 20 6f ostringstream o
0d70: 73 3b 20 6f 73 20 3c 3c 20 22 20 28 22 20 3c 3c s; os << " (" <<
0d80: 20 69 6e 74 28 28 63 6c 6f 63 6b 28 29 2d 73 74 int((clock()-st
0d90: 61 72 74 5f 74 69 6d 65 29 2f 43 4c 4f 43 4b 53 art_time)/CLOCKS
0da0: 5f 50 45 52 5f 53 45 43 2a 31 30 30 30 29 20 3c _PER_SEC*1000) <
0db0: 3c 20 22 20 6d 73 65 63 29 22 3b 20 72 65 74 75 < " msec)"; retu
0dc0: 72 6e 20 6f 73 2e 73 74 72 28 29 3b 20 7d 0d 0a rn os.str(); }..
0dd0: 74 65 6d 70 6c 61 74 65 3c 74 79 70 65 6e 61 6d template<typenam
0de0: 65 20 54 3e 20 6f 73 74 72 65 61 6d 26 20 6f 70 e T> ostream& op
0df0: 65 72 61 74 6f 72 3c 3c 28 6f 73 74 72 65 61 6d erator<<(ostream
0e00: 26 20 6f 73 2c 20 63 6f 6e 73 74 20 76 65 63 74 & os, const vect
0e10: 6f 72 3c 54 3e 26 20 76 29 0d 0a 20 7b 20 6f 73 or<T>& v).. { os
0e20: 20 3c 3c 20 22 7b 20 22 3b 0d 0a 20 20 20 66 6f << "{ ";.. fo
0e30: 72 28 74 79 70 65 6e 61 6d 65 20 76 65 63 74 6f r(typename vecto
0e40: 72 3c 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 r<T>::const_iter
0e50: 61 74 6f 72 20 69 74 3d 76 2e 62 65 67 69 6e 28 ator it=v.begin(
0e60: 29 3b 20 69 74 21 3d 76 2e 65 6e 64 28 29 3b 20 ); it!=v.end();
0e70: 2b 2b 69 74 29 0d 0a 20 20 20 6f 73 20 3c 3c 20 ++it).. os <<
0e80: 27 5c 22 27 20 3c 3c 20 2a 69 74 20 3c 3c 20 27 '\"' << *it << '
0e90: 5c 22 27 20 3c 3c 20 28 69 74 2b 31 3d 3d 76 2e \"' << (it+1==v.
0ea0: 65 6e 64 28 29 20 3f 20 22 22 20 3a 20 22 2c 20 end() ? "" : ",
0eb0: 22 29 3b 20 6f 73 20 3c 3c 20 22 20 7d 22 3b 20 "); os << " }";
0ec0: 72 65 74 75 72 6e 20 6f 73 3b 20 7d 0d 0a 76 6f return os; }..vo
0ed0: 69 64 20 76 65 72 69 66 79 5f 63 61 73 65 28 63 id verify_case(c
0ee0: 6f 6e 73 74 20 6c 6f 6e 67 20 6c 6f 6e 67 26 20 onst long long&
0ef0: 45 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 20 Expected, const
0f00: 6c 6f 6e 67 20 6c 6f 6e 67 26 20 52 65 63 65 69 long long& Recei
0f10: 76 65 64 29 20 7b 0d 0a 20 62 6f 6f 6c 20 6f 6b ved) {.. bool ok
0f20: 20 3d 20 28 45 78 70 65 63 74 65 64 20 3d 3d 20 = (Expected ==
0f30: 52 65 63 65 69 76 65 64 29 3b 0d 0a 20 69 66 28 Received);.. if(
0f40: 6f 6b 29 20 63 65 72 72 20 3c 3c 20 22 50 41 53 ok) cerr << "PAS
0f50: 53 45 44 22 20 3c 3c 20 74 69 6d 65 72 28 29 20 SED" << timer()
0f60: 3c 3c 20 65 6e 64 6c 3b 20 20 65 6c 73 65 20 7b << endl; else {
0f70: 20 63 65 72 72 20 3c 3c 20 22 46 41 49 4c 45 44 cerr << "FAILED
0f80: 22 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 " << timer() <<
0f90: 65 6e 64 6c 3b 0d 0a 20 63 65 72 72 20 3c 3c 20 endl;.. cerr <<
0fa0: 22 5c 74 6f 3a 20 5c 22 22 20 3c 3c 20 45 78 70 "\to: \"" << Exp
0fb0: 65 63 74 65 64 20 3c 3c 20 27 5c 22 27 20 3c 3c ected << '\"' <<
0fc0: 20 65 6e 64 6c 20 3c 3c 20 22 5c 74 78 3a 20 5c endl << "\tx: \
0fd0: 22 22 20 3c 3c 20 52 65 63 65 69 76 65 64 20 3c "" << Received <
0fe0: 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 < '\"' << endl;
0ff0: 7d 20 7d 0d 0a 23 64 65 66 69 6e 65 20 43 41 53 } }..#define CAS
1000: 45 28 4e 29 20 7b 63 65 72 72 20 3c 3c 20 22 54 E(N) {cerr << "T
1010: 65 73 74 20 43 61 73 65 20 23 22 20 3c 3c 20 4e est Case #" << N
1020: 20 3c 3c 20 22 2e 2e 2e 22 20 3c 3c 20 66 6c 75 << "..." << flu
1030: 73 68 3b 20 73 74 61 72 74 5f 74 69 6d 65 3d 63 sh; start_time=c
1040: 6c 6f 63 6b 28 29 3b 0d 0a 23 64 65 66 69 6e 65 lock();..#define
1050: 20 45 4e 44 09 20 76 65 72 69 66 79 5f 63 61 73 END. verify_cas
1060: 65 28 5f 2c 20 46 69 6e 64 69 6e 67 4b 69 64 73 e(_, FindingKids
1070: 28 29 2e 67 65 74 53 75 6d 28 6e 2c 20 71 2c 20 ().getSum(n, q,
1080: 41 2c 20 42 2c 20 43 29 29 3b 7d 0d 0a 69 6e 74 A, B, C));}..int
1090: 20 6d 61 69 6e 28 29 7b 0d 0a 0d 0a 43 41 53 45 main(){....CASE
10a0: 28 30 29 0d 0a 09 69 6e 74 20 6e 20 3d 20 35 3b (0)...int n = 5;
10b0: 20 0d 0a 09 69 6e 74 20 71 20 3d 20 32 3b 20 0d ...int q = 2; .
10c0: 0a 09 69 6e 74 20 41 20 3d 20 30 3b 20 0d 0a 09 ..int A = 0; ...
10d0: 69 6e 74 20 42 20 3d 20 31 3b 20 0d 0a 09 69 6e int B = 1; ...in
10e0: 74 20 43 20 3d 20 31 3b 20 0d 0a 09 6c 6f 6e 67 t C = 1; ...long
10f0: 20 6c 6f 6e 67 20 5f 20 3d 20 31 35 4c 4c 3b 20 long _ = 15LL;
1100: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 31 29 0d 0a ..END..CASE(1)..
1110: 09 69 6e 74 20 6e 20 3d 20 35 3b 20 0d 0a 09 69 .int n = 5; ...i
1120: 6e 74 20 71 20 3d 20 34 3b 20 0d 0a 09 69 6e 74 nt q = 4; ...int
1130: 20 41 20 3d 20 33 3b 20 0d 0a 09 69 6e 74 20 42 A = 3; ...int B
1140: 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 43 20 3d = 2; ...int C =
1150: 20 31 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 1; ...long long
1160: 20 5f 20 3d 20 34 33 33 37 36 4c 4c 3b 20 0d 0a _ = 43376LL; ..
1170: 45 4e 44 0d 0a 43 41 53 45 28 32 29 0d 0a 09 69 END..CASE(2)...i
1180: 6e 74 20 6e 20 3d 20 32 30 30 30 30 30 3b 20 0d nt n = 200000; .
1190: 0a 09 69 6e 74 20 71 20 3d 20 32 30 30 30 30 30 ..int q = 200000
11a0: 3b 20 0d 0a 09 69 6e 74 20 41 20 3d 20 31 32 33 ; ...int A = 123
11b0: 34 35 3b 20 0d 0a 09 69 6e 74 20 42 20 3d 20 36 45; ...int B = 6
11c0: 37 38 39 30 3b 20 0d 0a 09 69 6e 74 20 43 20 3d 7890; ...int C =
11d0: 20 31 31 31 32 31 33 31 34 31 3b 20 0d 0a 09 6c 111213141; ...l
11e0: 6f 6e 67 20 6c 6f 6e 67 20 5f 20 3d 20 31 33 33 ong long _ = 133
11f0: 33 37 38 34 30 38 34 32 38 32 33 37 4c 4c 3b 20 378408428237LL;
1200: 0d 0a 45 4e 44 0d 0a 2f 2a 0d 0a 43 41 53 45 28 ..END../*..CASE(
1210: 33 29 0d 0a 09 69 6e 74 20 6e 20 3d 20 3b 20 0d 3)...int n = ; .
1220: 0a 09 69 6e 74 20 71 20 3d 20 3b 20 0d 0a 09 69 ..int q = ; ...i
1230: 6e 74 20 41 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 nt A = ; ...int
1240: 42 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 43 20 3d B = ; ...int C =
1250: 20 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 ; ...long long
1260: 5f 20 3d 20 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 43 _ = LL; ..END..C
1270: 41 53 45 28 34 29 0d 0a 09 69 6e 74 20 6e 20 3d ASE(4)...int n =
1280: 20 3b 20 0d 0a 09 69 6e 74 20 71 20 3d 20 3b 20 ; ...int q = ;
1290: 0d 0a 09 69 6e 74 20 41 20 3d 20 3b 20 0d 0a 09 ...int A = ; ...
12a0: 69 6e 74 20 42 20 3d 20 3b 20 0d 0a 09 69 6e 74 int B = ; ...int
12b0: 20 43 20 3d 20 3b 20 0d 0a 09 6c 6f 6e 67 20 6c C = ; ...long l
12c0: 6f 6e 67 20 5f 20 3d 20 4c 4c 3b 20 0d 0a 45 4e ong _ = LL; ..EN
12d0: 44 0d 0a 2a 2f 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 D..*/..}..// END
12e0: 20 43 55 54 20 48 45 52 45 0d 0a CUT HERE..