1985 Problem A 任务调度

问题 A: 任务调度
时间限制: 1 Sec 内存限制: 32 MB
献花: 11 解决: 9
[献花][花圈][TK题库]
题目描述
读入任务调度序列，输出n个任务适合的一种调度方式。
输入
输入包含多组测试数据。
每组第一行输入一个整数n（n<100000），表示有n个任务。
接下来n行，每行第一个表示前序任务，括号中的任务为若干个后序任务，表示只有在前序任务完成的情况下，后序任务才能开始。若后序为NULL则表示无后继任务。
输出
输出调度方式，输出如果有多种适合的调度方式，请输出字典序最小的一种。
样例输入
4
Task0(Task1,Task2)
Task1(Task3)
Task2(NULL)
Task3(NULL)
样例输出
Task0 Task1 Task2 Task3

#define _CRT_SECURE_NO_WARNINGS
#include <unordered_map>
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <map>

using namespace std;

struct dispatch
{
    string taskname;
    int level;

    friend bool operator < (const dispatch a ,const dispatch b)
    {
        if (a.level != b.level) return a.level > b.level;
        else return a.taskname > b.taskname;
    }
};

void split(string str, priority_queue<dispatch> &que,unordered_map<string,int>&level)
{
    string pretask;
    int prelevel = -1;
    int pos = str.find('(');
    for (int i = 0; i < pos; ++i)
        pretask += str[i];
    str.erase(str.begin(), str.begin() + pos + 1);
    str.erase(str.end() - 1, str.end());

    if (level.count(pretask) == 0)
    {
        dispatch pre;
        level[pretask] = 0;

        pre.taskname = pretask;
        pre.level = prelevel = 0;

        level[pretask] = prelevel;
        que.push(pre);
    }
    else
        prelevel = level[pretask];  

    while (str.size())
    {
        string nexttask;
        dispatch next;
        int pos = str.find(',');
        if (pos != string::npos)
        {
            for (int i = 0; i < pos; ++i)
                nexttask += str[i];
            str.erase(str.begin(), str.begin() + pos + 1);

        }
        else
        {
            nexttask = str;
            str.clear();
        }

        if (nexttask != "NULL")
        {
            next.taskname = nexttask;
            next.level = prelevel + 1;
            level[nexttask] = next.level;
            que.push(next);
        }
    }
}

int main()
{
#ifdef _DEBUG
    freopen("data.txt", "r+", stdin);
#endif // _DEBUG

    int n;
    string task;
    priority_queue<dispatch> que;
    unordered_map<string, int>level;
    while (cin >> n)
    {
        while (n--)
        {
            cin >> task;
            split(task, que, level);
        }
        while (!que.empty())
        {
            cout << que.top().taskname;
            que.pop();
            if (!que.empty())
                cout << " ";
        }
        cout << endl;
    }



    return 0;
}
/************************************************************** Problem: 1985 User: Sharwen Language: C++ Result: 升仙 Time:1 ms Memory:1732 kb ****************************************************************/