HE CSP-S 2025游记

#include<bits/stdc++.h>
#define int long long
using namespace std;
int T,n;
bool k;

struct p{
  int a,b;
}ob[100005],mid;
bool cmp(p a,p b){
  if(a.a>b.a) return 1;
  if(a.a==b.a){
  	return a.b<b.b;
  }
  return 0;
}

priority_queue<p,vector<p>,greater<p> > q;
bool operator > (const p &a,const p &b){
  return a.b>b.b;
}




signed main(){
  freopen("adventure.in","r",stdin);
  freopen("adventure.out","w",stdout);
  cin>>T;
  while(T--){
  	int ans=0;
  	cin>>n>>k;
  	for(int i=1;i<=n;i++){
  		cin>>ob[i].a;
  	}
  	for(int i=1;i<=n;i++){
  		cin>>ob[i].b;
  	}
  	sort(ob+1,ob+1+n,cmp);
  	if(k==1){
  		mid.a=ob[1].a;
  		mid.b=ob[1].b;
  		q.push(mid);
  		ans+=mid.b;
  		for(int i=1;i<=n;i++){
  			ob[i].a=ob[i+1].a;
  			ob[i].a=ob[i+1].b;
  		}
  	}
  	for(int i=1;i<=n;i++){
  		mid.a=ob[i].a;
  		mid.b=ob[i].b;
  		if(i%2==1){
  			if(q.empty()) continue;
  			if(q.top().b<mid.b){
  				ans-=(q.top().b-mid.b);
  				q.pop();
  				q.push(mid);
  			}
  		}else{
  			q.push(mid);
  			ans+=mid.b;
  		}
  	}
  	while(!q.empty()){
  		q.pop();
  	}
  	cout<<ans<<endl;
  }
  fclose(stdin);
  fclose(stdout);
}