Problem Description
读入任务调度序列,输出n个任务适合的一种调度方式。
Input
输入包含多组测试数据。每组第一行输入一个整数n(n < 100000),表示有n个任务。接下来n行,每行第一个表示前序任务,括号中的任务为若干个后序任务,表示只有在前序任务完成的情况下,后序任务才能开始。若后序为NULL则表示无后继任务。
Output
输出调度方式,输出如果有多种适合的调度方式,请输出字典序最小的一种。
Sample Input
4
Task0(Task1,Task2)
Task1(Task3)
Task2(NULL)
Task3(NULL)
5
Task4(Task1,Task2)
Task2(Task3)
Task3(NULL)
Task1(Task0)
Task0(NULL)
4
Task3(Task2)
Task0(Task3,Task2)
Task1(Task2)
Task2(NULL)
5
Task1(Task2)
Task0(Task3,Task2)
Task2(Task4)
Task4(NULL)
Task3(NULL)
Sample Output
Task0 Task1 Task2 Task3
Task4 Task1 Task0 Task2 Task3
Task0 Task1 Task3 Task2
Task0 Task1 Task2 Task3 Task4
解题思路
- 将输入的数据转化为图:利用strtok_r()函数分离数字,数字即代表任务,也是图的顶点元素。
- 利用拓扑排序+小顶堆得到任务执行顺序并输出。
- 算法笔记p414拓扑排序&算法笔记p335堆
经验总结
char* strtok_r(char* str, const char* delim, char** saveptr);
-
strtok_r 是一个字符串分割函数,用于将字符串分割成多个子字符串。strtok_r 接受三个参数:
- str:一个指向要分割的字符串的指针
- delim:一个包含分隔符的字符串,用于指定分割子字符串的标记。在每次调用时,strtok_r 会查找 str 中的下一个分隔符。
- saveptr:一个指向指针的指针,用于存储分割状态。saveptr 保存被修改的 str 的值,以便下一次调用 strtok_r 时使用。
-
strtok_r 函数的返回值是一个指向当前分割子字符串的指针。
-
strtok_r 函数会修改传入的字符串,将分隔符替换为字符串结束符’\0’。
int strncmp(const char* str1, const char* str2, size_t n);
- strncmp 函数用于比较两个字符串的前 n 个字符是否相等,strncmp 函数接受三个参数:
- str1:要进行比较的第一个字符串。
- str2:要进行比较的第二个字符串。
- n:要比较的字符数。即比较字符串的前 n 个字符。
- 函数的返回值是一个整数,表示比较的结果:
- 如果 str1 和 str2 的前 n 个字符相等,则返回 0。
- 如果 str1 的前 n 个字符小于 str2 的前 n 个字符,则返回一个负整数。
- 如果 str1 的前 n 个字符大于 str2 的前 n 个字符,则返回一个正整数。
代码实现(C)
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdbool.h>
#define MaxSize 100000
// 图的邻接表实现数据结构定义
typedef int VertexType;
// 边表结点
typedef struct ArcNode {
int adjVertex; // 该弧所指向的顶点的位置
struct ArcNode *next; // 指向下一条弧的指针
} ArcNode;
// 顶点表结点
typedef struct VertexNode {
VertexType data; // 顶点信息
struct ArcNode *first; // 指向第一条依附该顶点的弧的指针
} AdjList;
// 图的链式存储(邻接表实现)
typedef struct ALGraph {
AdjList vertices[MaxSize]; // 邻接表
int verNum; // 顶点数
} ALGraph;
// 从字符串中提取数字
int extractTasks(const char *str, int *tasks) {
char *token; // 字符串分割后的前部
char *rest = (char *) str; // 字符串分割后的剩余部分
int taskNum = 0; // 字符串中数字的个数
// '('、')'、','这三个字符前不是TaskX就是NULL,因此以这三个字符为分割符
// strtok_r函数会把分割符替换为'\0'
// 如传入str="Task0(Task1,Task2)"
// 第一次执行strok_r,token="Task0",rest="Task1,Task2)"
while ((token = strtok_r(rest, "(),", &rest))) {
if (strncmp(token, "Task", 4) == 0) { // 如果分割出的字符串中前四个字符含有Task
int num = atoi(token + 4); // 将Task后的字符串转为数字
tasks[taskNum++] = num; // 存储到数组中
} // else token = "NULL"
}
return taskNum;
}
// 将输入的数据转化为图
ALGraph *inputDataToGraph(int n) {
// 声明图
ALGraph *G = (ALGraph *) malloc(sizeof(ALGraph));
// 吸收n后面的换行符
getchar();
// 赋值顶点数
G->verNum = n;
for (int i = 0; i < n; ++i) {
char str[MaxSize];
gets(str);
int task[MaxSize]; // 存放任务
// 提取任务个数
int taskNum = extractTasks(str, task);
// 给顶点表赋值
int index = task[0];
G->vertices[index].data = task[0];
ArcNode *first = (ArcNode *) malloc(sizeof(ArcNode));
first->next = NULL;
if (taskNum == 1) { // 当前任务无后序结点
G->vertices[index].first = NULL;
} else if (taskNum > 1) { // 连接第一个边表结点
first->adjVertex = task[1];
G->vertices[index].first = first;
}
// 连接后序边表结点
ArcNode *pre;
pre = first;
for (int j = 2; j < taskNum; ++j) {
ArcNode *nextArcNode = (ArcNode *) malloc(sizeof(ArcNode));
nextArcNode->next = NULL;
nextArcNode->adjVertex = task[j];
pre->next = nextArcNode;
pre = nextArcNode;
}
}
return G;
}
void swap(int *a, int *b) {
int t = *a;
*a = *b;
*b = t;
}
// 对heap数组在[low,high]范围进行向下调整
// 其中low为欲调整结点的数组下标,high一般为堆的最后一个元素的数组下标
void downAdjust(int low, int high, int *heap) {
int i = low, j = i * 2; // i为欲调整结点,j为其左孩子
while (j <= high) { // 存在孩子结点
// 如果右孩子存在,且右孩子的值小于左孩子
if (j + 1 <= high && heap[j + 1] < heap[j])
j = j + 1; // 让j存储右孩子下标
// 如果孩子中最小的权值比欲调整结点i小
if (heap[j] < heap[i]) {
swap(&heap[j], &heap[i]); // 交换最小权值的孩子与欲调整结点
i = j; // 保持i为欲调整结点
j = i * 2; // j为i的左孩子
} else
break; // 孩子的权值均比欲调整结点i大,则无需向下调整
}
}
// 建堆
void createHeap(int n, int *heap) {
for (int i = n / 2; i >= 1; --i)
downAdjust(i, n, heap);
}
// 对heap数组在[low,high]范围进行向上调整
// 其中low一般设置为1,high表示欲调整结点的数组下标
void upAdjust(int low, int high, int *heap) {
int i = high, j = i / 2; // i为欲调整结点,j为其父亲结点
while (j >= low) { // 父亲结点在[low,high]范围内
// 父亲权值大于欲调整结点i的权值
if (heap[j] > heap[i]) {
swap(&heap[j], &heap[i]); // 交换父亲结点和欲调整结点
i = j; // 保持i为欲调整结点
j = i / 2; // 保持j为i的父亲
} else
break; // 父亲的权值比欲调整结点i的权值小,调整结束
}
}
// 添加元素x
void insert(int x, int *n, int *heap) {
heap[++(*n)] = x; // 让元素个数加1,然后将数组末位赋值为x
upAdjust(1, *n, heap); // 向上调整新加入的结点n
}
// 删除堆顶元素
int deleteTop(int *n, int *heap) {
int top = heap[1];
heap[1] = heap[(*n)--]; // 用最后一个元素覆盖堆顶元素,并让元素个数减1
downAdjust(1, *n, heap); // 向下调整堆顶元素
return top;
}
// 入度
int inDegree[MaxSize];
// 计算图中各个顶点的入度
void countInDegree(ALGraph *graph) {
// 初始化
for (int i = 0; i < graph->verNum; ++i)
inDegree[i] = 0;
// 计算入度
for (int i = 0; i < graph->verNum; ++i) {
ArcNode *node = graph->vertices[i].first;
while (node) {
inDegree[node->adjVertex]++;
node = node->next;
}
}
}
// 拓扑排序
bool topologicalSort(ALGraph *G) {
int num = 0; // 记录加入拓扑排序的顶点数
int heapNum = 0; // 堆中元素的个数
int heapIndex = 1; // 堆中元素的起始下标
int heap[MaxSize]; // 小顶堆
countInDegree(G); // 计算图的入度
for (int i = 0; i < G->verNum; ++i)
if (inDegree[i] == 0) {
heap[heapIndex++] = i; //将所有入度为0的顶点入堆
heapNum++;
}
// 建堆
createHeap(heapNum, heap);
while (heapNum) { // 堆不为空
int u = deleteTop(&heapNum, heap); // 取堆顶元素u
ArcNode *i = G->vertices[u].first;
while (i) {
int v = G->vertices[i->adjVertex].data;
inDegree[v]--; // 顶点v的入度减1
if (inDegree[v] == 0) // 顶点v的入度减为0则入队
insert(v, &heapNum, heap);
i = i->next;
}
G->vertices[u].first = NULL; // 清空顶点u的所有出边(如无必要可不写)
num++; // 加入拓扑序列的顶点数加1
if (heapNum)
printf("Task%d ", u); // 输出顶点u,作为拓扑序列中的顶点
else
printf("Task%d\n", u);
}
if (num == G->verNum) return true; // 加入拓扑排序的顶点数为n,说明拓扑排序成功
else return false; // 加入拓扑排序的顶点数小于n,说明拓扑排序失败
}
int main() {
int n;
while (~scanf("%d", &n)) {
ALGraph *G = inputDataToGraph(n); // 将输入的数据转化为图
topologicalSort(G); // 拓扑排序
}
return 0;
}