HDU 5303 Delicious Apples (类似背包)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5303

题意：有一个长为len(0~len-1)的环，环上一些位置上有苹果树，每棵数上有若干个苹果，有一个大小为k的箩筐，当箩筐满了之后可以回到出发点0处把苹果卸下，问最少走多少距离才可以把所有苹果运回0点呢？

思路：由于是一个环，我们将其分成左右2边处理出取第i个苹果时的最短距离，由于苹果数量小于10^5，将苹果离散化，分别存入l和r这2个数组排序，取该苹果的最短距离disl[i]=dis[i-k]+l[i](右边同理)，res=(disl[l.size()]+disr[r.size()])*2

当前没有取的苹果个数小于等于k的时候可能会发生要绕一圈取完所有苹果的情况，因此我们枚举这一种情况res=min(res,(disl[l.size()-i]+disr[r.size()-(k-i)])*2+len)，注意r.size()-(k-i)可能小于0

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#define LL long long
using namespace std;

LL pos[100030],disl[100030],disr[100030];
vector <LL> l,r;

int main()
{
int t,len,n,k;
scanf("%d",&t);
while (t--)
{
int num=0;
l.clear();
r.clear();
memset(disl,0,sizeof(disl));
memset(disr,0,sizeof(disr));
scanf("%d%d%d",&len,&n,&k);
for (int i=0;i<n;i++)
{
int tmp1,tmp2;
scanf("%d%d",&tmp1,&tmp2);
while (tmp2--)
{
pos[num++]=tmp1;
}
}

for (int i=0;i<num;i++)
{
if (2*pos[i]<len) l.push_back(pos[i]);
else r.push_back(len-pos[i]);
}
sort(l.begin(),l.end());
sort(r.begin(),r.end());

for (int i=0;i<l.size();i++)
{
if (i+1<=k)
disl[i+1]=l[i];
else
disl[i+1]=disl[i+1-k]+l[i];
}

for (int i=0;i<r.size();i++)
{
if (i+1<=k)
disr[i+1]=r[i];
else
disr[i+1]=disr[i+1-k]+r[i];
}

LL res=(disl[l.size()]+disr[r.size()])*2;
//cout<<disl[l.size()]<<":"<<disr[r.size()]<<endl;
for (int i=0;i<=k && i<=l.size();i++)
{
int tmp1=l.size()-i;
int tmp2=r.size()-(k-i);
if (tmp2<0) tmp2=0;
res=min(res,(disl[tmp1]+disr[tmp2])*2+len);
}
printf("%I64d\n",res);
}
}