Trainticket/seat.cpp at master · Iaven00/Trainticket · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#include "seat.h"

Seat::Seat(int sitenum) {
	this->site.resize(sitenum - 1, 0); //sitenum个站点对应sitenum-1个区间，0代表区间无人
	this->sitenum = 0; //开始时第0个区间为空
	this->restsite = sitenum - 1;
}

Seat::~Seat() {

}

bool Seat::freeSeat(int startsite, int endsite) {
	if (startsite < 0 || endsite > site.size() || endsite <= startsite)
		return false;

	for (int i = startsite; i < endsite; i++) {
		if (site[i] == 1) {
			site[i] = 0;
			this->restsite++;
		}
	}
	if (endsite <= this->sitenum) {
		this->sitenum = startsite;
	}

	return true;
}


//
bool Seat::grabSeat( int startsite, int endsite) {
	if (!judgeSites(startsite, endsite)) return false;
	for (int i = startsite; i < endsite; i++) {
		site[i] = 1;
	}
	this->restsite -= endsite - startsite;
	if (startsite == this->sitenum) {
		for (int i = 0; i < site.size() - endsite; i++) {
			if (site.at(endsite + i) == 0) {
				this->sitenum = endsite + i;
				break;
			}
		}
	}
	return true;
}
void Seat::print() {
	std::cout << "sitenum: " <<(int)this->sitenum<<"   restsites: "<< (int)this->restsite<<std::endl;
	for (int i = 0; i < site.size(); i++)
		std::cout << site[i] << "  ";
	std::cout << std::endl;
}
bool Seat::judgeSites(int startsite, int endsite) {
	if (endsite - startsite > this->restsite || this->sitenum > startsite || startsite < 0 || startsite >= endsite || endsite>site.size()) { //大于剩余空区间
		return false;
	}
	for (int i = startsite; i < endsite; i++) {
		if (site.at(i) != 0) {
			return false;
		}
	}
	return true;
}