【枚舉】【二分】Codeforces Round #477 (rated, Div. 2, based on VK Cup 2018 Round 3) D. Resource Distribution

題意:有兩個服務要求被知足,服務S1要求x1數量的資源,S2要求x2數量的資源。有n個服務器來提供資源,第i臺能提供a[i]的資源。當你選擇必定數量的服務器來爲某個服務提供資源後,資源需求會等量地分擔給它們,要求每臺服務器承擔的資源需求不超過其所能提供的資源需求。給定一種合法的方案,每臺服務器要麼沒有被分配給任何一個服務,或者被分配給其中一個服務。數組

對服務器按能提供的資源從小到大排序。枚舉給S1分配的服務器數量i,而後在a數組中二分,就能夠獲得給S1提供的是哪i臺服務器,它們佔據了a數組中連續的一段。服務器

而後給S2哪些呢?分兩種狀況:①假若a數組最後存在k臺能知足S2,而且這k臺不與給S1的相交,那麼不妨就給它這最後的k臺。this

②假若這i臺以後,不存在a的某個後綴可以知足S2,那麼預處理一個東西:pre[i]表明(n-i+1)-need[i](假若要使用第i臺那麼至少要使用的總檯數),而後對這個玩意求個前綴max,若是S1的i臺前存在某個位置j,能使得pre[j]>=i,就是說j所對應的能空出來的臺數大於等於S1佔據掉的臺數,那麼不妨從第n臺,倒着數need[j]臺(跳過S1的i臺),分配給S2便可。spa

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
struct data{
	int val,pos;
	data(){}
	data(const int &val,const int &pos){
		this->val=val;
		this->pos=pos;
	}
};
bool cmp(const data &a,const data &b){
	return a.val!=b.val ? a.val<b.val : a.pos<b.pos;
}
int n,m1,m2;
data a[300005];
int premax[300005],premaxpos[300005],need[300005];
int main(){
	//freopen("d.in","r",stdin);
	scanf("%d%d%d",&n,&m1,&m2);
	for(int i=1;i<=n;++i){
		scanf("%d",&a[i].val);
		a[i].pos=i;
	}
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;++i){
		int tmp;
		if(m2%a[i].val==0){
			tmp=m2/a[i].val;
		}
		else{
			tmp=m2/a[i].val+1;
		}
		if((n-i+1)-tmp>premax[i-1]){
			premax[i]=(n-i+1)-tmp;
			premaxpos[i]=i;
			need[i]=tmp;
		}
		else{
			premax[i]=premax[i-1];
			premaxpos[i]=premaxpos[i-1];
			need[i]=need[i-1];
		}
	}
	int hou=0;
	for(int i=n;i>=1;--i){
		if(m2<=(ll)a[i].val*(ll)(n-i+1)){
			hou=i;
			break;
		}
	}
	if(!hou){
		puts("No");
		return 0;
	}
	for(int i=1;i<=n;++i){
		int tmp;
		if(m1%i==0){
			tmp=m1/i;
		}
		else{
			tmp=m1/i+1;
		}
		data *p=lower_bound(a+1,a+n+1,data(tmp,0),cmp);
		if(p-a+i-1>n){
			continue;
		}
		if(hou>=p-a+i){
			puts("Yes");
			printf("%d %d\n",i,n-hou+1);
			for(int j=p-a;j<p-a+i;++j){
				printf("%d%c",a[j].pos,j==p-a+i-1 ? '\n' : ' ');
			}
			for(int j=hou;j<=n;++j){
				printf("%d%c",a[j].pos,j==n ? '\n' : ' ');
			}
			return 0;
		}
		else if(premax[p-a-1]>=i){
			puts("Yes");
			printf("%d %d\n",i,need[p-a-1]);
			for(int j=p-a;j<p-a+i;++j){
				printf("%d%c",a[j].pos,j==p-a+i-1 ? '\n' : ' ');
			}
			int cnt=0;
			for(int j=n;j>=1;--j){
				if(j>=p-a && j<p-a+i){
					continue;
				}
				++cnt;
				printf("%d%c",a[j].pos,cnt==need[p-a-1] ? '\n' : ' ');
				if(cnt==need[p-a-1]){
					break;
				}
			}
			return 0;
		}
	}
	puts("No");
	return 0;
}
相關文章
相關標籤/搜索