[topCoder-每日一二题]--[5]

比较有意思的一个题目

问题：车上有2列椅子，同一列中相邻椅子的距离为1，两列之间的距离为2，椅子上有一些乘客，现在上来上个人，如何使他们之间的距离和最短？

解决：如果两列得某一列中有连续3个空位，则是最短距离为4。否则，按照2为坐标把空位存储到vector<pair>中，(x,y) 第一列为（0，i）,第二列为(2,i)，然后按照y排序，如果y相同，按照x排序。这样对排序好的数组依次取3个计算距离，取最小值。

程序：

#include <vector>
#include <map>
#include <algorithm>
#include <string>
#include <complex>
#include <iostream>
using namespace std;
struct cmp
{
 bool operator()(pair<int,int>p1,pair<int,int>p2)
 {
  return p1.second==p2.second?(p1.first<p2.first):p1.second<p2.second;
 }
};

class BusSeating
{
 private:
  double deal(pair<int,int>p1,pair<int,int>p2,pair<int,int>p3)
  {
   double d1 = sqrt((double)(p1.first-p2.first)*(p1.first-p2.first)+(p1.second-p2.second)*(p1.second-p2.second));
   double d2 = sqrt((double)(p1.first-p3.first)*(p1.first-p3.first)+(p1.second-p3.second)*(p1.second-p3.second));
   double d3 = sqrt((double)(p3.first-p2.first)*(p3.first-p2.first)+(p3.second-p2.second)*(p3.second-p2.second));
   return d1+d2+d3;
  }
 public:
 double getArrangement(string l, string r)
 {
  string s = "---";
  if(l.find(s)!=string::npos||r.find(s)!=string::npos)
   return 4.0;
  vector<pair<int,int> >px;
  int i = 0;
  while((i=l.find('-',i))!=string::npos)
   px.push_back(make_pair(0,i++));
  i =0;
  while((i=r.find('-',i))!=string::npos)
   px.push_back(make_pair(2,i++));
  sort(px.begin(),px.end(),cmp());
  double min = 100000;
  for(int i = 0; i< px.size()-2;i++)
  {
   double tmp = deal(px[i],px[i+1],px[i+2]);
   cout<<tmp<<endl;
   if(min > tmp+1e-9)
    min = tmp;
  }
  return min;
  
 }
};