QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#706326#7904. Rainbow Subarrayq1w2e3r4#RE 0ms3716kbC++142.9kb2024-11-03 10:34:282024-11-03 10:34:28

Judging History

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

  • [2024-11-03 10:34:28]
  • 评测
  • 测评结果:RE
  • 用时:0ms
  • 内存:3716kb
  • [2024-11-03 10:34:28]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;

const int MAXN = 1e6 + 10;
int T,N,K;

map<int,int> mp;
int a[MAXN];

int mid = 0;
int lsum = 0, rsum = 0;
int lcnt = 0, rcnt = 0, midcnt = 0;

int add(int x){
    if(midcnt == 0){
        mid = x;
        mp[mid] = 1;
        midcnt = 1;
        return 0;
    }
    mp[x]++;
    if(x < mid){
        lsum += x;
        lcnt++;
        if(lcnt > midcnt + rcnt){
            auto it = prev(mp.lower_bound(mid));
            int y = it->first, ycnt = it->second;
            lsum -= y * ycnt;
            lcnt -= ycnt;
            rsum += mid * midcnt;
            rcnt += midcnt;
            mid = y;
            midcnt = ycnt;
        }
        return lcnt * mid - lsum + rsum - rcnt * mid;
    }
    else if(x > mid){
        rsum += x;
        rcnt++;
        if(rcnt > midcnt + lcnt){
            auto it = next(mp.lower_bound(mid));
            int y = it->first, ycnt = it->second;
            rsum -= y * ycnt;
            rcnt -= ycnt;
            lsum += mid * midcnt;
            lcnt += midcnt;
            mid = y;
            midcnt = ycnt;
        }
        return lcnt * mid - lsum + rsum - rcnt * mid;
    }
    else if(x == mid){
        midcnt++;
        return lcnt * mid - lsum + rsum - rcnt * mid;
    }
    assert(0);
}

int del(int x){
    mp[x]--;
    if(mp[x] == 0) mp.erase(x);
    int flag = 0;
    if(x < mid){
        lsum -= x;
        lcnt--;
        if(rcnt > midcnt + lcnt){
            auto it = next(mp.lower_bound(mid));
            int y = it->first, ycnt = it->second;
            rsum -= y * ycnt;
            rcnt -= ycnt;
            lsum += mid * midcnt;
            lcnt += midcnt;
            mid = y;
            midcnt = ycnt;
        }
        return lcnt * mid - lsum + rsum - rcnt * mid;
    }
    else if(x > mid){
        rsum -= x;
        rcnt--;
        if(lcnt > midcnt + rcnt){
            auto it = prev(mp.lower_bound(mid));
            int y = it->first, ycnt = it->second;
            lsum -= y * ycnt;
            lcnt -= ycnt;
            rsum += mid * midcnt;
            rcnt += midcnt;
            mid = y;
            midcnt = ycnt;
        }
        return lcnt * mid - lsum + rsum - rcnt * mid;
    }
    else if(x == mid){
        midcnt--;
        return lcnt * mid - lsum + rsum - rcnt * mid;
    }
    if(flag) mp.erase(x);
    assert(0);
}

void prepare(){
    cin >> N >> K;
    for(int i = 1; i <= N; i++){
        cin >> a[i];
        a[i] -= i;
    }
    int r = 0;
    int ans = 0;
    for(int i = 1; i <= N; i++){
        while(r < N){
            if(add(a[r+1]) > K){
                del(a[r+1]);
                break;
            }
            r++;
        }
        ans = max(ans, r - i + 1);
        del(a[i]);
    }
    cout << ans << "\n";
}

int main(){
    cin >> T;
    while(T--){
        prepare();
    }
}

详细

Test #1:

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

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
Runtime Error

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
6
7
2
4
1
1
1
1
3
1
1
6
6
5
6
1

result: