QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#454409 | #6338. Chorus | snpmrnhlol | 0 | 1ms | 9812kb | C++14 | 2.8kb | 2024-06-24 21:16:51 | 2024-06-24 21:16:52 |
answer
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 1e6;
const ll inf = 1e18;
char v[N*2];
ll v2[N],v3[N];
ll ps[N + 1];
pair<ll,ll> dp[N + 1];
ll n,k;
ll cnt = 0,cnt2 = 0;
struct line{
ll k,b,id;
};
bool check(line a,line b,line c){
return (a.b - b.b)*(b.k - c.k) <= (b.b - c.b)*(a.k - b.k);
}
struct CHT{
vector <line> cht;
ll sz = 0, pt = 0;
void init(){
cht.clear();
sz = 0;
pt = 0;
}
void add(ll k,ll b,ll id){
while(sz >= 2 && check(cht[sz - 2],cht[sz - 1],{k,b,id})){
cht.pop_back();
sz--;
}
cht.push_back({k,b,id});
sz++;
}
pair<ll,ll> get(ll x){
assert(sz != 0);
pt = min(pt,sz - 1);
while(pt < sz - 1 && x*(cht[pt].k - cht[pt + 1].k) <= -(cht[pt].b - cht[pt + 1].b)){
pt++;
}
return {cht[pt].k*x + cht[pt].b,cht[pt].id};
}
};
CHT cht;
pair <ll,ll> check(ll x){
//cout<<"checking:"<<x<<'\n';
dp[0] = {0,0};
cht.init();
ll pt = 0;
for(ll i = 1;i <= n;i++){
dp[i] = {inf,0};
for(ll j = i - 1;j >= 0;j--){
ll v2j = max(v2[j],j);
if(v2j >= i){
dp[i] = min(dp[i],{dp[j].first + x,dp[j].second + 1});
//cout<<i<<"->"<<j<<' '<<dp[j].first + x<<' '<<dp[j].second + 1<<'\n';
}/*else{
dp[i] = min(dp[i],{dp[j].first + x + (i - v2j)*(n - j) - (ps[i] - ps[v2j]),dp[j].second + 1});
//cout<<i<<"->"<<j<<' '<<dp[j].first<<' '<<x<<' '<<(i - v2[j])*(n - j)<<' '<<(ps[i] - ps[v2[j]])<<' '<<dp[j].second + 1<<'\n';
}*/
}
while(pt < n && max(v2[pt],pt) < i){
ll v2pt = max(v2[pt],pt);
cht.add(-pt,dp[pt].first + x - v2pt*(n - pt) + ps[v2pt],dp[pt].second);
pt++;
}
if(cht.sz != 0){
auto id = cht.get(i);
dp[i] = min(dp[i],{id.first + i*n - ps[i],id.second + 1});
}
//cout<<dp[i].first<<' '<<dp[i].second<<' '<<x<<' '<<i<<'\n';
}
return dp[n];
//cout<<'\n';
}
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
cin>>n>>k;
for(ll i = 0;i < 2*n;i++){
cin>>v[i];
if(v[i] == 'B'){
v2[cnt] = i - cnt;
cnt++;
}
if(v[i] == 'A'){
v3[cnt2] = i - cnt2;
cnt2++;
}
}
for(ll i = 1;i <= n;i++){
ps[i] = ps[i - 1] + (n - v3[i - 1]);
}
ll l = 0,r = inf;
ll ans = 0;
while(l != r){
ll mij = (l + r)/2;
auto x = check(mij);
if(x.second >= k){
l = mij + 1;
}else{
r = mij;
}
}
check(l);
ans = dp[n].first - k*l;
cout<<ans;
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 16
Accepted
time: 1ms
memory: 9812kb
input:
1 1 BA
output:
1
result:
ok 1 number(s): "1"
Test #2:
score: -16
Wrong Answer
time: 1ms
memory: 9808kb
input:
7 5 ABBAAABBABABBA
output:
22
result:
wrong answer 1st numbers differ - expected: '3', found: '22'
Subtask #2:
score: 0
Skipped
Dependency #1:
0%
Subtask #3:
score: 0
Skipped
Dependency #1:
0%
Subtask #4:
score: 0
Skipped
Dependency #1:
0%
Subtask #5:
score: 0
Skipped
Dependency #1:
0%