E 求调

这个 E 我写了个很简短的做法,但是不知道为什么过不去,换了一种写法才过的。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
#include<cmath>
#include<string>
#include<set>
#include<map>
#include<queue>
#include<assert.h>
#define int long long
#define pii pair<int,int>
#define x first
#define y second
using namespace std;
const int N = 5e5 + 7;
int n,m;
pii s[N];
int a[N];
int ans;
void solve(){
	ans = 0;
	cin>>n>>m;
    for(int i=1;i<=n;++i) cin>>s[i].y;
    for(int i=1;i<=n;++i) cin>>s[i].x;
    sort(s+1,s+n+1);
    vector<int>vec;
    for(int i=1;i<=n;++i){
        pii tmp = make_pair(s[i].x+s[i].y+1,0);
        a[i] = upper_bound(s+i,s+n+1,tmp)-s-1;
    }
    for(int i=n;i>=1;--i){
    	int j = i - 1;
    	while(j>=1 && a[j]>j) j--;
    	j++;
    	vec.push_back(i-j+1);
    	i = j;
    }
    sort(vec.begin(),vec.end());
    reverse(vec.begin(),vec.end());
    for(int i=0;i<m && i<(int)vec.size();++i) ans += vec[i];
    cout<<ans<<'\n';
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int T;cin>>T;
	for(int _=1;_<=T;++_){
		solve();
	}
	return 0;
}
全部评论

相关推荐

昨天 11:12
重庆大学 C++
既然这么缺人,为什么挂我呢
希望被offer砸中...:其实不缺人
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务