thoughtworks笔试作业

#include "Badminton.h" #include "StringTool.h" #include <iostream> using namespace std; bool BookItem::SetBookStr(string s) { vector<string> subs; split(s, " ", subs); if (SetClientID(subs[0]) && SetDateAndDay(subs[1]) && SetTime(subs[2], this->GetBookday())) { if ((subs.size() == 4 && SetBookType("Book")) || SetBookType("Cancel")) { this->bookstr = s; return true; } } return false; } bool BookItem::SetClientID(string s) { string prttn = "[[:alnum:]]+"; regex r(prttn); if (regex_match(s, r)) { this->clientID = s; return true; } return false; } bool BookItem::SetDateAndDay(string s) { string prttn = "(([0-9]{3}[1-9]|[0-9]{2}[1-9][0-9]{1}|[0-9]{1}[1-9][0-9]{2}|[1-9][0-9]{3})-(((0[13578]|1[02])-(0[1-9]|[12][0-9]|3[01]))|((0[469]|11)-(0[1-9]|[12][0-9]|30))|(02-(0[1-9]|[1][0-9]|2[0-8]))))|((([0-9]{2})(0[48]|[2468][048]|[13579][26])|((0[48]|[2468][048]|[3579][26])00))-02-29)"; regex r(prttn); if (regex_match(s, r)) { this->date = s; this->day = this->CaculateWeekDay(stoi(s.substr(0, 4)), stoi(s.substr(5, 2)), stoi(s.substr(8, 2))); return true; } return false; } bool BookItem::SetTime(string s,WEEKDAY day) { string prttn = "(0\\d{1}|1\\d{1}|2[0-3]):[0-5]\\d{1}~(0\\d{1}|1\\d{1}|2[0-3]):[0-5]\\d{1}"; regex r(prttn); if (regex_match(s, r)) { TimeSpan tmp = TimeSpan(s); switch (day) { case sat: case sun: if (tmp.st_h < 9 || tmp.st_h >= 22 || tmp.ed_h <= 9 || tmp.ed_h >22 || tmp.st_h >= tmp.ed_h || tmp.ed_m != 0 || tmp.st_m != 0) return false; this->timespan = tmp; this->time = s; return true; case mon: case tue: case wed: case thu: case fri: if (tmp.st_h < 9 || tmp.st_h >= 22 || tmp.ed_h <= 9 || tmp.ed_h >22 || tmp.st_h >= tmp.ed_h || tmp.ed_m != 0 || tmp.st_m != 0) return false; this->timespan = tmp; this->time = s; return true; } } return false; } bool BookItem::SetBookType(string s) { this->bookOrcancel = s; return true; } void BookItem::UpdateIncome(){ int st_h = this->GetBookStime().first; int ed_h = this->GetBookEtime().first; switch (this->GetBookday()) { case sat: case sun: if (ed_h <= 12) this->income = (ed_h - st_h) * 40; else if (st_h <12 && ed_h <= 18) this->income = (12 - st_h) * 40 + (ed_h - 12) * 50; else if (st_h <12 && ed_h >18) this->income = (12 - st_h) * 40 + 300 + (ed_h - 18) * 60; else if (st_h <18 && ed_h <= 18) this->income = (ed_h - st_h) * 50; else if (st_h <18 && ed_h >18) this->income = (18 - st_h) * 50 + (ed_h - 18) * 60; else this->income = (ed_h - st_h) * 60; break; case mon: case tue: case wed: case thu: case fri: if (ed_h <= 12) this->income = (ed_h - st_h) * 30; else if (st_h <12 && ed_h <= 18) this->income = (12 - st_h) * 30 + (ed_h - 12) * 50; else if (st_h <12 && ed_h <= 20) this->income = (12 - st_h) * 30 + 300 + (ed_h - 18) * 80; else if (st_h <12 && ed_h <= 22) this->income = (12 - st_h) * 30 + 460 + (ed_h - 20) * 60; else if (st_h <18 && ed_h <= 18) this->income = (ed_h - st_h) * 50; else if (st_h < 18 && ed_h <= 20) this->income = (18 - st_h) * 50 + (ed_h - 18) * 80; else if (st_h < 18 && ed_h <= 22) this->income = (18 - st_h) * 50 + 160 + (22 - ed_h) * 80; else if (st_h < 20 && ed_h <= 20) this->income = (ed_h - st_h) * 80; else if (st_h < 20 && ed_h <= 22) this->income = (20 - st_h) * 80 + (ed_h - 20) * 60; else this->income = (ed_h - st_h) * 60; break; } if (this->GetBookType() == "Cancel") { if(this->GetBookday()==WEEKDAY::sat || this->GetBookday()==WEEKDAY::sun) this->income /= 4; else this->income /= 2; } } WEEKDAY BookItem::CaculateWeekDay(int y, int m, int d) { if (m == 1 || m == 2) { m += 12; y--; } int week = (d + 2 * m + 3 * (m + 1) / 5 + y + y / 4 - y / 100 + y / 400) % 7; WEEKDAY weekstr; switch (week) { case 0: weekstr = mon; break; case 1: weekstr = tue; break; case 2: weekstr = wed; break; case 3: weekstr = thu; break; case 4: weekstr = fri; break; case 5: weekstr = sat; break; case 6: weekstr = sun; break; } return weekstr; } bool BadmintonCourt::SetBookInfo(string bookinfo) { BookItem bkitm; if (bkitm.SetBookStr(bookinfo)) { //检查订单是否符合条件 if (bkitm.GetBookType() == "Book") { bool again = false; for (auto &itm : (this->bookinfo)[bkitm.GetBookDate()]) { if (itm.GetBookType() == "Book") { if (!(bkitm.GetBookEtime().first <= itm.GetBookStime().first\ || bkitm.GetBookStime().first >= itm.GetBookEtime().first)) { cout << "Error: the booking conflicts with existing bookings!" << endl; return false; } else continue; } else continue; } this->bookinfo[bkitm.GetBookDate()].push_back(bkitm); return true; } else { for (auto& itm : this->bookinfo[bkitm.GetBookDate()]) { if (itm.GetBookType() == "Book") { if (bkitm.GetclientID() == itm.GetclientID() && bkitm.GetBookTime() == itm.GetBookTime()) { itm.SetBookType("Cancel"); return true; } else continue; } } cout << "Error: the booking being cancelled does not exist!" << endl; return false; } } cout << "Error: the booking is invalid!" << endl; return false; } void BadmintonCourt::PrintIncome() { for (auto &item : bookinfo) { for (auto &item1 : item.second) { if (item1.GetBookType() == "Book") cout << item1.GetBookDate() << " " << item1.GetBookTime() << " " << item1.GetIncome() << "元" << endl; else cout << item1.GetBookDate() << " " << item1.GetBookTime() << " 违约金 " << item1.GetIncome() << "元" << endl; } } } void BadmintonCourt::UpdateIncome() { income = 0; for (auto &item : bookinfo) { for (auto &item1 : item.second) { item1.UpdateIncome(); if (item1.GetBookType() == "Book") { double tmp = item1.GetIncome(); income += tmp; } else { double tmp = item1.GetIncome(); income += tmp; } } } }