背景
航空公司的货运飞机往返于各个国家和货物分理中心之间,分理中心的货物运送到各个国家,再将各个国家发出的货物送到分理中心。
分理中心非常庞大,每个国家都有一个自己的货运站,每个货运站有两个平台用于堆放货物。其中 A 平台用于堆放运送到该国的货物,而 B 平台堆放着其它国家的货物。各个国家的货运站排成一个环形。
平时,有货运汽车沿着环形的线路在各个国家的货运站之间运行。货运汽车的仓库是非常狭窄的,先放进去的货物只能等后放进去的货物移走之后才能搬出来。
当货运汽车到达一个站点 X 时,它首先卸货。卸货的过程是:首先查看最外面的箱子是否标记着 X 标签,如果是,代表这是运送到 X 国的货物,则放到 A 平台上,如果不是,则把该货箱放到B平台货物队列的末尾,然后处理下一箱货。当 B 平台放满或者是车被搬空,则开始装货。装货的过程是:从 B 平台货物队列的开始处装车,把车装满或把 B 平台搬空则装货结束。车子开往下一个货运站。
卸下一箱货物和装载一箱货物均需要一分钟的时间,且假设当货车从一个货运站开往另一个货运站的时间为两分钟。
每天,货车和货运飞机都这样周而复始的运行着,直到有一天因为报酬的原因,分理中心的工人开始罢工。货运飞机依然运行,但却无法运走任何货物。
任务
你是老板的心腹,平时深受老板关照。因此,你决定停止罢工,独自担当起分理的任务。现在,分理中心已经堆积了大量的货物,你决定在开始前先写程序计算一下要完成搬运需要多长的时间。
输入
对于输入部分包含几组数据,第一行是一个整数SET,表明下面有几组数据,随后便是这几组输入数据。
对于每一组输入数据,其第一行为三个整数 N、S 和 Q。其中 N(2<=N<=100) 表示环中站的个数。S(1<=S<=100) 表示货车的最大容量,即货车同时可以装载多少箱货物。而 Q(1<=Q<=100) 表示 B 平台所能放置的货物的总数,假定该系统中所有货运站 B 平台的最大容量都相同。
从第二行往后一共有 N 行,每行一的第一个数字代表该站 B 平台上货物的总数,之后的每个数字代表每箱货物运送到哪个站。
输出
对于每组输入,你需要在独立的一行中输出完成该任务所需要的时间。
C++代码(含注释)
#include<iostream>
#include<vector>
#include<stack>
#include<queue>
using namespace std;
int main() {
int N; // 测试用例的数量
cin >> N;
while (N--) {
int numStations, maxCapacity, maxNumBoxes;
cin >> numStations >> maxCapacity >> maxNumBoxes;
stack<int> truck; // 货车是栈
vector<queue<int>> stations(numStations); // 每站的B平台是一个队列
int totalBoxes = 0; // 总共的货物数量
for (auto& stationQueue : stations) {
int numBoxes, destination;
cin >> numBoxes;
totalBoxes += numBoxes;
while (numBoxes--) {
cin >> destination;
stationQueue.push(destination);
}
} // 对N行输入进行录入
int currentStation = 0; // 当前所在的站点,范围是 [0, numStations)
int time = 0; // 时间
while (totalBoxes) {
// 装货
while (truck.size() < maxCapacity && !stations[currentStation].empty()) {
truck.push(stations[currentStation].front()); // 将B平台上的货物放入货车
stations[currentStation].pop(); // 从B平台上移除货物
time++; // 时间增加
}
// 运输
currentStation = (currentStation + 1) % numStations; // 当前站点移动到下一个站点,循环范围是 [0, numStations)
time += 2; // 运输时间为2分钟
// 卸货
while (!truck.empty() && (stations[currentStation].size() < maxNumBoxes || currentStation + 1 == truck.top())) {
if (currentStation + 1 == truck.top()) { // 如果当前站点就是该货物的目的地
truck.pop(); // 货车卸货
totalBoxes--; // 更新总货物数量
time++; // 时间增加
}
else {
stations[currentStation].push(truck.top()); // 将货物放回当前站点的B平台
truck.pop(); // 货车卸货
time++; // 时间增加
}
}
}
cout << time << endl; // 输出所需的总时间
}
return 0;
}