拓扑排序算法应用题,助力理解算法
题目出处:
http://acm.hdu.edu.cn/showproblem.php?pid=1285
题目描述:
有N个比赛队(1<=N<=500),编号依次为1,2,3,. . . . ,N进行比赛,比赛结束后,裁判委员会要
将所有参赛队伍从前往后依次排名,但现在裁判委员会不能直接获得每个队的比赛成绩,只知道每场比赛的结果,即P1赢P2,用P1,P2表示,排名时P1在P2之前。现在请你编程序确定排名。
输入格式:
输入有若干组,每组中的第一行为二个数N(1<=N<=500),M;其中N表示队伍的个数,M表示接着有M行的输入数据。接下来的M行数据中,每行也有两个整数P1,P2表示即P1队赢了P2队。
输出格式:
给出一个符合要求的排名。输出时队伍号之间有空格,最后一名后面没有空格。
其他说明:符合条件的排名可能不是唯一的,此时要求输出时编号小的队伍在前;输入数据保证是正确
的,即输入数据确保一定能有一个符合要求的排名。
样例输入:
4 3
1 2
2 3
4 3
样例输出:
1 2 4 3
解析:
这是一道拓扑排序的模板题,我们需要用邻接表的方式去构造图,我用的是vector的方式存储图形,由于要优先输出编号小的队伍,所以用了优先队列,详细解析在代码里。
代码:
#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#include<functional>
using namespace std;
const int maxN = 505;
int InDeg[maxN];//记录所有点的入度
vector<int>vec[maxN];//邻接表存储图
priority_queue<int,vector<int>,greater<int> >q;//优先队列,以从小到大输出
int N, M;//输入对象
int u, v,num=0;//num记录已输出点的数量
void topsort()
{
num = 0;
for(int i=1;i<=N;i++)//遍历所有的点,入度为0的点就推进优先队列里
if (!InDeg[i])
{
q.push(i);
}
while (!q.empty())
{
int now = q.top();
q.pop();
num++;
if (num == N)
{
cout << now << endl;
break;
}
cout << now << " ";
for (int i = 0; i < vec[now].size(); i++)//将所有跟这个点链接的其他点的入度减一并判断是否度数变为0,变为0就入队
{
if (--InDeg[vec[now][i]] == 0)
{
q.push(vec[now][i]);
}
}
}
}
int main()
{
while (cin >> N >> M)//多组输入
{
for (int i = 1; i <= N; i++)//每次都需要清空上一组的内容
vec[i].clear();
memset(InDeg, 0, sizeof(InDeg));//置0
while (!q.empty())
q.pop();
for (int i = 1; i <= M; i++)
{
cin >> u >> v;
vec[u].push_back(v);
InDeg[v]++;
}
topsort();
}
return 0;
}