北京大学MOOC 程序设计与算法（三）魔兽世界之一：备战

魔兽世界之一：备战

题目来源：

北京大学MOOC 程序设计与算法（三）第03周测验(2020春季)

题目描述：

005:魔兽世界之一：备战

总时间限制: 1000ms 内存限制: 65536kB

魔兽世界的西面是红魔军的司令部，东面是蓝魔军的司令部。两个司令部之间是依次排列的若干城市。
红司令部，City 1，City 2，……，City n，蓝司令部

两军的司令部都会制造武士。武士一共有 dragon 、ninja、iceman、lion、wolf 五种。每种武士都有编号、生命值、攻击力这三种属性。

双方的武士编号都是从1开始计算。红方制造出来的第n个武士，编号就是n。同样，蓝方制造出来的第n个武士，编号也是n。

武士在刚降生的时候有一个生命值。

在每个整点，双方的司令部中各有一个武士降生。

红方司令部按照iceman、lion、wolf、ninja、dragon的顺序循环制造武士。

蓝方司令部按照lion、dragon、ninja、iceman、wolf的顺序循环制造武士。

制造武士需要生命元。

制造一个初始生命值为m的武士，司令部中的生命元就要减少m个。

如果司令部中的生命元不足以制造某个按顺序应该制造的武士，那么司令部就试图制造下一个。如果所有武士都不能制造了，则司令部停止制造武士。

给定一个时间，和双方司令部的初始生命元数目，要求你将从0点0分开始到双方司令部停止制造武士为止的所有事件按顺序输出。
一共有两种事件，其对应的输出样例如下：

1. 武士降生
   输出样例： 004 blue lion 5 born with strength 5,2 lion in red headquarter
   表示在4点整，编号为5的蓝魔lion武士降生，它降生时生命值为5，降生后蓝魔司令部里共有2个lion武士。（为简单起见，不考虑单词的复数形式）注意，每制造出一个新的武士，都要输出此时司令部里共有多少个该种武士。

2. 司令部停止制造武士
   输出样例： 010 red headquarter stops making warriors
   表示在10点整，红方司令部停止制造武士

输出事件时：

首先按时间顺序输出；

同一时间发生的事件，先输出红司令部的，再输出蓝司令部的。

输入
第一行是一个整数，代表测试数据组数。

每组测试数据共两行。

第一行：一个整数M。其含义为， 每个司令部一开始都有M个生命元( 1 <= M <= 10000)。

第二行：五个整数，依次是 dragon 、ninja、iceman、lion、wolf 的初始生命值。它们都大于0小于等于10000。
输出
对每组测试数据，要求输出从0时0分开始，到双方司令部都停止制造武士为止的所有事件。
对每组测试数据，首先输出"Case:n" n是测试数据的编号，从1开始 。
接下来按恰当的顺序和格式输出所有事件。每个事件都以事件发生的时间开头，时间以小时为单位，有三位。

样例输入

1
20
3 4 5 6 7

样例输出

Case:1
000 red iceman 1 born with strength 5,1 iceman in red headquarter
000 blue lion 1 born with strength 6,1 lion in blue headquarter
001 red lion 2 born with strength 6,1 lion in red headquarter
001 blue dragon 2 born with strength 3,1 dragon in blue headquarter
002 red wolf 3 born with strength 7,1 wolf in red headquarter
002 blue ninja 3 born with strength 4,1 ninja in blue headquarter
003 red headquarter stops making warriors
003 blue iceman 4 born with strength 5,1 iceman in blue headquarter
004 blue headquarter stops making warriors

题目分析

基于题意，首先设计一个士兵（Soldier）的类，包含名字（name）和生命值（hp）属性。由于发现这道题作为系列之一比较简单，各种士兵之间的区别仅仅是name和hp不同，所以暂时没有必要为各种士兵分别设计派生类。

还需要设计指挥部（Headquarter）类，基本的属性有：名字（name），生命元（power）。此题中只需打印士兵降生的信息，因此无需将士兵队列作为成员变量放进指挥部类。

关于士兵的降生，题目的要求是红蓝双方分别按各自的顺序制造，如果不足则试图制造下一个，直到所有的均不能制造，这也是本题的重点所在。我的解决方案是设计一个全局的士兵池soldierPool：

