链接:点击打开链接
题意:在一个长L的数轴上,起点在0点,有可能有两种事件,第一种是数轴上某一点会出现一份食物,另一种是找食物,这种情况会前往一个最近的食物,如果两边的食物距离相同,那么就按最后一次移动的方向进行移动,如果两边都没有食物,就呆在原地,问最后走的路程是多少
代码:
#include <set>
#include <queue>
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
multiset<int> s;
multiset<int>::iterator ite,ite_tmp;
int main(){
int a,b,i,l,n,t,cas,dis,pos;
long long ans;
scanf("%d",&t);
for(cas=1;cas<=t;cas++){
s.clear();
scanf("%d%d",&l,&n);
ans=pos=0,dis=1;
for(i=1;i<=n;i++){
scanf("%d",&a);
if(a==0){
scanf("%d",&b);
s.insert(b);
}
else{ //记录当前位置和方向
if(s.size()==0) //二分找出离当前方向最近的两个点
continue;
if(s.size()==1){
ite=s.begin();
ans+=abs(*ite-pos);
if(*ite>pos)
dis=1;
else if(*ite<pos)
dis=0;
pos=*ite;
s.erase(ite);
continue;
}
ite=s.lower_bound(pos);
if(*ite==pos){
s.erase(ite);
continue;
}
if(ite==s.end()){ //注意不能直接删除s.end()
ite--;
ans+=abs(*ite-pos);
dis=0;
pos=*ite;
s.erase(ite);
continue;
}
if(ite==s.begin()){
ans+=abs(*ite-pos);
dis=1;
pos=*ite;
s.erase(ite);
continue;
}
ite_tmp=--ite;
ite=s.upper_bound(pos); //找出比当前点大的那个点
if(abs(*ite-pos)<abs(*ite_tmp-pos)){
ans+=abs(*ite-pos);
dis=1;
pos=*ite;
s.erase(ite);
}
else if(abs(*ite-pos)>abs(*ite_tmp-pos)){
ans+=abs(*ite_tmp-pos);
dis=0;
pos=*ite_tmp;
s.erase(ite_tmp);
}
else if(abs(*ite-pos)==abs(*ite_tmp-pos)){
ans+=abs(*ite_tmp-pos); //距离相同选方向和上一回一样的点
if(dis==1){
pos=*ite;
s.erase(ite);
}
else{
pos=*ite_tmp;
s.erase(ite_tmp);
}
}
}
}
printf("Case %d: %I64d\n",cas,ans);
}
return 0;
}