HDU 5303 多校第二场

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

题目大意：有一个长为L的环，n个苹果树，一个篮子最多装k个苹果，装完要回到起点卸下再出发，给出n个苹果树顺时针的位置及苹果的个数，求摘完所有苹果走的最小路程。

思路：总体是贪心，把环变成直线，分左右两条线，还有一种情况就是最后剩下小于k个的苹果绕环一圈拿完，对于每条线，

可以dp求得距离，有一个窍门，因为总的苹果数是小于1e5的，所以可以对于每个苹果都看成一个点。dis[i]表示取完第i个苹果所走的距离，有i<=k时，dis[i]=x[i];i>k时，dis[i]=dis[i-k]+x[i];...

#include <iostream>
#include <string.h>
#include <algorithm>
#include <vector>
#include <cstdio>
using namespace std;

#define REP(i,a,b) for(int i=a;i<b;++i)
#define mset(a,b) memset(a,b,sizeof a)

typedef long long LL;

const int maxn = 101111;

LL x[maxn];
vector <LL> l,r;
LL ldis[maxn],rdis[maxn];
int a,n,L,k;

void init()
{
l.clear();
r.clear();
mset(ldis,0);
mset(rdis,0);
}

int main()
{
int t;
scanf("%d",&t);
while(t--)
{
init();
//cin>>L>>n>>k;
scanf("%I64d%d%d",&L,&n,&k);
int cnt=0;
REP(i,0,n){
LL pos;
//cin>>pos>>a;
scanf("%d%d",&pos,&a);
REP(i,0,a){
x[cnt++]=pos;
}
}
//sort(x,x+cnt);
REP(i,0,cnt){
if(x[i]*2<=L) l.push_back(x[i]);
else{
r.push_back(L-x[i]);
}
}
sort(l.begin(),l.end());
sort(r.begin(),r.end());
int lsz=l.size();
int rsz=r.size();
for(int i=0;i<lsz;++i){
int j=i+1;
if(j<=k) ldis[j]=l[i];
else{
ldis[j]=ldis[j-k]+l[i];
}
}
for(int i=0;i<rsz;++i){
int j=i+1;
if(j<=k) rdis[j]=r[i];
else{
rdis[j]=rdis[j-k]+r[i];
}
}
LL ans=((ldis[lsz]+rdis[rsz])<<1);
for(int i=1;i<=lsz && i<k;++i){
int rpo = max(0,rsz-k+i);
LL dis = ((ldis[lsz-i]+rdis[rpo])<<1)+L;
ans=min(ans,dis);
}
printf("%I64d\n",ans);
}
}