QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#602427#6338. ChorusDimash0 0ms3904kbC++233.6kb2024-10-01 00:54:282024-10-01 00:54:29

Judging History

你现在查看的是最新测评结果

  • [2024-10-01 00:54:29]
  • 评测
  • 测评结果:0
  • 用时:0ms
  • 内存:3904kb
  • [2024-10-01 00:54:28]
  • 提交

answer

#include <bits/stdc++.h>
    
using namespace std;
    
typedef long long ll;
const int  N = 5e3 + 12, MOD = (int)1e9 + 7;
 
int n, k, x[N], y[N], pref[ 2 * N], col[N], opt[N], sum[N];
ll dp[N], c[N][N], add = 0;

void make(string &s) {
    string t;
    deque<int> bf;
    int bal = 0;
    for(int i = 0; i < n + n; i++) {
        if(s[i] == 'A') {
            bal++;
            t += 'A';
            if(!bf.empty()) {
                int j = bf[0];
                add += (int)bf.size();
                t += 'B';
                bal--;
                bf.pop_front();
            }
        } else {
            if(bal) {
                bal--;
                t += 'B';
            } else {
                bf.push_back(i);
            }
        }
    }
    s = t;
}
void prec() {
    for(int i = 1; i <= n; i++) {
        for(int j = i; j >= 1; j--) {
            c[j][i] = c[j + 1][i];
            if(y[j] <= x[i]) {
                c[j][i] += pref[x[i]] - pref[y[j] - 1];;
            }
            ll cnt = pref[x[i]] - pref[y[j] - 1], cst = - cnt * (cnt - 1) / 2;
            cst += x[i] * cnt;
            cst -= (sum[x[i]] - sum[y[j] - 1]);
        }
    }
    for(int i = 1; i < n; i++) {
        for(int j = 1; j < n; j++) {
            assert(c[i][j + 1] + c[i + 1][j] >= c[i][j] + c[i + 1][j + 1]);
        }
    }
}
pair<ll, ll> calc_slow(ll pen) {
    for(int i = 1; i <= n; i++) {
        dp[i] = 1e18;
        col[i] = n + 1;
        for(int j = i; j >= 1; j--) {
            if(dp[j - 1] + c[j][i] + pen <= dp[i]) {
                dp[i] = dp[j - 1] + c[j][i] + pen;
                col[i] = col[j - 1] + 1;
                opt[i] = j;
            }
        }
    }
    return {dp[n], col[n]};
}

ll str[N], val[N], str_[N], val_[N];
void dodo(string &s) {
    string t;
    deque<int> c, d;
    for(int i = 0; i < n + n; i++) {
        if(s[i] == 'A') {
            c.push_back(i);
        } else {
            d.push_back(i);
        }
    }
    vector<pair<int, char>> xx;
    for(int i = 0; i < n; ) {
        int j = i + 1, col = 1;
        ll f = d[i] + 1;
        while(j < n && c[j] < d[i]) {
            f += d[j] + 1;
            j++;
            col++;
        }
        str_[d[i] + 1] = col;
        val_[d[i] + 1] = f;
        xx.push_back({c[i], 'A'});
        xx.push_back({d[i], 'B'});
        i = j;
    }
    sort(xx.begin(), xx.end());
    for(int i = 0; i < (int)xx.size(); i++) {
        t += xx[i].second;
        if(xx[i].second == 'B') {
            str[i + 1] = str_[xx[i].first + 1];
            val[i + 1] = val[xx[i].first + 1];
        }
    }
    s = t;
    n = (int)s.size() / 2;
}
void test() {
    cin >> n >> k;
    int _x = 0, _y = 0;
    string s;
    cin >> s;
    // cout << s << '\n';
    make(s);
    // cout << s << '\n';
    dodo(s);
    // cout << s << '\n';
    for(int i = 1; i <= n + n; i++) {
        char a = s[i - 1];
        pref[i] = pref[i - 1];
        sum[i] = sum[i - 1];
        if(a == 'A') {
            x[++_x] = i;
        } else { 
            y[++_y] = i;
            pref[i] += str[i];
            sum[i] += val[i];
        }
    }
    prec();
    ll l = -1, r = 1e12, res;
    while(r - l > 1) { 
        ll mid = (l + r) / 2;
        auto [val, t] = calc_slow(mid);
        if(t <= k) {
            r = mid;
            res = val - k * 1ll * mid;
        } else {
            l = mid;
        }
    }
    cout << res + add << '\n';
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); 
    
    int t = 1; 
    // cin >> t;
    
    while(t--) 
        test();
    return 0;
}

詳細信息

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 16
Accepted
time: 0ms
memory: 3904kb

input:

1 1
BA

output:

1

result:

ok 1 number(s): "1"

Test #2:

score: 16
Accepted
time: 0ms
memory: 3668kb

input:

7 5
ABBAAABBABABBA

output:

3

result:

ok 1 number(s): "3"

Test #3:

score: 16
Accepted
time: 0ms
memory: 3752kb

input:

10 3
BABBABAABAABBABBBAAA

output:

26

result:

ok 1 number(s): "26"

Test #4:

score: 0
Wrong Answer
time: 0ms
memory: 3624kb

input:

10 2
AAABBABABBAAABBBAABB

output:

6

result:

wrong answer 1st numbers differ - expected: '11', found: '6'

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%