SDUT 3545 装备合成 (模拟) -- 解题报告

题面

装备合成
Time Limit: 1000MS Memory limit: 65536K

题目描述
小白很喜欢玩儿LOL，但是无奈自己是个坑货，特别是在装备的选择和合成上，总是站在泉水里为选装备而浪费时间。现在小白试图解决这个问题，为了使问题简单化，我们把游戏中的装备合成规则简化如下：
（1）装备的合成关系构成一棵合成关系树，如图(a)所示，装备的下级装备叫合成小件，直接连接的叫直接合成小件；上级装备叫合成件，直接连接的叫直接合成件。合成树上的每件装备都可以直接购买，如果钱不够也可以先购买这个装备的合成小件。
（2）每个装备都有一个合成价格和一个总价格，装备的总价格=该装备的合成价格+该装备所有直接合成小件的总价格之和（图(a)关系树上面显示的数字就是对应装备的总价格）。
（3）初始的时候（未拥有装备的任何部分），装备的当前价格等于总价格，当前价格会随着购买的小件而发生相应的变化。
（4）如果购买了某个装备，那么它所有上级可合成装备的当前价格都会相应地减少，减少的量等于这个装备的当前价格。
（5）如果购买了某个装备，那么它下级所有合成小件都会被购买，也就是说，这些小件都变成已拥有的状态，但是物品栏里只有最终合成的装备，因为是小件合成了这件装备。
（6）我们认为关系树上的每个装备的直接合成小件都不超过2件，关系树上同一件物品只会出现一次，我们也不考虑物品的出售问题，我们假设物品栏容量无限，金钱无限。

现在问题来了，按照格式给定一棵合成关系树，再给定一系列操作，完成相应的查询，具体格式见输入。

输入
多组输入，对于每组数据：
第一行有两个整数n m，分别代表关系树中物品的数量和后续的操作数。（1 <= n,m <= 10^4）
接下来的n行，每行对应一件物品的描述，格式为equipment p k，分别代表装备的名称，合成价格，直接合成小件的个数，然后有k个合成小件的名称synthesis1 synthesis2 … synthesisk。（1 <= p <=10^4 , k <= 2 , 所有物品名称只包含小写英文字母，长度不超过20）
接下来的m行，每行对应一个操作，格式如下：
（1）B equipment ，表示物品equipment被购买。
（2）Q equipment ，表示要查询物品equipment的当前价格。
其中equipment代表物品的名称，保证合法且存在于合成关系树中。
注意：如果B指令购买的装备是当前已经拥有的装备（包括小件），那么忽略该条B购买指令。特别地，如果你已经拥有某件装备，那么查询该物品的当前价格时应该输出这件物品的总价格而不是0

输出
对于每次Q查询指令，输出一个整数，代表查询的物品的当前价格。

示例输入
8 10
bannerofcommand 600 2 aegisofthelegion fiendishcodex
aegisofthelegion 400 2 nullmagicmantle crystallinebracer
fiendishcodex 465 1 amplifyingtome
nullmagicmantle 450 0
crystallinebracer 100 2 rubycrystal rejuvenationbead
amplifyingtome 435 0
rubycrystal 400 0
rejuvenationbead 150 0
B crystallinebracer
Q fiendishcodex
Q bannerofcommand
B nullmagicmantle
Q aegisofthelegion
B aegisofthelegion
Q bannerofcommand
Q crystallinebracer
B bannerofcommand
Q bannerofcommand

示例输出
900
2350
400
1500
650
3000

提示

来源
“师创杯”山东理工大学第八届ACM程序设计竞赛
*/

解题思路

没什么可说的，大模拟题一道，对于每一种装备可以用 map 做名称到下标的映射，然后建树，强行模拟。

不多说了，直接上代码：

#include <iostream>#include <cstring>#include <string>#include <map>using namespace std;struct node {    int idx;            //建树之后的下标    int total, rmd;     //总价、剩余价格    int left, right;    //左右子节点的下标    bool non_root;      //是否是根节点} t[10001];map<string, int> mp;int n, m, cnt;bool bought;void Initialize() {    mp.clear();    memset(t, 0, sizeof(t));    cnt = 1;}int GetIndex(string s) {    if(mp[s]) return mp[s];     //如果已有映射，直接返回下标    else return mp[s] = cnt++;  //否则先存储映射再返回下标}void Build(int root) {    if(!t[root].left) { //如果是叶子节点        t[root].idx = cnt++;        t[root].total = t[root].rmd;        return;    }    if(t[root].left) {  //递归建树，回溯时累加装备价值        Build(t[root].left);        t[root].idx = cnt++;        t[root].rmd += t[t[root].left].rmd;     }    if(t[root].right) { //递归建树，回溯时累加装备价值        Build(t[root].right);        t[root].rmd += t[t[root].right].rmd;    }    t[root].total = t[root].rmd;    //更新总价值}int Buy(int root, int idx) {    if(t[root].rmd == 0) return 0;  //如果剩余金钱已为0，则直接返回    int cost;    if(idx == t[root].idx)        cost = t[root].rmd;    if(idx < t[root].idx)        cost = Buy(t[root].left, idx);    if(idx > t[root].idx)        cost = Buy(t[root].right, idx);    t[root].rmd -= cost;    //剩余价格减去此子物品的花费    return cost;}int Query(int root, int idx) {    if(idx == t[root].idx) {        if(t[root].rmd==0 || bought) return t[root].total;  //已合成则返回总价        else return t[root].rmd;    }    if(t[root].rmd == 0) bought = true; //如果中途物品剩余价格为0，则标记为true    if(idx < t[root].idx)        return Query(t[root].left, idx);    else return Query(t[root].right, idx);}int main(int argc, char const *argv[]) {    ios::sync_with_stdio(false);    string name, op;    int cost, num, root;    while(cin >> n >> m) {        Initialize();        for(int i=0; i<n; ++i) {            cin >> name >> cost >> num;            int root_idx = GetIndex(name);            t[root_idx].rmd = cost;            if(num >= 1) {  //初始化左子树                cin >> name;                int sub_idx = GetIndex(name);                t[root_idx].left = sub_idx;                t[sub_idx].non_root = true;            }            if(num == 2) {  //初始化右子树                cin >> name;                int sub_idx = GetIndex(name);                t[root_idx].right = sub_idx;                t[sub_idx].non_root = true;            }        }        //寻找根节点的下标        for(root=1; t[root].non_root; ++root);        cnt = 1;        Build(root);    //建树        while(m--) {            cin >> op >> name;            int idx = t[GetIndex(name)].idx;            if(op == "B") Buy(root, idx);            else {                bought = false;                cout << Query(root, idx) << endl;            }        }    }    return 0;}