Codeup墓地-1985

Codeup墓地-1985
src:http://codeup.cn/problem.php?id=1985
1985: 任务调度
时间限制: 1 Sec 内存限制: 32 MB
题目描述
读入任务调度序列，输出n个任务适合的一种调度方式。

输入
输入包含多组测试数据。

每组第一行输入一个整数n（n<100000），表示有n个任务。

接下来n行，每行第一个表示前序任务，括号中的任务为若干个后序任务，表示只有在前序任务完成的情况下，后序任务才能开始。若后序为NULL则表示无后继任务。

输出
输出调度方式，输出如果有多种适合的调度方式，请输出字典序最小的一种。
样例输入
4
Task0(Task1,Task2)
Task1(Task3)
Task2(NULL)
Task3(NULL)
样例输出
Task0 Task1 Task2 Task3

//C++11 unordered_map priority_queue 难！
#include<bits/stdc++.h>
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;
        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(){
    int n;
    string task;
    priority_queue<dispatch> que;
    unordered_map<string, int>level;
    while (~scanf("%d",&n)){
        getchar();
        while (n--){
            getline(cin,task);
            split(task, que, level);
        }
        while (!que.empty()){
            printf("%s",que.top().taskname.c_str());
            que.pop();
            if (!que.empty())printf(" ");
        }
        printf("\n");
    }
    return 0;
}