QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#756795#7904. Rainbow SubarraywkrTL 0ms3676kbC++203.5kb2024-11-16 21:58:292024-11-16 21:58:32

Judging History

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

  • [2024-11-16 21:58:32]
  • 评测
  • 测评结果:TL
  • 用时:0ms
  • 内存:3676kb
  • [2024-11-16 21:58:29]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define close ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)

const int N = 1e5 + 5;

int n, k, m;
vector<int> input;
int a[N];

struct SEG
{
#define lson id << 1
#define rson id << 1 | 1
    int sum;
    int sz;
} seg[N << 2];

int getIndex(int v)
{
    return lower_bound(input.begin(), input.end(), v) - input.begin() + 1;
}

void up(int id)
{
    seg[id].sum = seg[lson].sum + seg[rson].sum;
    seg[id].sz = seg[lson].sz + seg[rson].sz;
}

void build(int id, int l, int r)
{
    seg[id].sum = 0;
    seg[id].sz = 0;
    if (l == r)
        return;
    int mid = (l + r) >> 1;
    build(lson, l, mid);
    build(rson, mid + 1, r);
}

void change(int id, int l, int r, int x, int v, int le)
{
    if (l == r)
    {
        seg[id].sum += v;
        seg[id].sz += le;
        return;
    }
    int mid = (l + r) >> 1;
    if (x <= mid)
        change(lson, l, mid, x, v, le);
    else
        change(rson, mid + 1, r, x, v, le);
    up(id);
}

int querySum(int id, int l, int r, int ql, int qr)
{
    if (ql <= l && r <= qr)
        return seg[id].sum;
    int mid = (l + r) >> 1, res = 0;
    if (ql <= mid)
        res += querySum(lson, l, mid, ql, qr);
    if (qr > mid)
        res += querySum(rson, mid + 1, r, ql, qr);
    return res;
}

int querySz(int id, int l, int r, int ql, int qr)
{
    if (ql <= l && r <= qr)
        return seg[id].sz;
    int mid = (l + r) >> 1, res = 0;
    if (ql <= mid)
        res += querySz(lson, l, mid, ql, qr);
    if (qr > mid)
        res += querySz(rson, mid + 1, r, ql, qr);
    return res;
}

int findKth(int id, int l, int r, int index)
{
    if (l == r)
        return l;
    int mid = (l + r) >> 1;
    if (seg[lson].sz >= index)
        return findKth(lson, l, mid, index);
    else
        return findKth(rson, mid + 1, r, index - seg[lson].sz);
}

bool check(int len)
{
    build(1, 1, m);
    for (int i = 1; i <= len; i++)
    {
        change(1, 1, m, getIndex(a[i]), a[i], 1);
    }
    int index = findKth(1, 1, m, (len + 1) / 2);
    int nowMid = input[index - 1];
    int cnt = querySz(1, 1, m, 1, index) * nowMid - querySum(1, 1, m, 1, index) +
              querySum(1, 1, m, index, m) - querySz(1, 1, m, index, m) * nowMid;
    for (int i = 1; i + len <= n; i++)
    {
        change(1, 1, m, getIndex(a[i]), -a[i], -1);
        change(1, 1, m, getIndex(a[i + len]), a[i + len], 1);
        index = findKth(1, 1, m, (len + 1) / 2);
        nowMid = input[index - 1];
        int ans = querySz(1, 1, m, 1, index) * nowMid - querySum(1, 1, m, 1, index) +
                  querySum(1, 1, m, index, m) - querySz(1, 1, m, index, m) * nowMid;
        cnt = min(cnt, ans);
    }
    return cnt <= k;
}

void solved()
{
    cin >> n >> k;
    input.clear();
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        a[i] -= i;
        input.push_back(a[i]);
    }
    sort(input.begin(), input.end());
    input.erase(unique(input.begin(), input.end()), input.end());
    m = input.size();
    int l = 1, r = n, ans = 0;
    while (l <= r)
    {
        int mid = (l + r) >> 1;
        if (check(mid))
        {
            ans = mid;
            l = mid + 1;
        }
        else
        {
            r = mid - 1;
        }
    }
    cout << ans << endl;
}

signed main()
{
    close;
    int T;
    cin >> T;
    while (T--)
    {
        solved();
    }
    return 0;
}

詳細信息

Test #1:

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

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
3
2
6
5
7
2
4
1
4
1
1
3
2
2
7
8
7
7
1
7
6
2
4
3
1
6
7
7
3
4
3
9
3
8
6
6
3
1
6
3
1
2
4
6
4
6
4
1
4
7
1
6
3
5
6
6
1
7
5
3
1
6
4
5
3
2
2
6
2
3
10
1
4
3
2
4
5
1
7
5
5
5
8
5
3
6
3
5
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
6
2
6
2
4
1
5
4
5
3
2
4
1
2
1
4
5
8
3
7
3
3
3...

result: