QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#756099#7904. Rainbow SubarrayLoxilanteWA 1393ms7984kbC++202.3kb2024-11-16 19:01:482024-11-16 19:01:48

Judging History

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

  • [2024-11-16 19:01:48]
  • 评测
  • 测评结果:WA
  • 用时:1393ms
  • 内存:7984kb
  • [2024-11-16 19:01:48]
  • 提交

answer

#define F_C
#include <bits/stdc++.h>
#define rep(i, l, r) for(int i = l; i < r; i++)
#define hrp(i, l, r) for(int i = l; i <= r; i++)
#define rev(i, r, l) for(int i = r; i >= l; i--)
#define int ll
using namespace std;
typedef long long ll;
template<typename tn = int> tn next(void) { tn k; cin>>k; return k; }
#ifndef LOCAL
#define D(...) 0
#endif
const int U = 1e6;
const int kInf = 1e17;
int w[U], n, k, sum[U];
std::multiset<int> a, b;
int suma, sumb;
void init() {
  suma = sumb = 0;
  a.clear(), b.clear();
  a.insert(-kInf), b.insert(kInf);
}
void adjust() {
  while (a.size() > b.size() + 1) {
    std::multiset<int>::iterator it = (--a.end());
    b.insert(*it);
    a.erase(it);
    sumb += *it;
    suma -= *it;
  }
  while (b.size() > a.size()) {
    std::multiset<int>::iterator it = b.begin();
    a.insert(*it);
    b.erase(it);
    suma += *it;
    sumb -= *it;
  }
}
void add(int val_) {
  if (val_ <= *b.begin()) a.insert(val_), suma += val_;
  else b.insert(val_), sumb += val_;
  adjust();
}
void del(int val_) {
  std::multiset<int>::iterator it = a.lower_bound(val_);
  if (it != a.end()) {
    a.erase(it);
    suma -= *it;
  } else {
    it = b.lower_bound(val_);
    b.erase(it);
    sumb -= *it;
  }
  adjust();
}
int get_middle() {
  return *a.rbegin();
}
bool check(int len)
{
    init();
    hrp(i, 1, len) add(w[i]);
    int p1 = get_middle(), ans = (int)(((a.size()-1)*p1-suma)+(sumb-(b.size()-1)*p1));
    hrp(i, 1, n-len)
    {
        int p = get_middle();
        ans = min(ans, (int)(((a.size()-1)*p-suma)+(sumb-(b.size()-1)*p)));
        del(w[i]); add(w[i+len]);
    }
    return ans <= k;
}
signed main(void)
{
    #ifdef LOCAL
//	freopen("C:\\Users\\Loxil\\Desktop\\IN.txt", "r", stdin);
//	freopen("C:\\Users\\Loxil\\Desktop\\OUT.txt", "w", stdout);
    #endif
    
    ios::sync_with_stdio(false);
    cin.tie(0);
    
    int T = next();
    while(T--)
    {
        cin>>n>>k;
        hrp(i, 1, n) cin>>w[i], w[i] += n-i;

        int l = 1, r = n, ans = 1;
        while(l <= r)
        {
            int mid = l+r>>1;
            if (check(mid)) ans = max(ans, mid), l = mid+1;
            else r = mid-1;
        }
        cout<<ans<<endl;
    }
    
    return 0;
}
/*
1
5 6
1 1 1 1 1
 */

詳細信息

Test #1:

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

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: 1393ms
memory: 7984kb

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

result:

wrong answer 3rd lines differ - expected: '3', found: '2'