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;
}