QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#795716#7904. Rainbow Subarrayimwinter#TL 0ms3588kbC++201.8kb2024-11-30 23:47:162024-11-30 23:47:16

Judging History

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

  • [2024-11-30 23:47:16]
  • 评测
  • 测评结果:TL
  • 用时:0ms
  • 内存:3588kb
  • [2024-11-30 23:47:16]
  • 提交

answer

#include <bits/stdc++.h>

using i64 = long long;

void sol() {
    int n;
    std::cin >> n;

    i64 k;
    std::cin >> k;

    std::vector<int> a(n);
    for (int i = 0 ; i < n; i++) {
        std::cin >> a[i];
        a[i] -= i;
    }

    auto cal= [&](std::priority_queue<int>& lo, std::priority_queue<int, std::vector<int>, std::greater<int>>& hi) {
        if (lo.size() > hi.size() + 1) {
            hi.push(lo.top());
            lo.pop();
        } else if (hi.size() > lo.size()) {
            lo.push(hi.top());
            hi.pop();
        }
        return lo.top();
    };

    int ans = 0;
    int lo = 0;
    std::unordered_map<int, int> freq;
    std::priority_queue<int> q;
    std::priority_queue<int, std::vector<int>, std::greater<int>> gq;
    
    for (int hi = 0; hi < n; ++hi) {
        q.push(a[hi]);
        int mid = cal(q, gq);
        freq[a[hi]]++;
        
        int opt = 0;
        for (auto& _ : freq) {
            opt += abs(_.first - mid) * _.second;
        }

        while (opt > k) {
            freq[a[lo]]--;
            if (freq[a[lo]] == 0) {
                freq.erase(a[lo]);
            }
            lo++;
            q = std::priority_queue<int>();
            gq = std::priority_queue<int, std::vector<int>, std::greater<int>>();
            for (int i = lo; i <= hi; ++i) {
                q.push(a[i]);
                mid = cal(q, gq);
            }
            
            opt = 0;
            for (auto& _ : freq) {
                opt += abs(_.first - mid) * _.second;
            }
        }

        ans = std::max(ans, hi - lo + 1);
    }
    std::cout << ans << "\n";
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int _;
    std::cin >>_;

    while (_--) {
        sol();
    }

    return 0;
}

詳細信息

Test #1:

score: 100
Accepted
time: 0ms
memory: 3588kb

input:

5
7 5
7 2 5 5 4 11 7
6 0
100 3 4 5 99 100
5 6
1 1 1 1 1
5 50
100 200 300 400 500
1 100
3

output:

4
3
5
1
1

result:

ok 5 lines

Test #2:

score: -100
Time Limit Exceeded

input:

11102
2 167959139
336470888 134074578
5 642802746
273386884 79721198 396628655 3722503 471207868
6 202647942
268792718 46761498 443917727 16843338 125908043 191952768
2 717268783
150414369 193319712
6 519096230
356168102 262263554 174936674 407246545 274667941 279198849
9 527268921
421436316 3613460...

output:

1
4
2
2
6
5
6
2
4
1
4
1
1
3
2
2
6
8
7
7
1
7
6
2
4
3
1
6
7
7
3
4
3
8
3
8
5
6
3
1
6
3
1
2
4
6
4
6
4
1
4
7
1
6
3
5
6
6
1
7
4
3
1
6
4
4
3
2
2
6
2
3
10
1
4
3
2
4
4
1
6
5
5
4
7
5
3
6
3
4
5
8
5
4
5
2
1
5
2
3
3
4
8
1
3
1
2
2
8
3
1
6
8
1
8
4
5
6
6
8
4
8
3
2
8
4
5
5
2
6
2
4
1
5
4
5
3
2
4
1
2
1
4
5
7
3
7
3
3
3...

result: