题目描述
XCQ队长要退役啦,由于队长常年刷题,机位上摆着各类算法书,一个人实在是搬不动,所以他叫来了他的小弟ZZX帮忙搬书。队长突然灵机一动,给ZZX出了一个问题。
假设机位上有n本书,每本书的体积分别为ai,那么如果使用容量为V的书包来装书最少要几个书包呢?(注意书是不可拆分的,整体要么装到一个书包,要么不装)。
ZZX(冥思苦想): 不会呀。
XCQ: 看你这么弱,我就把问题再简化一下好啦,让每个书包只装不超过两本书, 那么用k(2 * k >= n)个体积为V的书包能否装完n本书呢?
ZZX(又是一阵冥思苦想): 队长放过我吧,实在是不会啊。
XCQ: 长江后浪推前浪, 既然如此,就把这个问题交给后浪们吧。
输入
第一行包含一个数字T(T <= 60), 表示测试数据组数。然后一空行,随后有T组数据。
每组测数据格式如下:
第一行包含三个数字n,k,V(1 <= n <= 1000000, n <= 2 * k <= 1000000, 1 <= V <= 1000000),表示有n本书,使用k个容量为V的书包。
第二行n个数,表示每本书的体积ai(ai <= 1000000)。所有数据保证给出的n个数是非递减的。
每组数据后面有一空行。
输出
每组数据输出一行, 如果能够装完所有书本输出"Yes",否则输出"No",引号不需要输出,注意字母大小写。
样例输入
3
4 5 8
1 5 6 8
3 2 7
3 5 7
10 5 108
15 15 20 28 38 44 46 52 69 94
样例输出
Yes
No
No
题目大意:n本书,k个大小为v的书包,问k个书包能否装满n本书,条件是每个书包最多放两本书,而且书的大小是非降序
解题思路:题目条件约束较多,所以题目比较简单,直接最大的书和最小的书放一起就行,如果放不下就单独放最大的书,
也就是简单的贪心算法,但不是最优解。
我前年(2017)写这个题时,有一个困扰
比如1,2,9,10四本书,书包容量为v=13,我们可以2和10放一起,1和9放一起,这样是或许是最优解(这就是当时我的思路),但是:2和10放一起的话,那么2和9,1和10也绝对可以放一起,没必要把一个书包凑得那么满,
也就是说:w<x<y<z
if a[x]+a[z]<=v 则 a[x]+a[y]<=v && a[w]+a[z]<=v;
所以我们直接最大的和最小的凑一起就行了,没必要把一个书包凑得很满
代码中有注释:
#include <iostream>
#include <cstdio>
using namespace std;
int a[1000000];
int main()
{
int t,n,k,v;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&k,&v);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
int i,j,flag=1;
i=0,j=n-1;//两个指针
while(k!=0 && i<=j)//i>j 说明书都放完了,反之没放完
{
if(a[i]+a[j]<=v)//第一本和最后一本放一个包里
{
k--;
i++;
j--;
}
else if(a[j]<=v)//最后一本单独放一个包里
{
k--;
j--;
}
else//书大于v了,放不下
{
flag=0;
break;
}
}
if(i<=j) flag=0;//书没放完
printf(flag?"Yes\n":"No\n");
}
return 0;
}