悬关
查看原帖
悬关
561529
Infinite_Energy楼主2023/5/11 13:11
#include<bits/stdc++.h>
using namespace std;
long long k,n,m,num[10000010],lim,exnum[1000010],ans[1000010],now,tot;
struct node{
	long long val,pos;
}a[1000010],b[1000010];
bool cmp(node x,node y){
	if(x.val!=y.val){
		return x.val<y.val;
	}else{
		return x.pos<y.pos;
	}
}
long long read(){
	char ch=getchar();
	long long sgn=1,x=0;
	while(ch<'0'||ch>'9'){
		if(ch=='-'){
			sgn=-1;
		}
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		x=(x<<3)+(x<<1)+(ch&15);
		ch=getchar();
	}
	return x*sgn;
}
void write(long long n,bool p){
	if(n<0){
		putchar('-');
		n=-n;
	}
	if(n==0){
		if(p==true){
			putchar('0');
		}
		return;
	}
	write(n/10,0);
	putchar(n%10+'0');
}
int main(){
	n=read();
	m=read();
	k=read();
	for(int i=1;i<=n;i++){
		a[i].val=read();
		a[i].pos=i;
		lim=max(lim,a[i].val);
		num[a[i].val]++;
	}
	for(int i=1;i<=m;i++){
		b[i].val=read();
		b[i].pos=i;
		lim=max(lim,b[i].val);
	}
	sort(b+1,b+m+1,cmp);
	for(int i=lim;i>=0;i--){
		if(i==0){
			if(num[i]>k){
				cout<<-1<<endl;
				return 0;
			}
		}
		if(num[i]<=k){
			exnum[i]=k-num[i];
		}else{
			num[i-1]+=num[i]-k;
		}
	}
	now=1;
	for(int i=0;i<=lim;i++){
		if(exnum[i]>0){
			while(b[now].val<=i-1&&now<=m){
				now++;
			}
			while(exnum[i]>=1&&now<=m){
				tot++;
				ans[tot]=b[now].pos;
				exnum[i]--;
				now++;
			}
		}
	}
	write(tot,1);
	putchar('\n');
	sort(ans+1,ans+tot+1);
	for(int i=1;i<=tot;i++){
		write(ans[i],1);
		putchar(' ');
	}
	return 0;
}

在机房一个小时没调出来。服了。WA8。

2023/5/11 13:11
加载中...