【USACO FEB 2010 SILVER】吃巧克力(Chocolate Eating)

【USACO FEB 2010 SILVER】吃巧克力

時間限制: 1 Sec  內存限制: 128 MB

題目描述

    Bessie拿到了N (1 <= N <= 50,000)塊巧克力。她決定想個辦法吃掉這些巧克力，使得它在吃巧克力的這段時間裏，最不開心的一天儘可能的開心。並且一共吃D (1 <= D <= 50,000)天。
     每塊巧克力有一個開心值H_i (1 <= H_i <= 1,000,000)，當某天你吃下那塊巧克力時，你將獲得那塊巧克力的開心值。每一天的開心值是所有當天吃掉的巧克力的總開心值之和。每天晚上Bessie睡覺之後，它的開心值會減半。也就是說，比如昨天Bessie的開心值爲50，那麼今天早上我一醒來我就會有25點的開心值，捨去小數點後數字。另外，Bessie還有一個怪癖，她喜歡按照巧克力本來的排列順序吃。
     Bessie第一天的開心值爲0，求一個每天吃巧克力的方案，使得Bessie最不開心的一天儘可能的開心。

輸入

第1行：2個整數N和D

接下來N行，每行1個整數H_i

輸出

第1行：1個整數，表示最不開心的一天的最大開心值

接下來N行，每行一個整數，表示第i塊巧克力是第幾天吃的

樣例輸入

5 5
10
40
13
22
7

樣例輸出

24
1
1
3
4
5

就是二分

一開始傳參數時mid寫的int類型，LL轉換爲int也沒警告，然後一直亂wa對拍了幾萬組極限數據也沒找到問題。。

以後注意

#include<cstdio>
#include<climits>
#define LL long long
LL cbs[50100];
int tag[50100];
int n,d;
bool tryf(LL mid){
	bool flag=true;
	int at=1;
	LL sum=0;
	for(int i=1;i<=d;i++){
		if(sum)sum=sum>>1;
		while(sum<mid)tag[at]=i,sum+=cbs[at++];
		if(at>n+1){flag=false;break;}
	}
	for(;at<=n;at++)tag[at]=d;
	return flag;
}
int main(){
	
	//freopen("in.txt","r",stdin);
	//freopen("out1.txt","w",stdout);
	LL l=0,r=0,mid;
	scanf("%d%d",&n,&d);
	for(int i=1;i<=n;i++)
		scanf("%lld",&cbs[i]),r+=cbs[i];
	cbs[n+1]=10LL*INT_MAX;
	LL ans;
	while(l<r){
		mid=(l+r+1)>>1;
		if(tryf(mid))ans=mid,l=mid;
		else r=mid-1;
	}
	tryf(ans);
	printf("%lld\n",ans);
	bool f=false;
	for(int i=1;i<=n;i++){
		if(tag[i]==d)f=true;
		if(f)printf("%d\n",d);
		else printf("%d\n",tag[i]);
	}
}