Soldier soldierPool[5] = {Soldier("dragon"), Soldier("ninja"),
                          Soldier("iceman"), Soldier("lion"), Soldier("wolf")};

然后在Headquarter类中设置一个seq数组记录指挥部的出兵顺序以及一个pos标记表示当前在seq数组中的位置。最后还需要一个totalCount变量记录指挥部制造士兵的编号以及一个count数组记录每种士兵的数量。

main函数的设计就很简单了，数据读入，类初始化，然后进入while循环让双方轮流制造士兵。注意当一方不能制造士兵后只需要打印一次停止信息，这里需要用一个flag标记一下。具体可以参考下面的代码。

完整代码

#include <iostream>
#include <cstring>
#include <iomanip>

using namespace std;

class Soldier {
private:
    char _name[20]{};
    int _hp;

    friend class HeadQuarter;

public:
    explicit Soldier(const char *name, int hp = 0) {
        strcpy(_name, name);
        _hp = hp;
    }

    void SetHp(int hp) {
        _hp = hp;
    }

};

Soldier soldierPool[5] = {Soldier("dragon"), Soldier("ninja"),
                          Soldier("iceman"), Soldier("lion"), Soldier("wolf")};

class HeadQuarter {
private:
    char _name[20]{};
    int _power;
    int _seq[5]{};
    int _totalCount;
    int _count[5]{};
    int _pos;
public:
    HeadQuarter(const char *name, int power, const int *seq) {
        _totalCount = 0;
        _pos = 0;
        for (int i = 0; i < 5; ++i) _count[i] = 0;
        strcpy(_name, name);
        _power = power;
        for (int i = 0; i < 5; i++) {
            _seq[i] = seq[i];
        }
    }

    bool CanBornSoldier() {
        for (int i = 0; i < 5; ++i) {
            if (_power >= soldierPool[i]._hp)
                return true;
        }
        return false;
    }

    int BornSoldier() {
        if (_pos >= 5) _pos -= 5;
        for (int i = 0; i < 5; ++i) {
            if (_power >= soldierPool[_seq[_pos]]._hp) {
                _power -= soldierPool[_seq[_pos]]._hp;
                ++_count[_seq[_pos]];
                ++_totalCount;
                int tempPos = _pos;
                ++_pos;
                return _seq[tempPos];
            } else {
                ++_pos;
                if (_pos >= 5) _pos -= 5;
            }
        }
        return -1;
    }

    void PrintBornInfo(int pos, int time) {
        if (pos >= 0) {
            cout << setfill('0') << setw(3) << right << time << " " << _name << " "
                 << soldierPool[pos]._name << " " << _totalCount << " born with strength "
                 << soldierPool[pos]._hp << "," << _count[pos] << " " << soldierPool[pos]._name << " in "
                 << _name << " headquarter" << endl;
        }
    }

    void PrintStopInfo(int time) {
        cout << setfill('0') << setw(3) << right << time << " " << _name << " headquarter stops making warriors" << endl;
    }
};

int main() {
    int caseTotal;
    cin >> caseTotal;
    int caseNum = 0;
    while (caseNum < caseTotal) {
        ++caseNum;
        int time = 0;
        int power;
        cin >> power;
        int redSeq[5] = {2, 3, 4, 1, 0}, blueSeq[5] = {3, 0, 1, 2, 4};
        HeadQuarter red("red", power, redSeq);
        HeadQuarter blue("blue", power, blueSeq);
        int tempHp;
        for (int i = 0; i < 5; ++i) {
            cin >> tempHp;
            soldierPool[i].SetHp(tempHp);
        }
        cout << "Case:" << caseNum << endl;
        bool flag1 = true, flag2 = true;
        while (true) {
            if (red.CanBornSoldier()) {
                int index = red.BornSoldier();
                red.PrintBornInfo(index, time);
            } else {
                if (flag1)
                    red.PrintStopInfo(time);
                flag1 = false;
            }
            if (blue.CanBornSoldier()) {
                int index = blue.BornSoldier();
                blue.PrintBornInfo(index, time);
            } else {
                if (flag2)
                    blue.PrintStopInfo(time);
                flag2 = false;
            }
            if (!red.CanBornSoldier() && !blue.CanBornSoldier() && !flag1 && !flag2) break;
            ++time;
        }
    }
    return 0;
}