QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#693895#7904. Rainbow SubarrayczrqWA 142ms5924kbC++171.4kb2024-10-31 16:55:372024-10-31 16:55:38

Judging History

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

  • [2024-10-31 16:55:38]
  • 评测
  • 测评结果:WA
  • 用时:142ms
  • 内存:5924kb
  • [2024-10-31 16:55:37]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std; 
typedef long long ll;
const int M = 5e5 + 10;
int t,n,k;
int a[M];
multiset<int> s1,s2;
long long sum1=0,sum2=0;
void adjust(){
	while(s1.size()>s2.size()+1){
		multiset<int>::iterator it=(--s1.end());
		sum2+=*it; sum1-=*it;
		s2.insert(*it);
		s1.erase(it);
	}
	while(s1.size()<s2.size()){
		multiset<int>::iterator it=s2.begin();
		sum1+=*it; sum2-=*it;
		s1.insert(*it); s2.erase(it);
	}
}
void add(int tmp){
	if(tmp>*s2.begin()){
		s2.insert(tmp);
		sum2+=tmp;
	}
	else {
		s1.insert(tmp); sum1+=tmp;
	}
	adjust();
}
void del(int tmp){
	multiset<int>::iterator it=s1.lower_bound(tmp);
	if(it!=s1.end()){
		s1.erase(it);
		sum1-=tmp;
	}
	else{
		it=s2.lower_bound(tmp); s2.erase(it);
		sum2-=tmp;
	}
	adjust();
}
void solve(){
	cin>>n>>k;
	s1.clear(); s2.clear(); sum1=sum2=0;
	for(int i=1;i<=n;i++){
		cin>>a[i]; a[i]-=i;
	}
	if(n==1) {
		cout<<1<<endl; return;
	}
	int mid=a[1],sum=0,res=1;
	add(a[1]);
	for(int i=1,j=2;i<=j&&j<=n&&i<=n;i++){
		while(j<=n){
			add(a[j]); int tmp=*(s1.rbegin());
			sum=tmp*s1.size()-sum1+sum2-tmp*s2.size();
			//cout<<tmp<<" ";
			if(sum<=k&&j<=n) {
				res=max(res,j-i+1); j++;
			}
			else {
				del[a[j]];	break;
			}
			
			
		}
		del(a[i]);
	}
	cout<<res<<endl;
}
int main(){
	cin>>t;
	while(t--){
		solve();
	}
} 


 

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

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: 142ms
memory: 5924kb

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:

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

result:

wrong answer 1st lines differ - expected: '1', found: '2'