#include <bits/stdc++.h>
using namespace std;
#define int long long
struct segtree{
int l, r, i; double m, b; int cnt;
segtree* lc = 0, *rc = 0;
segtree* getmem();
segtree(int l, int r) : l(l), r(r), m(0), b(1e9), cnt(0){
i = (l + r) / 2;
if(l==r) return;
lc = getmem(); *lc = segtree(l, i);
rc = getmem(); *rc = segtree(i+1, r);
};
segtree() : segtree(-1, -1){};
void u(double qm, double qb, int qcnt){
if(qm * i + qb < m*i+b){
swap(m, qm); swap(b, qb); swap(cnt, qcnt);
}
if(l == r) return;
if(qm > m) lc->u(qm, qb, qcnt);
else rc->u(qm, qb, qcnt);
}
pair<double, int> q(int qi){
pair<double, int> cur = {m*qi+b, cnt};
if(i == qi) return cur;
if(qi < i) return min(cur, lc->q(qi));
else return min(cur, rc->q(qi));
}
} mem[(int)1e7]; int memsz = 0;
segtree* segtree::getmem(){return &mem[memsz++];}
signed main(){
int n, k; cin >> n >> k;
string s; cin >> s;
double l = 0, r = 1e12;
double res = 0, m = 0;
int cnt = 0;
vector<int> b;
for(int i = 0; i < 2*n; i++){
if(s[i] == 'B') b.push_back(i);
}
while(r - l > 1e-6){
m = (l + r) / 2;
memsz = 0;
segtree ac(0, n);
long double sl = 0;
ac.u(0, 0, 0);
int cnta = 0;
long double cur = 0;
int funnynumber = 0;
int cntactive = 0;
for(int i = 0; i < 2 * n; i++){
if(s[i] == 'A'){
if(cntactive>0) funnynumber -= cntactive;
cntactive++;
if(cntactive>0) funnynumber += b[cnta] - i - cntactive;
cnta++;
cur += sl;
auto tmp = ac.q(cnta);
tmp.first += m-0.00001; tmp.second++;
if(cnta == n) {
res = tmp.first + cur;
cnt = tmp.second;
}
//if(r-l < 2e-6)
//cout<<i<<","<<cnta<<","<<cur<<","<<tmp.first+cur-tmp.second*m<<","<<tmp.second<<endl;
//if(r-l < 2e-6) cout<<funnynumber<<","<<b[cnta-1] - i - cntactive<<endl;
//if(r-l < 2e-6)
//cout<<cnta<<","<<tmp.first + (cnta)*cnta+funnynumber-tmp.second*m<<endl;
//cout<<cnta<<","<<tmp.first + (cnta + max(0LL, b[cnta-1] - i - cntactive))*cnta-funnynumber-tmp.second*m<<endl;
ac.u(-cnta, tmp.first + (cnta)*cnta+funnynumber, tmp.second);
}
else{
cntactive--;
sl++;
}
}
if(cnt < k) r = m;
else l = m;
}
//cout<<cnt<<","<<k<<endl;
cout<<(int)round(res-m*k)<<endl;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
183 ms |
626396 KB |
Output is correct |
2 |
Correct |
98 ms |
626516 KB |
Output is correct |
3 |
Correct |
98 ms |
626416 KB |
Output is correct |
4 |
Correct |
98 ms |
626768 KB |
Output is correct |
5 |
Correct |
95 ms |
626516 KB |
Output is correct |
6 |
Correct |
99 ms |
626516 KB |
Output is correct |
7 |
Correct |
91 ms |
626568 KB |
Output is correct |
8 |
Correct |
89 ms |
626516 KB |
Output is correct |
9 |
Correct |
87 ms |
626628 KB |
Output is correct |
10 |
Correct |
85 ms |
626468 KB |
Output is correct |
11 |
Correct |
87 ms |
626528 KB |
Output is correct |
12 |
Correct |
89 ms |
626520 KB |
Output is correct |
13 |
Correct |
88 ms |
626528 KB |
Output is correct |
14 |
Correct |
86 ms |
626512 KB |
Output is correct |
15 |
Correct |
88 ms |
626604 KB |
Output is correct |
16 |
Correct |
86 ms |
626512 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
183 ms |
626396 KB |
Output is correct |
2 |
Correct |
98 ms |
626516 KB |
Output is correct |
3 |
Correct |
98 ms |
626416 KB |
Output is correct |
4 |
Correct |
98 ms |
626768 KB |
Output is correct |
5 |
Correct |
95 ms |
626516 KB |
Output is correct |
6 |
Correct |
99 ms |
626516 KB |
Output is correct |
7 |
Correct |
91 ms |
626568 KB |
Output is correct |
8 |
Correct |
89 ms |
626516 KB |
Output is correct |
9 |
Correct |
87 ms |
626628 KB |
Output is correct |
10 |
Correct |
85 ms |
626468 KB |
Output is correct |
11 |
Correct |
87 ms |
626528 KB |
Output is correct |
12 |
Correct |
89 ms |
626520 KB |
Output is correct |
13 |
Correct |
88 ms |
626528 KB |
Output is correct |
14 |
Correct |
86 ms |
626512 KB |
Output is correct |
15 |
Correct |
88 ms |
626604 KB |
Output is correct |
16 |
Correct |
86 ms |
626512 KB |
Output is correct |
17 |
Correct |
92 ms |
626416 KB |
Output is correct |
18 |
Correct |
91 ms |
626516 KB |
Output is correct |
19 |
Correct |
89 ms |
626584 KB |
Output is correct |
20 |
Correct |
88 ms |
626516 KB |
Output is correct |
21 |
Correct |
89 ms |
626512 KB |
Output is correct |
22 |
Correct |
91 ms |
626512 KB |
Output is correct |
23 |
Correct |
92 ms |
626772 KB |
Output is correct |
24 |
Correct |
90 ms |
626568 KB |
Output is correct |
25 |
Correct |
90 ms |
626516 KB |
Output is correct |
26 |
Correct |
89 ms |
626512 KB |
Output is correct |
27 |
Correct |
88 ms |
626404 KB |
Output is correct |
28 |
Correct |
88 ms |
626512 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
183 ms |
626396 KB |
Output is correct |
2 |
Correct |
98 ms |
626516 KB |
Output is correct |
3 |
Correct |
98 ms |
626416 KB |
Output is correct |
4 |
Correct |
98 ms |
626768 KB |
Output is correct |
5 |
Correct |
95 ms |
626516 KB |
Output is correct |
6 |
Correct |
99 ms |
626516 KB |
Output is correct |
7 |
Correct |
91 ms |
626568 KB |
Output is correct |
8 |
Correct |
89 ms |
626516 KB |
Output is correct |
9 |
Correct |
87 ms |
626628 KB |
Output is correct |
10 |
Correct |
85 ms |
626468 KB |
Output is correct |
11 |
Correct |
87 ms |
626528 KB |
Output is correct |
12 |
Correct |
89 ms |
626520 KB |
Output is correct |
13 |
Correct |
88 ms |
626528 KB |
Output is correct |
14 |
Correct |
86 ms |
626512 KB |
Output is correct |
15 |
Correct |
88 ms |
626604 KB |
Output is correct |
16 |
Correct |
86 ms |
626512 KB |
Output is correct |
17 |
Correct |
92 ms |
626416 KB |
Output is correct |
18 |
Correct |
91 ms |
626516 KB |
Output is correct |
19 |
Correct |
89 ms |
626584 KB |
Output is correct |
20 |
Correct |
88 ms |
626516 KB |
Output is correct |
21 |
Correct |
89 ms |
626512 KB |
Output is correct |
22 |
Correct |
91 ms |
626512 KB |
Output is correct |
23 |
Correct |
92 ms |
626772 KB |
Output is correct |
24 |
Correct |
90 ms |
626568 KB |
Output is correct |
25 |
Correct |
90 ms |
626516 KB |
Output is correct |
26 |
Correct |
89 ms |
626512 KB |
Output is correct |
27 |
Correct |
88 ms |
626404 KB |
Output is correct |
28 |
Correct |
88 ms |
626512 KB |
Output is correct |
29 |
Correct |
133 ms |
626768 KB |
Output is correct |
30 |
Correct |
130 ms |
626632 KB |
Output is correct |
31 |
Correct |
130 ms |
626520 KB |
Output is correct |
32 |
Correct |
109 ms |
626724 KB |
Output is correct |
33 |
Correct |
118 ms |
626768 KB |
Output is correct |
34 |
Correct |
128 ms |
626764 KB |
Output is correct |
35 |
Correct |
127 ms |
626768 KB |
Output is correct |
36 |
Correct |
129 ms |
626576 KB |
Output is correct |
37 |
Correct |
133 ms |
626708 KB |
Output is correct |
38 |
Correct |
127 ms |
626516 KB |
Output is correct |
39 |
Correct |
130 ms |
626768 KB |
Output is correct |
40 |
Correct |
136 ms |
626516 KB |
Output is correct |
41 |
Correct |
132 ms |
626512 KB |
Output is correct |
42 |
Correct |
119 ms |
626560 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
183 ms |
626396 KB |
Output is correct |
2 |
Correct |
98 ms |
626516 KB |
Output is correct |
3 |
Correct |
98 ms |
626416 KB |
Output is correct |
4 |
Correct |
98 ms |
626768 KB |
Output is correct |
5 |
Correct |
95 ms |
626516 KB |
Output is correct |
6 |
Correct |
99 ms |
626516 KB |
Output is correct |
7 |
Correct |
91 ms |
626568 KB |
Output is correct |
8 |
Correct |
89 ms |
626516 KB |
Output is correct |
9 |
Correct |
87 ms |
626628 KB |
Output is correct |
10 |
Correct |
85 ms |
626468 KB |
Output is correct |
11 |
Correct |
87 ms |
626528 KB |
Output is correct |
12 |
Correct |
89 ms |
626520 KB |
Output is correct |
13 |
Correct |
88 ms |
626528 KB |
Output is correct |
14 |
Correct |
86 ms |
626512 KB |
Output is correct |
15 |
Correct |
88 ms |
626604 KB |
Output is correct |
16 |
Correct |
86 ms |
626512 KB |
Output is correct |
17 |
Correct |
92 ms |
626416 KB |
Output is correct |
18 |
Correct |
91 ms |
626516 KB |
Output is correct |
19 |
Correct |
89 ms |
626584 KB |
Output is correct |
20 |
Correct |
88 ms |
626516 KB |
Output is correct |
21 |
Correct |
89 ms |
626512 KB |
Output is correct |
22 |
Correct |
91 ms |
626512 KB |
Output is correct |
23 |
Correct |
92 ms |
626772 KB |
Output is correct |
24 |
Correct |
90 ms |
626568 KB |
Output is correct |
25 |
Correct |
90 ms |
626516 KB |
Output is correct |
26 |
Correct |
89 ms |
626512 KB |
Output is correct |
27 |
Correct |
88 ms |
626404 KB |
Output is correct |
28 |
Correct |
88 ms |
626512 KB |
Output is correct |
29 |
Correct |
133 ms |
626768 KB |
Output is correct |
30 |
Correct |
130 ms |
626632 KB |
Output is correct |
31 |
Correct |
130 ms |
626520 KB |
Output is correct |
32 |
Correct |
109 ms |
626724 KB |
Output is correct |
33 |
Correct |
118 ms |
626768 KB |
Output is correct |
34 |
Correct |
128 ms |
626764 KB |
Output is correct |
35 |
Correct |
127 ms |
626768 KB |
Output is correct |
36 |
Correct |
129 ms |
626576 KB |
Output is correct |
37 |
Correct |
133 ms |
626708 KB |
Output is correct |
38 |
Correct |
127 ms |
626516 KB |
Output is correct |
39 |
Correct |
130 ms |
626768 KB |
Output is correct |
40 |
Correct |
136 ms |
626516 KB |
Output is correct |
41 |
Correct |
132 ms |
626512 KB |
Output is correct |
42 |
Correct |
119 ms |
626560 KB |
Output is correct |
43 |
Correct |
715 ms |
627728 KB |
Output is correct |
44 |
Correct |
1172 ms |
628060 KB |
Output is correct |
45 |
Correct |
1153 ms |
628196 KB |
Output is correct |
46 |
Correct |
697 ms |
628056 KB |
Output is correct |
47 |
Correct |
876 ms |
628060 KB |
Output is correct |
48 |
Incorrect |
1122 ms |
628052 KB |
Output isn't correct |
49 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
183 ms |
626396 KB |
Output is correct |
2 |
Correct |
98 ms |
626516 KB |
Output is correct |
3 |
Correct |
98 ms |
626416 KB |
Output is correct |
4 |
Correct |
98 ms |
626768 KB |
Output is correct |
5 |
Correct |
95 ms |
626516 KB |
Output is correct |
6 |
Correct |
99 ms |
626516 KB |
Output is correct |
7 |
Correct |
91 ms |
626568 KB |
Output is correct |
8 |
Correct |
89 ms |
626516 KB |
Output is correct |
9 |
Correct |
87 ms |
626628 KB |
Output is correct |
10 |
Correct |
85 ms |
626468 KB |
Output is correct |
11 |
Correct |
87 ms |
626528 KB |
Output is correct |
12 |
Correct |
89 ms |
626520 KB |
Output is correct |
13 |
Correct |
88 ms |
626528 KB |
Output is correct |
14 |
Correct |
86 ms |
626512 KB |
Output is correct |
15 |
Correct |
88 ms |
626604 KB |
Output is correct |
16 |
Correct |
86 ms |
626512 KB |
Output is correct |
17 |
Correct |
92 ms |
626416 KB |
Output is correct |
18 |
Correct |
91 ms |
626516 KB |
Output is correct |
19 |
Correct |
89 ms |
626584 KB |
Output is correct |
20 |
Correct |
88 ms |
626516 KB |
Output is correct |
21 |
Correct |
89 ms |
626512 KB |
Output is correct |
22 |
Correct |
91 ms |
626512 KB |
Output is correct |
23 |
Correct |
92 ms |
626772 KB |
Output is correct |
24 |
Correct |
90 ms |
626568 KB |
Output is correct |
25 |
Correct |
90 ms |
626516 KB |
Output is correct |
26 |
Correct |
89 ms |
626512 KB |
Output is correct |
27 |
Correct |
88 ms |
626404 KB |
Output is correct |
28 |
Correct |
88 ms |
626512 KB |
Output is correct |
29 |
Correct |
133 ms |
626768 KB |
Output is correct |
30 |
Correct |
130 ms |
626632 KB |
Output is correct |
31 |
Correct |
130 ms |
626520 KB |
Output is correct |
32 |
Correct |
109 ms |
626724 KB |
Output is correct |
33 |
Correct |
118 ms |
626768 KB |
Output is correct |
34 |
Correct |
128 ms |
626764 KB |
Output is correct |
35 |
Correct |
127 ms |
626768 KB |
Output is correct |
36 |
Correct |
129 ms |
626576 KB |
Output is correct |
37 |
Correct |
133 ms |
626708 KB |
Output is correct |
38 |
Correct |
127 ms |
626516 KB |
Output is correct |
39 |
Correct |
130 ms |
626768 KB |
Output is correct |
40 |
Correct |
136 ms |
626516 KB |
Output is correct |
41 |
Correct |
132 ms |
626512 KB |
Output is correct |
42 |
Correct |
119 ms |
626560 KB |
Output is correct |
43 |
Correct |
715 ms |
627728 KB |
Output is correct |
44 |
Correct |
1172 ms |
628060 KB |
Output is correct |
45 |
Correct |
1153 ms |
628196 KB |
Output is correct |
46 |
Correct |
697 ms |
628056 KB |
Output is correct |
47 |
Correct |
876 ms |
628060 KB |
Output is correct |
48 |
Incorrect |
1122 ms |
628052 KB |
Output isn't correct |
49 |
Halted |
0 ms |
0 KB |
- |