QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#694079#7904. Rainbow SubarrayczrqWA 147ms8064kbC++171.5kb2024-10-31 17:14:022024-10-31 17:14:04

Judging History

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

  • [2024-10-31 17:14:04]
  • 评测
  • 测评结果:WA
  • 用时:147ms
  • 内存:8064kb
  • [2024-10-31 17:14:02]
  • 提交

answer

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


 

詳細信息

Test #1:

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

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: 147ms
memory: 8064kb

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

result:

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