poj1505(二分+贪心)

"最大值尽量小"是一种很常见的优化目标。

关乎于炒书。

题目见此:

http://poj.org/problem?id=1505

我的copy的代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int book[505];
bool flag[505];
int m,k;	//m:the number of books;	k:the number of scribers.
int cnt;	//cnt is short for count. 
int copy(long long x)
{
    cnt=1;
    long long sum=0;
    memset(flag,false,sizeof(flag));
    
    for(int i=m-1;i>=0;i--)
    {
        sum+=book[i];
        if(sum>x)
        {
            cnt++;
            sum=book[i];
            flag[i]=true;
        }
 
    }
    return cnt;
}

void print()
{
    printf("%d",book[0]);
    for(int i=1;i<m;i++)
    {
        if(flag[i-1])   printf(" /");
        printf(" %d",book[i]);
    }
    printf("\n");
}

int main()
{
        int T;
        scanf("%d",&T);
        long long l,r;
        while(T--)
        {
            scanf("%d%d",&m,&k);
            l=r=0;
            for(int i=0;i<m;i++)
            {
                scanf("%d",&book[i]);
				if(book[i]>l) l=book[i];
                r+=book[i];
            }
            
            long long mid;
            while(l<r)
            {
                mid=(l+r)/2;
                if(copy(mid)<=k)    r=mid;
                else l=mid+1;
            }
            int cnt=copy(r);
            for(int i=0;i<m&&cnt<k;i++)
            {
                if(!flag[i])   flag[i]=true,cnt++;
            }
          print(); 
        }
    return 0;
}

  

猜你喜欢

转载自www.cnblogs.com/dragondragon/p/11373604.html