QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#454353 | #6338. Chorus | snpmrnhlol | 0 | 0ms | 9824kb | C++14 | 2.4kb | 2024-06-24 20:02:04 | 2024-06-24 20:02:04 |
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;
};
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){
while(sz >= 2 && check(cht[cnt - 2],cht[cnt - 1],{k,b})){
cht.pop_back();
sz--;
}
cht.push_back({k,b});
sz++;
}
ll get(ll x){
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;
}
};
pair <ll,ll> check(ll x){
//cout<<"checking:"<<x<<'\n';
dp[0] = {0,0};
for(ll i = 1;i <= n;i++){
dp[i] = {inf,0};
for(ll j = i - 1;j >= 0;j--){
if(v2[j] >= 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 - v2[j])*(n - j) - (ps[i] - ps[v2[j]]),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';
}
}
//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;
ans = dp[n].first - dp[n].second*mij;
}
}
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: 0ms
memory: 9744kb
input:
1 1 BA
output:
1
result:
ok 1 number(s): "1"
Test #2:
score: -16
Wrong Answer
time: 0ms
memory: 9824kb
input:
7 5 ABBAAABBABABBA
output:
4
result:
wrong answer 1st numbers differ - expected: '3', found: '4'
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%