QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#708097#7904. Rainbow SubarrayXiaoYang3WA 939ms14636kbC++233.7kb2024-11-03 19:32:262024-11-03 19:32:27

Judging History

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

  • [2024-11-03 19:32:27]
  • 评测
  • 测评结果:WA
  • 用时:939ms
  • 内存:14636kb
  • [2024-11-03 19:32:26]
  • 提交

answer

#include <bits/stdc++.h>
using ll = long long;
using namespace std;
const int N = 5e5 + 5;
int n, m, na[N], nb[N], tot;
ll k;
struct node {
    int data;
    ll sum;
} tree[N * 4 + 50];
void pushup(int id) {
    tree[id].data = tree[id << 1].data + tree[id << 1 | 1].data;
    tree[id].sum = tree[id << 1].sum + tree[id << 1 | 1].sum;
}
// x,y,的cnt个数
int queryxy(int id, int l, int r, int ql, int qr) {
    if (ql <= l && r <= qr) {
        return tree[id].data;
    }
    int mid = (l + r) >> 1;
    if (qr <= mid) {
        return queryxy(id << 1, l, mid, ql, qr);
    } else if (ql > mid) {
        return queryxy(id << 1 | 1, mid + 1, r, ql, qr);
    } else {
        return queryxy(id << 1, l, mid, ql, qr) +
               queryxy(id << 1 | 1, mid + 1, r, ql, qr);
    }
}
// x,y,的sum
ll querysum(int id, int l, int r, int ql, int qr) {
    if (ql <= l && r <= qr) {
        return tree[id].sum;
    }
    int mid = (l + r) >> 1;
    if (qr <= mid) {
        return querysum(id << 1, l, mid, ql, qr);
    } else if (ql > mid) {
        return querysum(id << 1 | 1, mid + 1, r, ql, qr);
    } else {
        return querysum(id << 1, l, mid, ql, qr) +
               querysum(id << 1 | 1, mid + 1, r, ql, qr);
    }
}
void change(int id, int l, int r, int x, int cnt) {
    if (l == r) {
        tree[id].data += cnt;
        tree[id].sum = tree[id].data * nb[l];
        return;
    }
    int mid = (l + r) >> 1;
    if (x <= mid) {
        change(id << 1, l, mid, x, cnt);
    } else {
        change(id << 1 | 1, mid + 1, r, x, cnt);
    }
    pushup(id);
}
// 第cnt个的位置
int querycnt(int id, int l, int r, int cnt) {
    if (l == r) {
        return l;
    }
    int mid = l + r >> 1;
    if (cnt > tree[id << 1].data) {
        return querycnt(id << 1 | 1, mid + 1, r, cnt - tree[id << 1].data);
    } else {
        return querycnt(id << 1, l, mid, cnt);
    }
}

void solve() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        cin >> na[i];
        na[i] -= i;
        nb[i] = na[i];
    }
    sort(nb + 1, nb + 1 + n);
    int l = 1, r = n;
    for (int i = 1; i <= n; i++) {
        int pos = upper_bound(nb + 1, nb + 1 + n, nb[i]) - nb;
        l = max(l, pos - i + 1);
    }
    tot = unique(nb + 1, nb + 1 + n) - nb - 1;
    for (int i = 1; i <= n; i++) {
        na[i] = lower_bound(nb + 1, nb + 1 + tot, na[i]) - nb;
    }
    auto ck = [&](int mid) -> bool {
        ll res = 0;
        for (int i = 1; i <= tot * 4; i++) {
            tree[i].data = tree[i].sum = 0;
        }
        for (int i = 1; i < mid; i++) {
            change(1, 1, tot, na[i], 1);
        }
        for (int i = mid; i <= n; i++) {
            change(1, 1, tot, na[i], 1);
            int pos = querycnt(1, 1, tot, (mid + 1) / 2);
            ll su0 = 0, su1 = 0;
            if (pos > 1) {
                su0 = querysum(1, 1, tot, 1, pos - 1);
                int ct0 = queryxy(1, 1, tot, 1, pos - 1);
                su0 = nb[pos] * ct0 - su0;
            }
            su1 = querysum(1, 1, tot, pos, tot);
            int ct0 = queryxy(1, 1, tot, pos, tot);
            su1 = su1 - nb[pos] * ct0;
            if (su0 + su1 <= k) {
                // cout << pos << ' ' << su0 << ' ' << su1;
                return 1;
            }
            change(1, 1, tot, na[i - mid + 1], -1);
        }

        return 0;
    };

    while (l <= r) {
        int mid = l + r >> 1;
        if (ck(mid)) {
            l = mid + 1;
        } else {
            r = mid - 1;
        }
    }
    cout << l - 1 << '\n';
}
int main() {
    std::ios::sync_with_stdio(false);

    std::cin.tie(nullptr);
    int _ = 1;
    cin >> _;
    while (_--) {
        solve();
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 5592kb

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
Wrong Answer
time: 939ms
memory: 14636kb

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
7
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:

wrong answer 120th lines differ - expected: '8', found: '7'